题目链接:http://acm.nyist.net/JudgeOnline/problem.php?pid=489
大意:给一个n*m的方格,每个小方格只能填0或1, 现在给你每一行的和, 每一列的和, 问你是否存在这样的填法。如果有输出Not Sure , 否则输出 Terrible。
这是网络流中的最大流的一种的经典的用法(方格填充的可行性)。 建图方法, 创建一个超级源点和汇点, 把每一行看成一个点, 把每个点都与源点连一条边,容量为这一行的和; 把每一列看成一个点, 把每个点都与汇点连一条边,容量为这一列的和;然后在把每一行和每一列连一条边,容量为1(相当于第i行第j列的小方格可以填1)。 当然如果行的总和与列的总和不相等直接就可以判不可以。
#include<stdio.h>
#include<string.h>
#include<vector>
#include<queue>
#include<algorithm>
using namespace std;
const int maxn = 1000 + 10;
const int inf = 0x3f3f3f3f;
struct Edge
{
int from, to, cap, flow;
Edge(int u, int v, int c, int f) : from(u), to(v), cap(c), flow(f) {}
};
struct Dinic
{
int s, t, n, m;
vector<int> G[maxn];
vector<Edge> edges;
int d[maxn], cur[maxn];
bool inq[maxn];
void inin(int n)
{
this->n = n;
for(int i = 0; i <= n; i++)
G[i].clear();
edges.clear();
}
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(inq, false, sizeof(inq));
queue<int> Q;
d[s] = 0;
Q.push(s);
inq[s] = true;
while(!Q.empty())
{
int x = Q.front();
Q.pop();
for(int i = 0; i < G[x].size(); i++)
{
Edge& e = edges[G[x][i]];
if(!inq[e.to] && e.cap>e.flow)
{
d[e.to] = d[x] + 1;
Q.push(e.to);
inq[e.to] = true;
}
}
}
return inq[t];
}
int dfs(int x, int a)
{
if(x==t || a==0)
return a;
int f, flow = 0;
for(int& i = cur[x]; i < G[x].size(); i++)
{
Edge& e = edges[G[x][i]];
if(d[e.to]==d[x]+1 && (f = dfs(e.to, min(a, e.cap-e.flow)))>0)
{
e.flow += f;
flow += f;
edges[G[x][i]^1].flow -= f;
a -= f;
if(a == 0)
break;
}
}
return flow;
}
int max_flow(int s, int t)
{
this->s = s;
this->t = t;
int flow = 0;
while(bfs())
{
memset(cur, 0, sizeof(cur));
flow += dfs(s, inf);
}
return flow;
}
};
int main()
{
int T, n, m, s, t, sum1, sum2;
Dinic solve;
scanf("%d", &T);
while(T--)
{
scanf("%d%d", &n, &m);
s = 0, t = m + n + 1;
solve.inin(t);
sum1 = 0;
for(int i = 1; i <= n; i++)
{
int a;
scanf("%d", &a);
sum1 += a;
solve.AddEdge(s, i, a);
for(int j = 1; j <= m; j++)
solve.AddEdge(i, j+n, 1);
}
sum2 = 0;
for(int i = 1; i <= m; i++)
{
int a;
scanf("%d", &a);
sum2 += a;
solve.AddEdge(n+i, t, a);
}
bool flag = false;
if(sum1 == sum2)
{
int ans = solve.max_flow(s, t);
if(ans == sum1)
flag = true;
}
if(flag)
puts("Not Sure");
else
puts("Terrible");
}
return 0;
}