规律就是杨辉三角啊.
对于一个矩阵a,我们能在O(1)时间内求出子矩阵[x1x2][y1y2]的和。
设sum[i][j]为子矩阵[1i][1j]的和。则由容斥原理得:
sum[0][j]=sum[i][0]=0
a[x1x2][y1y2]=sum[x2][y2]-sum[x1-1][y2]-sum[x2][y1-1]+sum[x1-1][y1-1]。
其实就是降低时间复杂度,拿空间换时间。每一个元素先开始预处理得到以(1,1)为左上角,自身为右下角的矩阵里所有元素之和。
之后你要是想计算某个区间内的所有元素之和就是:
s[x2][y2]+s[x1-1][y1-1]-s[x1][y1-1]-s[x1-1][y1]
#include<iostream>
using namespace std;
long long t,n,m,k,f[2005][2005],s[2005][2005];
int main()
{
cin>>t>>k;
for(int i=0;i<=2000;i++)
{
f[i][i]=1;
f[i][0]=1;
}
for(int i=1;i<=2000;i++)
for(int j=1;j<i;j++)
{
f[i][j]=(f[i-1][j-1]+f[i-1][j])%k;
}
for(int i=1;i<=2000;i++)
for(int j=1;j<=2000;j++)
{
s[i][j]=s[i][j-1]+s[i-1][j]-s[i-1][j-1];
if(f[i][j]==0&&j<=i)
s[i][j]++;
}
while(t--)
{
cin>>n>>m;
cout<<s[n][m]<<endl;
}
}