codeforces#290 Fox And Dinner

传送门：点击打开链接

题意：给出n个数，要求将这n个数分成若干组，每组数的个数>=3并且围成一个环，要求环内任意相邻的两个数之和为质数。（ai>=2)

思路：首先相邻的两个数之和必须为一个奇数，所以一个奇数必须和两个偶数相邻，一个偶数必须和两个奇数相邻，并且可以推出任意一个环中count(奇数)==count(偶数）。这就变成了一个匹配的问题。而且一脸最大流的样子：

    所有奇数点和源点相连，容量为2；所有偶数点和汇点相连，容量为2；所有奇数点和满足和值为质数的偶数点相连（单向），容量为1,。跑出来的最大流为n则说明有解，否则无解。另外需要注意的是，生成方案时，需要在最大流跑完之后，这时非0的边才表示最大流中的边。中间过程的非0边有可能在后面被回退。

代码写的太啰嗦了：

#include<iostream>
#include<stdio.h>
#include<string.h>
#include<vector>
using namespace std;

int d[300][300];
int f[300][300];
int vis[20005];
int prime[20005];
int a[300];
int n,T;
int pre[300];
int q[1000];

int MaxFlow(){
	int maxflow=0;
	while(1){
		memset(pre,0,sizeof(pre));
		memset(vis,0,sizeof(vis));
		int head=1,rear=1;
		q[rear]=0;
		vis[0]=1;
		int ok=0;
		while(head<=rear){
			int t=q[head];
			head++;
			for(int i=1;i<=T;i++){
				if(f[t][i]<d[t][i]&&!vis[i]){
					rear++;
					q[rear]=i;
					vis[i]=1;
					pre[i]=t;
					if(i==T){
					   	ok=1;
						break;
					}
				}
			}
			if(ok) break;
		}
		if(ok){
			maxflow++;
			int t=T;
			while(t){
				f[pre[t]][t]++;
				f[t][pre[t]]--;
				t=pre[t];
			}
		}
		else{
			break;
		}
	}
	return maxflow;
}

int cnt[300];
int num;
int ans[300][300];
void dfs(int cur){
	if(vis[cur]) return;
	cnt[num]++;
	vis[cur]=1;
	ans[num][cnt[num]]=cur;
	for(int i=1;i<=n;i++){
		if(!vis[i]&&f[cur][i]){
			dfs(i);
		}
	}
}
			
int main(){
	int i,j;
	memset(prime,0,sizeof(prime));
	memset(vis,0,sizeof(vis));
	for(i=2;i<=20000;i++){
		if(!vis[i]){
			prime[i]=1;
			for(j=i;j<=20000;j+=i){
				vis[j]=1;
			}
		}
	}
	scanf("%d",&n);
	for(i=1;i<=n;i++){
		scanf("%d",&a[i]);
	}
	if(n&1){
		printf("Impossible\n");
		return 0;
	}
	T=n+1;
	for(i=1;i<=n;i++){
		if(a[i]&1){
			d[0][i]=2;
		}
		else{
			d[i][T]=2;
		}
		for(j=i+1;j<=n;j++){
			if(prime[a[i]+a[j]]){
				if(a[i]&1) d[i][j]=1;
				else d[j][i]=1;

			}
		}
	}
	int flow=MaxFlow();
	if(flow==n){
		memset(vis,0,sizeof(vis));
		vis[0]=1;
		vis[T]=1;
		for(i=1;i<=n;i++){	
			if(!vis[i]){
				num++;
				dfs(i);
			}
		}
		printf("%d\n",num);
		for(i=1;i<=num;i++){
			printf("%d ",cnt[i]);
			for(j=1;j<=cnt[i];j++){
				printf("%d ",ans[i][j]);
			}
			printf("\n");
		}
	}
	else{
		printf("Impossible\n");
	}
	return 0;
}