题目链接:洛谷P1361
题意:给定两块地A,B,以及编号1-n的n个种子,每个种子种在A,B地里可以收获不同的价值,不同的种子组合放在不同地块可以产生不同的额外价值。要求最大的价值
思路:将所有种子看成一个点,分别和A,B地块连边,边的容量为价值,不同种子组合则构造新的两个虚点,并分别和A,B地块、组合中的种子连边,其中和种子连边时,边权设为inf,这样可以避免边被删除,之后暴力跑dinic求最小割即可
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
#include <map>
#include <queue>
#include <functional>
#include <vector>
#include <stack>
#include <set>
#include <bitset>
using namespace std;
typedef long long ll;
const int maxn=9000+50;
const ll inf=0x3f3f3f3f3f;
const int MOD=998244353;
const int HASH=131;
int n, m, cnt;//点,边,前向星计数
int head[maxn];//前向星
int level[maxn];//分层
int cur[maxn];//当前弧优化
struct Edge
{
int to;
ll val;
int next;
}edge[maxn*maxn];
void add(int u,int v,ll val)
{
edge[cnt].to=v;
edge[cnt].val=val;
edge[cnt].next=head[u];
head[u]=cnt++;
/*反向存边,在原边基础上+1*/
edge[cnt].to=u;
edge[cnt].val=0;
edge[cnt].next=head[v];
head[v]=cnt++;
}
bool find_level(int s,int t)//源点和汇点,该bfs函数用来确定深度(层次)
{
queue<int> q;
memset(level,0,sizeof(level));
for(int i=1;i<=n;i++)
{
cur[i]=head[i];
}
int u=s;
level[u]=1;
q.push(u);
while(!q.empty())
{
u=q.front();
q.pop();
for(int i=head[u];~i;i=edge[i].next)
{
int v=edge[i].to;
ll val=edge[i].val;
if(!level[v]&&val)//层次不为0且容量不为0
{
level[v]=level[u]+1;
q.push(v);
}
}
if(level[t])//如果t有分层则继续下一步的处理
{
return true;
}
}
return false;
}
ll updata(int u,ll u_val,int t)//dfs更新
{
if(u==t)
{
return u_val;
}
ll used=0;//使用多少容量
for(int &i=cur[u];~i;i=edge[i].next)//弧优化,神奇
{
int v=edge[i].to;
ll val=edge[i].val;
if(level[v]==level[u]+1&&val)//如果是相邻两层且有剩余容量
{
ll tmp=updata(v,min(u_val-used,val),t);//dfs递归,找最小容量
edge[i].val-=tmp;
edge[i^1].val+=tmp;
used+=tmp;
if(used==u_val) return used;//达到最大值,本条增广路搜索完毕
}
}
if(used==0) level[u]=-1;//找不到增广路,减枝,节点作废
return used;
}
ll Dinic(int s,int t)
{
ll ans=0;
while(find_level(s,t))
{
ans+=updata(s,inf,t);
}
return ans;
}
void init()
{
cnt=0;
memset(head,-1,sizeof(head));
}
int main()
{
init();
cin>>n;
ll sum=0;//求出所有收益
for(int i=2;i<=n+1;i++)
{
int tmp;
cin>>tmp;
add(1,i,tmp);
sum+=tmp;
}
for(int i=2;i<=n+1;i++)
{
int tmp;
cin>>tmp;
add(i,n+2 ,tmp);
sum+=tmp;
}
cin>>m;
n=n+2;
int t=n;
int s=1;
for(int i=1;i<=m*2;i+=2)
{
int k,c1,c2;
cin>>k>>c1>>c2;
sum+=c1+c2;
add(1,i+n,c1);//构造虚点1
add(i+n+1,n,c2);//构造虚点2
for(int j=1;j<=k;j++)
{
int tmp;
cin>>tmp;
add(i+n,tmp+1,inf);//虚点和组内每个点连接,权值赋为inf
add(tmp+1,i+n+1,inf);
}
}
n=n+2*m;
cout<<sum-Dinic(s,t)<<endl;//所有收益减去最小割得到的就是答案
return 0;
}