蒟蒻求教卡空间
查看原帖
蒟蒻求教卡空间
116015
bellmanford楼主2020/5/3 09:25

我感觉开的不大,用的是并查集,但还是MLE了。。。

代码:

#include<iostream>
#include<cstdio>
using namespace std;

const int M=305,N=2e6+5;

int n,m,q,a[M][M];
int fa[N],Fa[N],ans[N],Ans[N];
int nxt[4][2]={{1,0},{0,1},{-1,0},{0,-1}};
bool b[M][M];
struct Query{
	int x,y,c,nc;
}Q[N];

int read(){
	int x=0,y=1;
	char ch=getchar();
	while(ch<'0'||ch>'9'){
		if(ch=='-') y=-1;
		ch=getchar();
	}
	while(ch>='0'&&ch<='9'){
		x=x*10+ch-'0';
		ch=getchar();
	}
	return x*y;
}

int num(int x,int y){
	return n*(x-1)+y;
}

int find(int x){
	if(x!=fa[x]) fa[x]=find(fa[x]);
	return fa[x];
}

int Find(int x){
	if(x!=Fa[x]) Fa[x]=Find(Fa[x]);
	return Fa[x];
}

void solve(int l,int r){
	for(int i=l;i<=r;i++){
		int x=Q[i].x,y=Q[i].y;
		if(b[x][y]){
			continue ;
		}
		b[x][y]=1;
		ans[i]=1;
		for(int k=0;k<4;k++){
			int nx=x+nxt[k][0],ny=y+nxt[k][1];
			if(nx<1||nx>n||ny<1||ny>m) continue ;
			if(b[nx][ny]){
				int Fx=Find(num(x,y)),Fy=Find(num(nx,ny));
				if(Fx==Fy) continue ;
				ans[i]--;
				Fa[Fx]=Fy;
			}
		}
	}
//	printf("%d %d:",l,r);
//	for(int i=l;i<=r;i++) printf("%d ",ans[i][0]);printf("\n");
	for(int i=l;i<=r;i++){
		int x=Q[i].x,y=Q[i].y;
		b[x][y]=0;
		Fa[num(x,y)]=num(x,y);
	}
	for(int i=r;i>=l;i--){
		int x=Q[i].x,y=Q[i].y;
		if(a[x][y]==Q[i].nc){
			continue ;
		}
		a[x][y]=Q[i].nc;
		ans[i]--;
		fa[num(x,y)]=0;
		for(int k=0;k<4;k++){
			int nx=x+nxt[k][0],ny=y+nxt[k][1];
			if(nx<1||nx>n||ny<1||ny>m) continue ;
			if(a[x][y]==a[nx][ny]){
				if(fa[num(x,y)]==0){
					fa[num(x,y)]=find(num(nx,ny));
					ans[i]++;
					continue ;
				}
				int fx=find(num(x,y)),fy=find(num(nx,ny));
				if(fx==fy) continue ;
				ans[i]++;
				fa[fx]=fy;
			}
		}
		if(fa[num(x,y)]==0){
			fa[num(x,y)]=num(x,y);
		}
	}
//	printf("%d %d:",l,r);
//	for(int i=l;i<=r;i++) printf("%d ",ans[i][1]);printf("\n");
//	for(int i=r;i>=l;i--){
//		if(i==q) continue ;
//		Ans[i]=Ans[i+1]-anss;
//	}
}

int main(){
	n=read(),m=read(),q=read();
	for(int i=1;i<=q;i++){
		Q[i].x=read();
		Q[i].y=read();
		Q[i].c=read();
		Q[i].nc=a[Q[i].x][Q[i].y];
		a[Q[i].x][Q[i].y]=Q[i].c;
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++){
			fa[num(i,j)]=Fa[num(i,j)]=num(i,j);
		}
	}
	Ans[q]=0;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++){
			Ans[q]++;
			for(int k=0;k<4;k++){
				int nx=i+nxt[k][0],ny=j+nxt[k][1];
				if(nx<1||nx>n||ny<1||ny>m) continue ;
				if(a[i][j]==a[nx][ny]){
					int fx=find(num(i,j)),fy=find(num(nx,ny));
					if(fx==fy) continue ;
					Ans[q]--;
					fa[fx]=fy;
				}
			}
		}
	}
//	printf("%d\n",Ans[q]);
	int last=q;
	for(int i=q-1;i>=1;i--){
		if(Q[i].c!=Q[last].c){
			solve(i+1,last);
			last=i;
		}
	}
	solve(1,last);
	for(int i=q-1;i>=1;i--){
		Ans[i]=Ans[i+1]-ans[i+1];
	}
	int pre=1;
	for(int i=1;i<=q;i++){
		printf("%d\n",Ans[i]);
	}
	return 0;
}
2020/5/3 09:25
加载中...