题目链接: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;
}