[洛谷P3941] 入阵曲

题目背景

丹青千秋酿，一醉解愁肠。
无悔少年枉，只愿壮志狂。

入阵曲

题解在代码里。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
long long a[1000][1000],sum[1000][1000],ans=0,tot,n,m,k,flag[1100000],b[1110000];
int main()
{
    cin>>n>>m>>k;
    for(int i=1;i<=n;i++)
    for(int j=1;j<=m;j++)
    {
        scanf("%lld",&a[i][j]);
        sum[i][j]=sum[i][j-1]+a[i][j]+sum[i-1][j]-sum[i-1][j-1];//sum数组存矩阵的和；
    }
    for(int i=0;i<n;i++)//枚举行
    for(int j=i+1;j<=n;j++)//枚举行
    {
        //memset(flag,0,sizeof(flag));//不能用memset，速度太慢，会超时13个点
        flag[0]=1;//因为mod k等于0说明它自己就能组成一个矩阵mod k等于0；所以一开始赋值为1
        for(int t=1;t<=m;t++)//枚举列
        {
            b[t]=(sum[i][t]-sum[j][t])%k;
            if(b[t]<0) b[t]+=k;//可能小于0，所以要加k变为正数
            ans+=flag[b[t]];//关键地方，前面算出有多少mod k余吧b[t]，当前 b[t]-之前mod k余b[t]的可以得到一个mod k余0的矩阵；
            flag[b[t]]++;//因为此时多了一个上面得出来的；
        }
        for(int t=1;t<=m;t++)
        flag[b[t]]=0;//清空flag数组
    }
    printf("%lld",ans);
    return 0;
}

博主蒟蒻，可以随意转载，但必须附上原文链接k-z-j。