codeforces+B. Divide Candies+取模

题目链接：http://codeforces.com/contest/1056/problem/B
题目大意：给你一个n阶矩阵，aij=i^2 + j^2。
现在给你n，和一个m问有多少元素能整除m(n<=1e9,m<=1000)

(1,2) (2,1)1^2 + 2^2=5, 5%5==0
(1,3)(3,1)
(2,4)(4,2)
(2,6)(6,2)
(3,4)(4,3)
(3,6)(6,3)
(5,5)

思路：由于a,b的数量级很大,而m的数量级很小,又因为求((a * a)+(b * b))%m == 0,即求 ((a%m * a%m)+(b%m * b%m))%m == 0。
所以预先处理m*m的矩阵就会发现。

矩阵是重复的。

所以就很好计算了

分成三部分

思路：mod知识很重要

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

int a[1005][1005];
int main()
{
    fill(a[0], a[0]+1005*1005, 0);
    int n, m;
    ll cut=0;
    scanf("%d%d",&n,&m);

    for(int i=1;i<=m;i++)
    {
        for(int j=1;j<=m;j++)
        {
            int s=i*i+j*j;
            if(s%m==0)
                a[i][j]=1, cut++;
        }
    }

    cut*=(ll)(n/m)*(n/m);/*计算重复的矩阵1*/

    ll s=0;
    for(int i=1;i<=n%m;i++)/*计算剩余的右半部分2*/
    {
        for(int j=1;j<=m;j++)
        {
            s+=a[i][j]+a[j][i];
        }
    }
    s*=n/m;
    for(int i=1;i<=n%m;i++)/*计算剩余的左下半部分3*/
    {
        for(int j=1;j<=n%m;j++)
        {
            s+=a[i][j];
        }
    }
    cout<<cut+s<<endl;

    return 0;

}