P2763 试题库问题

题目

题目

思路

显然又是一道恶心OIer喜闻乐见的拆点+输出方案。
code:

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#include<cmath>
#include<queue>
#include<deque>
using namespace std;
long long n1,n2,m,s,t,x,y,tot=2,head[4101],dep[4101],l,r,u[4101],r2;
long long w,ans;
long long mn(long long x,long long y)
{
	return (x>y?y:x);
}
struct f{
	long long to,net;
	long long w;
} a[100011];
void add(long long x,long long y,long long w)
{
	a[tot].to=y,a[tot].w=w,a[tot].net=head[x],head[x]=tot++;
	return;
}
bool bfs()
{
	memset(dep,0,sizeof(dep));
	l=0,r=0;
	u[r++]=s;
	dep[s]=1;
	while (l<r)
	{
		r2=r;
		for (long long i=l;i<r;i++)
		{
			for (long long j=head[u[i]];j;j=a[j].net)
			{
				if (a[j].w!=0&&dep[a[j].to]==0)
				{
					u[r2++]=a[j].to;
					dep[a[j].to]=dep[u[i]]+1;
				}
			}
		}
		l=r,r=r2;
	}
	return dep[t];
}
long long yyy;
long long anss[277001],yyyyy[277001];
long long dfs(long long d,long long in)
{
	if (d==t)
	{
		return in;
	}
	long long out=0;
	long long uw=dep[d]+1;
	for (long long j=head[d];j&&in;j=a[j].net)
	{
		if (a[j].w==0||dep[a[j].to]!=uw) continue;
		long long s=dfs(a[j].to,mn(in,a[j].w));
		out+=s,in-=s;
		a[j].w-=s,a[j^1].w+=s;
	}
	if (out==0)
	{
		dep[d]=0;
	}
	return out;
}
int main()
{
	scanf("%lld%lld",&n1,&n2);
	long long ss=0;
	for (int i=1;i<=n1;i++)
	{
		cin>>yyyyy[i];
		ss+=yyyyy[i];
	}
	s=n1+2*n2+1,t=n1+2*n2+2;
	for (int i=1;i<=n1;i++)
	{
		add(s,i,yyyyy[i]);
		add(i,s,0);
	}
	for (int j=n1+1;j<=n1+n2;j++)
	{
		add(j,j+n2,1);
		add(j+n2,j,0);
		add(j+n2,t,1);
		add(t,j+n2,0);
	}
	for (int j=1;j<=n2;j++)
	{
		cin>>x;
		for (int i=1;i<=x;i++)
		{
			cin>>y;
			add(y,n1+j,1);
			add(n1+j,y,0);
		}
	}
	while (bfs())
	{
		ans+=dfs(s,1e20);
	}
	if (ans!=ss)
	{
		cout<<"No Solution!";
		return 0;
	}
	for (int i=1;i<=n1;i++)
	{
		cout<<i<<':';
		for (int j=head[i];j;j=a[j].net) if (a[j].w==0&&a[j].to-n1<=n2) cout<<' '<<a[j].to-n1;
		cout<<endl;
	}
    return 0;
}