【二分】Pythagorean Triples—CF1487D

Pythagorean Triples---CF1487D

思路

联立 a 2 = b + c 2 a^2 = b + c^2 a2=b+c2、 a 2 + b 2 = c 2 a^2 + b^2 = c^2 a2+b2=c2 得:
a 2 = 2 ∗ c − 1 a^2 = 2 * c - 1 a2=2∗c−1、 b = c − 1 b = c - 1 b=c−1。

对于一个固定的 a a a, b b b、 c c c 的值都是固定的,只要满足 a ≤ b a\le b a≤b 和 c ≤ n c\le n c≤n 即可。

使用二分可以求出对应的 c c c 的值满足条件的 a a a 的取值范围,另外要排除 a = 1 , 2 a = 1, 2 a=1,2 这两个值,因为这时候 a > b a > b a>b。

C o d e Code Code

cpp 复制代码
#include <bits/stdc++.h>
#define int long long
#define sz(a) ((int)a.size())
#define all(a) a.begin(), a.end()
using namespace std;
using PII = pair<int, int>;
using i128 = __int128;
const int N = 2e5 + 10;

int n;

// c的值是否符合条件(先不考虑奇数这个条件)
int judge(int a) {
	if ((a * a + 1) / 2 <= n
		&& a <=(a * a + 1) / 2 - 1) {
		return 1;
	}
	return 0;
}

void solve() {
	cin >> n;
	cout << "       ";
	
	if (n <= 2) {
		cout << 0 << "\n";
		return;
	}
	
	// 二分求a的最大值(先不考虑得到的c是否是整数)
	int l = 3, r = n;
	while (l < r) {
		int mid = (l + r + 1) / 2;
		if (judge(mid)) {
			l = mid;
		} else {
			r = mid - 1;
		}
	}
	if (l % 2 == 0) {
		l --;
	}
	
	// 现在满足条件的a的取值范围是[3, l]中的所有奇数
	if (judge(l)) {
		cout << l / 2 << "\n";
	} else {
		cout << "0\n";
	}
}

signed main() {
	ios::sync_with_stdio(0), cin.tie(0), cout.tie(0);
	int T = 1;
	cin >> T; cin.get();
	while (T --) solve();
	return 0;
}
相关推荐
鄃鳕44 分钟前
python 字典 列表 类比c++【python】
c++·python
im_AMBER44 分钟前
算法笔记 05
笔记·算法·哈希算法
夏鹏今天学习了吗1 小时前
【LeetCode热题100(46/100)】从前序与中序遍历序列构造二叉树
算法·leetcode·职场和发展
吃着火锅x唱着歌1 小时前
LeetCode 2389.和有限的最长子序列
算法·leetcode·职场和发展
嶔某1 小时前
二叉树的前中后序遍历(迭代)
算法
WWZZ20251 小时前
快速上手大模型:机器学习2(一元线性回归、代价函数、梯度下降法)
人工智能·算法·机器学习·计算机视觉·机器人·大模型·slam
孤狼灬笑1 小时前
深度学习经典分类(算法分析与案例)
rnn·深度学习·算法·cnn·生成模型·fnn
dragoooon341 小时前
[优选算法专题四.前缀和——NO.26二维前缀和]
算法
保持低旋律节奏2 小时前
C++——list链表
c++·链表·list
苏小瀚3 小时前
算法---位运算
java·算法