Divide Candies CodeForces - 1056B(思维+数学)

最新推荐文章于 2024-08-29 08:15:00 发布

原创最新推荐文章于 2024-08-29 08:15:00 发布 · 398 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#math

收录于

本文介绍了一种解决特定数学问题的方法：给定a、b的范围及模数m，求解(a² + b²) % m == 0的整数解数量。通过预处理余数出现次数并利用双重循环来高效计算所有符合条件的对数。

传送门

题意：给出a,b，范围为[1,n]，然后问(a^2+b^2)%m==0的情况有多少个，n<=1e9，m<=1e3

题解：(a^2+b^2)%m=((a^2)%m+(b^2)%m)%m=((a%m*a%m)%m+(b%m*b%m)%m)%m，所以可以通过枚举取余m的情况，然后统计个数即可。

附上代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1050;
int n,m;
int cnt[N];
ll ans;
int main(void){
    scanf("%d%d",&n,&m);
    //处理出1到n中 %m余数为i的数的个数
    //处理前面km个数
    for(int i=0;i<m;i++){
        cnt[i]=n/m;
    }
    //处理后面n-km个数
    for(int i=(n/m)*m+1;i<=n;i++){
        cnt[i%m]++;
    }
    //枚举余数即可
    for(int i=0;i<m;i++){
        for(int j=0;j<m;j++){
            if(((i*i)%m+(j*j)%m)%m==0){
                ans+=(ll)cnt[i]*cnt[j];
            }
        }
    }
    printf("%lld\n",ans);
    return 0;
}