CF510E. Fox And Dinner

CF510E. Fox And Dinner

https://codeforces.com/contest/510

分析:

• 由于(a_i>2)， 相邻两个数一定一奇一偶，按奇偶建立二分图。
• 环上每个点度数都为2，因此只需要找是否每个点都能匹配两个。
• 建图跑dinic即可。

代码:

#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <algorithm>
#include <cmath>
using namespace std;
#define N 1050
#define M 2000050
#define S 1049
#define T 1048
#define inf 0x3f3f3f3f
int head[M],nxt[M],to[M],cnt=1,n,flow[M];
int dep[N],Q[N],a[N];
int vis[M],pri[M],pri_cnt,ch[N][2];
int vec[N][N],vsize[N],inv[N];
void sieve() {
	int i,j,lim=15000;
	for(i=2;i<=lim;i++) {
		if(!vis[i]) {
			pri[++pri_cnt]=i;
		}
		for(j=1;j<=pri_cnt&&i*pri[j]<=lim;j++) {
			vis[i*pri[j]]=1;
			if(i%pri[j]==0) break;
		}
	}
}
inline void add(int u,int v,int f) {
	to[++cnt]=v; nxt[cnt]=head[u]; head[u]=cnt; flow[cnt]=f;
	to[++cnt]=u; nxt[cnt]=head[v]; head[v]=cnt; flow[cnt]=0;
}
bool bfs() {
	int i;
	memset(dep,0,sizeof(dep));
	int l=0,r=0;
	Q[r++]=S; dep[S]=1;
	while(l<r) {
		int x=Q[l++];
		for(i=head[x];i;i=nxt[i]) if(flow[i]&&!dep[to[i]]) {
			dep[to[i]]=dep[x]+1;
			if(to[i]==T) return 1;
			Q[r++]=to[i];
		}
	}
	return 0;
}
int dfs(int x,int mf) {
	int i,nf=0;
	if(x==T) return mf;
	for(i=head[x];i;i=nxt[i]) if(flow[i]&&dep[to[i]]==dep[x]+1) {
		int tmp=dfs(to[i],min(mf-nf,flow[i]));
		if(!tmp) dep[to[i]]=0;
		nf+=tmp;
		flow[i]-=tmp;
		flow[i^1]+=tmp;
		if(nf==mf) break;
	}
	return nf;
}
void dinic() {
	while(bfs()) {
		while(dfs(S,inf)>0);
	}
}
bool check(int x) {
	int i;
	for(i=1;i<=pri_cnt&&pri[i]*pri[i]<=x;i++) if(x%pri[i]==0) return 0;
	return 1;
}
int main() {
	sieve();
	scanf("%d",&n);
	int i,j,x;
	for(i=1;i<=n;i++) scanf("%d",&a[i]);
	for(i=1;i<=n;i++) {
		if(!(a[i]&1)) {
			add(S,i,2);
			for(j=1;j<=n;j++) if(i!=j&&(a[j]&1)) {
				
				if(check(a[i]+a[j])) {
					add(i,j,1);
					if(i==1&&j==3) puts("FUCK");
				}
			}
		}else {
			add(i,T,2);
		}
	}
	dinic();
	int ERR=0;
	for(x=1;x<=n;x++) {
		if(!(a[x]&1)) {
			for(i=head[x];i;i=nxt[i]) if(to[i]!=S) {
				if(flow[i]==0) {
					if(!ch[x][0]) ch[x][0]=to[i];
					else ch[x][1]=to[i];
				}
			}
			if(!ch[x][1]) {ERR=1; break;}
		}else {
			for(i=head[x];i;i=nxt[i]) if(to[i]!=T) {
				if(flow[i]==1) {
					if(!ch[x][0]) ch[x][0]=to[i];
					else ch[x][1]=to[i];
				}
			}
			if(!ch[x][1]) {ERR=1; break;}
		}
	}
	if(ERR) {puts("Impossible"); return 0;}
	memset(vis,0,sizeof(vis));
	int tot=0;
	for(i=1;i<=n;i++) if(!vis[i]) {
		int tp=0;
		memset(inv,0,sizeof(inv));
		vec[++tot][++tp]=i; inv[i]=1;
		for(j=1;;j++) {
			int p=vec[tot][j];
			inv[p]=1;
			if(inv[ch[p][0]]&&inv[ch[p][1]]) break;
			if(inv[ch[p][0]]) {
				vec[tot][++tp]=ch[p][1];
			}else {
				vec[tot][++tp]=ch[p][0];
			}
		}
		for(j=1;j<=tp;j++) vis[vec[tot][j]]=1;
		vsize[tot]=tp;
	}
	printf("%d
",tot);
	for(i=1;i<=tot;i++) {
		printf("%d ",vsize[i]);
		for(j=1;j<=vsize[i];j++) {
			printf("%d ",vec[i][j]);
		}
		puts("");
	}
}