QAQ
性质,如果一个 a1%k==a2%k,那么这两个数的和就是k的倍数
预处理矩阵和,然后枚举2个行,1个列 转换成上面的问题
#include <cstdio>
#include <iostream>
#include <cstring>
#define ll long long
using namespace std;
ll map[500][500],s[500][500];
ll cnt[1100000],b[1100000];
int main()
{
int n,m,k;
ll ans=0;
scanf("%d%d%d",&n,&m,&k);
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
scanf("%d",&map[i][j]);
for(int i=1;i<=n;i++)
for(int j=1;j<=m;j++)
s[i][j]=s[i-1][j]+s[i][j-1]-s[i-1][j-1]+map[i][j];//子矩阵和
for(int i=0;i<n;i++)
for(int j=i+1;j<=n;j++)
{
cnt[0]=1;//0本身可以是种方案,也可以与0的搭配。。
for(int l=1;l<=m;l++)
{
b[l]=(s[j][l]-s[i][l]+k)%k;
ans+=cnt[b[l]];//与每一个-都会出k的倍数
cnt[b[l]]++;
}
for(int l=1;l<=m;l++)
cnt[b[l]]=0;//用memset超时(废话)
}
printf("%lld\n",ans);
return 0;
}
QAQ
贪深度深的点。
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
const int maxm=1e5+100;
bool is[maxm],vis[maxm];
struct node{
int id,deep;
}a[maxm];
int head[maxm],net[2*maxm],to[2*maxm];
int cnt;
void add(int x,int y)
{
cnt++;
to[cnt]=y;
net[cnt]=head[x];
head[x]=cnt;
}
void get_deep(int x,int fat)
{
vis[x]=1;
for(int i=head[x];i;i=net[i])
{
int p=to[i];
//printf("%d\n",p);
if(p==fat) continue;
//if(p==fat) continue;
a[p].deep=a[x].deep+1;
get_deep(p,x);
}
}
void dfs(int x,int dis,int fat)
{
vis[x]=1;is[x]=1;
if(dis==0) return;
for(int i=head[x];i;i=net[i])
{
int p=to[i];
if(p==fat) continue;
dfs(p,dis-1,x);
}
}
bool comp(node x,node y)
{
return x.deep>y.deep;
}
int main()
{
int n,k,t;
scanf("%d%d%d",&n,&k,&t);
for(int i=1;i<n;i++)
{
int d,c;
cin>>d>>c;
add(d,c),add(c,d);
//printf("-%d %d %d-\n",i,d,c);
a[i].id=i;
}
a[n].id=n;
get_deep(1,0);
sort(a+1,a+n+1,comp);
/*for(int i=head[1];i;i=net[i])
printf("%d ",to[i]);*/
int ans=0;
for(int i=1;i<=n;i++)
if(!is[a[i].id])
{
ans++;
dfs(a[i].id,2*k,0);
memset(vis,0,sizeof(vis));
}
/*for(int i=1;i<=n;i++)
printf("%d %d\n",a[i].id,a[i].deep); */
return printf("%d",ans)*0;
}