2023“钉耙编程”中国大学生算法设计超级联赛（2）Coin

2023“钉耙编程”中国大学生算法设计超级联赛（2）Coin

题目大意

小$F$组织了一场游戏，有$n$个人参与。开始时，每人都有一枚硬币，且每人都有一个数$a_i$表示他能持有硬币的上限。

游戏有$m$轮，在每一轮中，小$F$会选择不同的两个人$A$和$B$，然后这两个人有三种选择：

• $A$给$B$一枚硬币
• $B$给$A$一枚硬币
• 他们什么都不做

在这$n$个人中，有$k$个人是小$F$的朋友。现在小$F$想知道，在经过$m$轮游戏后，在所有可能的情况下，他的$k$个朋友所持有的硬币的最大总数是多少。

有$T$组数据。

$1\le T\le 10,1\le n,m\le 3000,1\le k\le n$

题解

这道题可以用网络流来解决。

我们先建立一个最大流模型：

• 源点$S$向每一个人连一条流量为$1$的边
• 将每个人按时间顺序拆成$m$个点，$m$个点从前往后依次连一条流量为$a_i$的边
• 对于每次操作给定的$(A,B)$，在各自对应时间点上的点连两条流量为$1$的边（正向边流量为$1$，反向边流量也为$1$）
• 对于$k$个朋友点，他们的最后一个时间的点向汇点$T$连一条流量为$a_i$的边

求一个最大流，即可得到答案。

但是，这样建图的话，点的数量和边的数量都为$O(n\times m)$，我们考虑优化。

我们发现，在图上很多点都是没有用的。所以，我们并不需要将每个人都拆成$m$个点，只需在每个时间将给定的$(A,B)$中的$A$和$B$都新开一个点，再连边即可。

这样的话，点数为$O(m)$，边数为$O(m+k)$，时间复杂度不超过$O(n\times m)$。

code

#include<bits/stdc++.h>
using namespace std;
const int inf=1e9;
int tq,n,m,k,x,y,s,t,cnt=0,a[3005],b[3005],lst[3005];
int tot=1,d[100005],l[100005],r[100005],w[100005];
int ans,dis[100005],vd[100005];
void add(int xx,int yy,int zz){
	l[++tot]=r[xx];d[tot]=yy;r[xx]=tot;w[tot]=zz;
}

int aug(int i,int augco){
	if(i==t) return augco;
	int augc=augco,md=cnt-1,dl;
	for(int u=r[i];u;u=l[u]){
		int j=d[u];
		if(w[u]>0){
			if(dis[i]==dis[j]+1){
				dl=aug(j,min(augc,w[u]));
				augc-=dl;
				w[u]-=dl;
				w[u^1]+=dl;
				if(dis[s]>=cnt) return augco-augc;
				if(!augc) break;
			}
			md=min(md,dis[j]);
		}
	}
	if(augco==augc){
		--vd[dis[i]];
		if(!vd[dis[i]]) dis[s]=cnt;
		dis[i]=md+1;
		++vd[dis[i]];
	}
	return augco-augc;
}
void sap(){
	memset(dis,0,sizeof(dis));
	memset(vd,0,sizeof(vd));
	ans=0;vd[0]=cnt;
	while(dis[s]<cnt){
		ans+=aug(s,inf);
	}
}
int main()
{
	scanf("%d",&tq);
	while(tq--){
		scanf("%d%d%d",&n,&m,&k);
		s=++cnt;t=++cnt;
		for(int i=1;i<=n;i++){
			scanf("%d",&a[i]);
			lst[i]=++cnt;
			add(s,lst[i],1);add(lst[i],s,0);
		}
		for(int i=1;i<=m;i++){
			scanf("%d%d",&x,&y);
			add(lst[x],++cnt,a[x]);add(cnt,lst[x],0);
			lst[x]=cnt;
			add(lst[y],++cnt,a[y]);add(cnt,lst[y],0);
			lst[y]=cnt;
			add(lst[x],lst[y],1);add(lst[y],lst[x],1);
		}
		for(int i=1;i<=k;i++){
			scanf("%d",&b[i]);
			add(lst[b[i]],t,a[b[i]]);add(t,lst[b[i]],0);
		}
		sap();
		printf("%d\n",ans);
		tot=1;cnt=0;
		memset(r,0,sizeof(r));
	}
}