UVALive-2531 The K-League （最大流建模+枚举）

题目大意：有n支足球队，已知每支球队的已胜场数和任意两支球队之间还需要的比赛场数a[i][j]，求最终可能夺冠的所有球队。

题目分析：枚举所有的球队，对于球队 i 让它在接下来的比赛中全部获胜，如果这样球队 i 还不能夺冠，那么 i 不是要找的，否则，就是答案。这道题可以建立公平分配问题的模型。将任意两个之间还有比赛的球队视作一个节点X（由u和v构成，即a[u][v]>0），则对于X有两个选择：u获胜或v获胜。这样的一个X相当于一个“任务”，对于每个任务X有两个“处理器”（u和v）供选择。建图如下：增加源点s和汇点t，由s出发向所有的X均连一条弧，容量为a[u][v]，然后由X出发向其对应的u和v各连一条容量为无穷大的弧，最后对每个球队u都连一条指向汇点t的弧，容量为total-win(u)，其中，total为 i 已经获胜的场数，win(u)为u已经获胜的场数。当源点出发的所有弧都饱和时，当前枚举的球队 i 则可能夺冠。

代码如下：

# include<iostream>
# include<cstdio>
# include<cmath>
# include<string>
# include<vector>
# include<list>
# include<set>
# include<map>
# include<queue>
# include<cstring>
# include<algorithm>
using namespace std;

# define LL long long
# define REP(i,s,n) for(int i=s;i<n;++i)
# define CL(a,b) memset(a,b,sizeof(a))
# define CLL(a,b,n) fill(a,a+n,b)

const double inf=1e30;
const int INF=1<<30;
const int N=1000;

struct Edge
{
int fr,to,cap,fw;
Edge(int _fr,int _to,int _cap,int _fw):fr(_fr),to(_to),cap(_cap),fw(_fw){}
};
vector<Edge>edges;
vector<int>G
;
int W
,L
,n,cur
,vis
,d
;
int a[30][30],s,t,ans[30];

void addEdge(int u,int v,int cap)
{
edges.push_back(Edge(u,v,cap,0));
edges.push_back(Edge(v,u,0,0));
int m=edges.size();
G[u].push_back(m-2);
G[v].push_back(m-1);
}

void read()
{
scanf("%d",&n);
REP(i,0,n) scanf("%d%d",W+i,L+i);
REP(i,0,n) REP(j,0,n) scanf("%d",&a[i][j]);
}

void init()
{
s=0,t=n*n+n+1;
edges.clear();
REP(i,0,t+1) G[i].clear();
}

bool BFS()
{
CL(vis,0);
queue<int>q;
vis[s]=1;
d[s]=0;
q.push(s);
while(!q.empty()){
int u=q.front();
q.pop();
REP(i,0,G[u].size()){
Edge &e=edges[G[u][i]];
if(!vis[e.to]&&e.cap>e.fw){
vis[e.to]=1;
d[e.to]=d[u]+1;
q.push(e.to);
}
}
}
return vis[t];
}

int DFS(int u,int a)
{
if(u==t||a==0) return a;
int flow=0,f;
for(int &i=cur[u];i<G[u].size();++i){
Edge &e=edges[G[u][i]];
if(d[e.to]==d[u]+1&&(f=DFS(e.to,min(a,e.cap-e.fw)))>0){
e.fw+=f;
edges[G[u][i]^1].fw-=f;
flow+=f;
a-=f;
if(a==0) break;
}
}
return flow;
}

int Dinic()
{
int flow=0;
while(BFS()){
CL(cur,0);
flow+=DFS(s,INF);
}
return flow;
}

void solve()
{
ans[0]=0;
REP(k,0,n){
int total=W[k];
REP(i,0,n) total+=a[k][i];
int flag=0;
REP(i,0,n) if(W[i]>total){
flag=1;
break;
}
if(flag) continue;
init();
int full=0;
REP(i,0,n){
REP(j,i+1,n){
full+=a[i][j];
if(a[i][j]>0) addEdge(s,i*n+j+1,a[i][j]);
addEdge(i*n+j+1,n*n+i+1,INF);
addEdge(i*n+j+1,n*n+j+1,INF);
}
if(W[i]<total) addEdge(n*n+i+1,t,total-W[i]);
}
int flow=Dinic();
if(flow==full) ans[++ans[0]]=k+1;
}
REP(i,1,ans[0]+1) printf("%d%c",ans[i],(i==ans[0])?'\n':' ');
}

int main()
{
int T;
scanf("%d",&T);
while(T--)
{
read();
solve();
}
　　return 0;
}