B. Divide Candies （*1600）

ACM真是一只大坑啊，这道数学题想了好久，还是看了题解才会做，真心难受

题意：给定数n，m，求(1,1)->(n,n)之间所有符合条件的数对(x,y)的个数，条件为(x^2+y^2)%m == 0

思路：预处理a^2%m，记录下0~m-1的结果然后计算

向着大佬的方向追赶

https://codeforces.com/contest/1056/problem/B

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<set>
#include<map>
#include<queue>
#include<cmath>
#define inf 0x3f3f3f3f
#define ll long long
#define mod 1000000007
using namespace std;

int main()
{
    ll m, n;
    vector<int> v[1005]; //保存“平方之后的余数”相同的k,k为a%m，b%m之类的数
    //因此之后把a%m当成a处理就ok
    int y[1005];//保存k的平方%m的结果
    scanf("%I64d%I64d",&n,&m);
    for(int i = 0; i < m; i ++)
    {
        v[(i*i)%m].push_back(i);
        y[i] = (i*i)%m;
    }
    ll ans = 0;
    for(int i = 0; i < m; i ++)//i表示a%m的结果，y[i]表示a*a%m的结果，k表示b*b%m的结果
    {
        int k = (m-y[i])%m;//若y[i]==0，k==0，否则k+y[i]==m
        for(int j = 0; j < v[k].size(); j ++)
        {
            ans += (n/m+(((n%m)>=i)&&i>0))*(n/m+(((n%m)>=v[k][j])&&v[k][j]>0));//v[k][j]表示b%m之后的结果
        }
    }
    printf("%I64d\n",ans);
    return 0;
}