舞蹈链板子

#include<bits/stdc++.h>
using namespace std;
#define rg register
inline int read(){
	rg char ch=getchar();
	rg int x=0,f=0;
	while(!isdigit(ch)) f|=(ch=='-'),ch=getchar();
	while(isdigit(ch)) x=(x<<1)+(x<<3)+(ch^48),ch=getchar();
	return f?-x:x;
}
int n,m;
int mp[550][550],ans[550],anscnt,cnt,row,y[5500];
int c[550],u[5500],d[5500],l[5500],r[5500],x[5500];
void init(){
//	memset(u,0,sizeof u);memset(d,0,sizeof d);
//	memset(l,0,sizeof l);memset(r,0,sizeof r);
//	memset(ans,0,sizeof ans);memset(c,0,sizeof c);
//	memset(x,0,sizeof x);memset(y,0,sizeof y);
	anscnt=0;row=0,cnt=m;
	for(int i=0;i<=m;++i)
		l[i]=i-1,r[i]=i+1,u[i]=d[i]=i;
	l[0]=m;r[m]=0;
}
void link(int i,int j){
	++cnt;
	x[cnt]=i;y[cnt]=j;
	d[cnt]=j;
	d[u[j]]=cnt;
	u[cnt]=u[j];
	u[j]=cnt;
	l[cnt]=cnt-1;
	r[cnt]=cnt+1;
	++c[j];
}
void del(int k){
	l[r[k]]=l[k];r[l[k]]=r[k];
	for(int i=d[k];i!=k;i=d[i]){
		for(int j=r[i];j!=i;j=r[j]){
			u[d[j]]=u[j];
			d[u[j]]=d[j];
			--c[y[j]];
		}
	}
}
void reserve(int k){
	l[r[k]]=k;r[l[k]]=k;
	for(int i=u[k];i!=k;i=u[i]){
		for(int j=l[i];j!=i;j=l[j]){
			u[d[j]]=j;
			d[u[j]]=j;
			++c[y[j]];
		}
	}
}
bool solve(){
	if(r[0]==0) return true;
	int k=r[0];
	for(int i=r[0];i!=0;i=r[i])
		k=c[i]<c[k]?i:k;
	del(k);
	++anscnt;
	for(int i=d[k];i!=k;i=d[i]){
		ans[anscnt]=x[i];
		for(int j=r[i];j!=i;j=r[j])
			del(y[j]);
		if(solve()) return true;
		for(int j=l[i];j!=i;j=l[j])
			reserve(y[j]);
	}
	--anscnt;
	reserve(k);
	return false;
}

int main(){
	n=read(),m=read();
	init();
	for(int i=1;i<=n;++i) 
		for(int j=1;j<=m;++j)
			mp[i][j]=read();
	for(int first,i=1;i<=n;++i){
		first=cnt+1;
		for(int j=1;j<=m;++j){
			if(mp[i][j]){
				link(i,j);
			}
		}
		r[cnt]=first;l[first]=cnt;
	}
	if(solve()){
		for(int i=1;i<=anscnt;++i){
			printf("%d ",ans[i]);
		}
	}else printf("No Solution!");
	return 0;
}

posted @ 2022-07-23 16:20  __int256  阅读(32)  评论(0编辑  收藏  举报