题目描述
\(n\)个宝石,\(m\)个保安,每个保安监控着一些宝石。
偷走第\(i\)个宝石能卖\(a[i]\)元,贿赂第\(i\)个保安需要\(b[i]\)元,你能偷走某个宝石当且仅当监控它的保安都被你贿赂过了。
问你的最大收益是多少。
输入
第一行包含一个正整数\(T(1\leq T\leq 10)\),表示测试数据的组数。
每组数据第一行包含两个正整数\(n,m(1\leq n,m\leq 50)\)。
第二行包含\(n\)个正整数\(a[1],a[2],\dots,a[n](1\leq a[i]\leq 10000)\)。
第三行包含\(m\)个正整数\(b[1],b[2],\dots,b[m](1\leq b[i]\leq 10000)\)。
接下来\(n\)行,每行一个长度为\(m\)的01串,如果第\(i\)行第\(j\)个字符为'1',说明第\(i\)个宝石被第\(j\)个保安监控着,否则说明第\(i\)个宝石没有被第\(j\)个保安监控着。
输出
对于每组数据输出一行一个整数,即最大收益。
题解:考虑以及获得了所有的宝石,此时删去一些必要的边作为代价,总收入减去花费的最小代价就是最大流
#include<bits/stdc++.h>
using namespace std;
typedef vector<string> VS;
const int N = 110,INF = 1e9;
struct Edge {
int from, to, cap, flow;
Edge(int u, int v, int c, int f) : from(u), to(v), cap(c), flow(f) {}
};
bool operator < (const Edge& a, const Edge& b) {
return a.from < b.from || (a.from == b.from && a.to < b.to);
}
struct ISAP {
int n, m, s, t;
vector<Edge> edges;
vector<int> G[N];
bool vis[N];
int d[N],cur[N],p[N],num[N];
void AddEdge(int from, int to, int cap) {
edges.push_back(Edge(from, to, cap, 0));
edges.push_back(Edge(to, from, 0, 0));
m = edges.size();
G[from].push_back(m - 2);
G[to].push_back(m - 1);
}
bool BFS() {
memset(vis, 0, sizeof(vis));
queue<int> Q;
Q.push(t);
vis[t] = 1;
d[t] = 0;
while (!Q.empty()) {
int x = Q.front();
Q.pop();
for (int i = 0; i < G[x].size(); i++) {
Edge& e = edges[G[x][i] ^ 1];
if (!vis[e.from] && e.cap > e.flow) {
vis[e.from] = 1;
d[e.from] = d[x] + 1;
Q.push(e.from);
}
}
}
return vis[s];
}
void init(int n) {
this->n = n;
for (int i = 0; i < n; i++) G[i].clear();
edges.clear();
}
int Augment() {
int x = t, a = INF;
while (x != s) {
Edge& e = edges[p[x]];
a = min(a, e.cap - e.flow);
x = edges[p[x]].from;
}
x = t;
while (x != s) {
edges[p[x]].flow += a;
edges[p[x] ^ 1].flow -= a;
x = edges[p[x]].from;
}
return a;
}
int Maxflow(int s, int t) {
this->s = s;
this->t = t;
int flow = 0;
BFS();
memset(num, 0, sizeof(num));
for (int i = 0; i < n; i++) num[d[i]]++;
int x = s;
memset(cur, 0, sizeof(cur));
while (d[s] < n) {
if (x == t) {
flow += Augment();
x = s;
}
int ok = 0;
for (int i = cur[x]; i < G[x].size(); i++) {
Edge& e = edges[G[x][i]];
if (e.cap > e.flow && d[x] == d[e.to] + 1) {
ok = 1;
p[e.to] = G[x][i];
cur[x] = i;
x = e.to;
break;
}
}
if (!ok) {
int m = n - 1;
for (int i = 0; i < G[x].size(); i++) {
Edge& e = edges[G[x][i]];
if (e.cap > e.flow) m = min(m, d[e.to]);
}
if (--num[d[x]] == 0) break;
num[d[x] = m + 1]++;
cur[x] = 0;
if (x != s) x = edges[p[x]].from;
}
}
return flow;
}
};
ISAP FG;
int a[N],b[N];
void solve()
{
int n,m;
cin>>n>>m;
VS g(n);
int s = 0,t = n+m+1;
FG.init(n+m+2);
int sum = 0;
for(int i=1;i<=n;++i) cin>>a[i], sum +=a[i];
for(int i=1;i<=m;++i) cin>>b[i];
//建立从源点到宝石的边
for(int i=1;i<=n;++i) FG.AddEdge(s,i,a[i]);
//建立从警察到汇点的边
for(int i=1;i<=m;++i) FG.AddEdge(i+n,t,b[i]);
//注意g[i]的下标是[0,n-1]
for(int i=0;i<n;++i) cin>>g[i];
//若有宝石和警察的关系建一条INF边不能删除
for(int i=0;i<n;++i)
for(int j=0;j<m;++j)
if(g[i][j]=='1') FG.AddEdge(i+1,j+1+n,INF);
//一条S->T的路径表示保留的一个宝石同时没有贿赂守护其的警察是非法方案
cout<<sum - FG.Maxflow(s,t)<<'\n';
}
int main()
{
int T;
cin>>T;
while(T--)
{
solve();
}
}
标签:宝石,int,flow,最小,转化,leq,edges,Edge
From: https://www.cnblogs.com/ruoye123456/p/18370547