【codechef】判断一个数是否为某组勾股数的最大数？

最新推荐文章于 2021-02-26 07:29:15 发布

ccyy-

最新推荐文章于 2021-02-26 07:29:15 发布

阅读量1.4k

点赞数 1

分类专栏：困难数论

本文链接：https://blog.csdn.net/cacyth/article/details/49535313

版权

困难同时被 2 个专栏收录

109 篇文章 0 订阅

订阅专栏

数论

15 篇文章 0 订阅

订阅专栏

如果一个数有 4n+1 形式的素因子，那么它可以是某组勾股数里最大的那个数。
否则不是。

https://www.codechef.com/OCT15/problems/ADTR

vector<bool>generate(int n){
    vector<bool> input(n + 1, true);
    vector<bool> ans(n + 1, false);    
    int sqrtN = (int)sqrt(n);
    for (int i = 2; i <= sqrtN; i ++) { //素数打表（高效）
        if(!input[i]) 
            continue;
        for(int j = i * i; j <= n; j += i) 
            input[j] = false;
    }
    for(long int i = 2; i < n ;i++){
    	if(input[i]){
    		if(i%4==1){
    			for(long int j=i;j<=n;j+=i) //因子符合，所有倍数都符合
    				ans[j] = true; 
    		}
    	}
    }
    return ans;
}
int main(){
	vector<bool> primes = generate(5000005);
	if(primes[a])
		printf("YES\n");
	else
		printf("NO\n");