codeforces-546E. Soldier and Traveling(网络流)

记录一个菜逼的成长。。

题目链接

51nod上的翻译

这种从一个初始状态到达一个目标状态的，从感觉上来说有很多种可能的，可以用网络流。

网络流最重要的是建图。

源点与城市的边的容量是 a[i]

城市与汇点的边的容量是 b[i]

城市与自身的边是 INF ，表示可以不移动。

城市与相邻的城市的容量也是 INF ,表示可以移动到相邻的城市。

然后跑Dinic求最大流。

如果等于 ∑a[i] 和 ∑b[i] 则说明是可以的。

i 城市与j城市之间的边的容量减少的值就是从i移动到j的士兵数量

#include <bits/stdc++.h>
using namespace std;
#define fin freopen("D://in.txt","r",stdin)
#define fout freopen("D://out.txt","w",stdout)
typedef long long LL;
const int INF = ;
const int MAX_V =  + ;
int a[MAX_V],b[MAX_V];
int ans[MAX_V][MAX_V];
struct edge{
  int to,cap,rev;
  edge(){}
  edge(int _to,int _cap,int _rev):to(_to),cap(_cap),rev(_rev){}
};
vector<edge>G[MAX_V];
int level[MAX_V];
int iter[MAX_V];
int n,m;
void add(int from,int to,int cap)
{
  G[from].push_back(edge(to,cap,G[to].size()));
  G[to].push_back(edge(from,,G[from].size()-));
}
void bfs(int s)
{
  memset(level,-,sizeof(level));
  queue<int>que;
  level[s] = ;
  que.push(s);
  while(!que.empty()){
    int f = que.front();
    que.pop();
    for( int i = ; i < G[f].size(); i++ ){
      edge &e = G[f][i];
      if(e.cap >  && level[e.to] == -){
        level[e.to] = level[f] + ;
        que.push(e.to);
      }
    }
  }
}
int dfs(int v,int t,int f)
{
  if(v == t)return f;
  for( int &i = iter[v]; i < G[v].size(); i++ ){
    edge &e = G[v][i];
    if(e.cap >  && level[v] < level[e.to]){
      int d = dfs(e.to,t,min(e.cap,f));
      if(d > ){
        e.cap -= d;
        G[e.to][e.rev].cap += d;
        return d;
      }
    }
  }
  return ;
}
int max_flow(int s,int t)
{
  int flow = ;
  for(;;){
    bfs(s);
    if(level[t] == -)return flow;
    memset(iter,,sizeof(iter));
    int f;
    while((f = dfs(s,t,INF)) > )
      flow += f;
  }
}
int main()
{
  while(~scanf("%d%d",&n,&m)){
    int suma = ,sumb = ;
    int s = ,e = *n+;
    for( int i = ; i <= n; i++ ){
      scanf("%d",a+i);
      add(,i,a[i]);
      suma += a[i];
    }
    for( int i = ; i <= n; i++ ){
      scanf("%d",b+i);
      add(i+n,e,b[i]);
      add(i,i+n,INF);
      sumb += b[i];
    }
    for( int i = ; i < m; i++ ){
      int u,v;
      scanf("%d%d",&u,&v);
      add(u,v+n,INF);
      add(v,u+n,INF);
    }
    int flow = max_flow(s,e);
    for( int i = ; i <= n; i++ ){
      for( int j = ; j < G[i].size(); j++ ){
        if(G[i][j].to == e)continue;
        ans[i][G[i][j].to-n] = INF - G[i][j].cap;
      }
    }
    int flag = ;
    if(flow == suma && flow == sumb)flag = ;

    if(flag){
      puts("YES");
      for( int i = ; i <= n; i++ ){
        for( int j = ; j <= n; j++ ){
          printf("%d ",ans[i][j]);
        }
        puts("");
      }
    }
    else {
      puts("NO");
    }
  }
  return ;
}