LightOJ 1262 Diagonal Sum最大ストリーム


タイトル:http://www.lightoj.com/volume_showproblem.php?problem=1262
n*mの行列があり、すべての斜め対角線の和を与え、すべての要素は[1,100]の間にこの行列を復元します。
考え:zjbztianyaが集まっているコードを見ました。いい料理ですね。
#include 
#include 
#include 
#include 
#include 
#define debug() puts("here");
using namespace std;

const int N = 210;
const int INF = 0x3f3f3f3f;
struct edge
{
    int to, cap, id, next;
} g[N*N*2];
int cnt, nv, head[N], level[N], gap[N], cur[N], pre[N];
int ans[N*N];
int cas;
void add_edge(int v, int u, int cap)
{
    g[cnt].to = u, g[cnt].cap = cap, g[cnt].next = head[v], head[v] = cnt++;
    g[cnt].to = v, g[cnt].cap = 0, g[cnt].next = head[u], head[u] = cnt++;
}
int sap(int s, int t)
{
    memset(level, 0, sizeof level);
    memset(gap, 0, sizeof gap);
    memcpy(cur, head, sizeof head);
    gap[0] = nv;
    int v = pre[s] = s, flow = 0, aug = INF;
    while(level[s] < nv)
    {
        bool flag = false;
        for(int &i = cur[v]; i != -1; i = g[i].next)
        {
            int u = g[i].to;
            if(g[i].cap > 0 && level[v] == level[u] + 1)
            {
                flag = true;
                pre[u] = v;
                v = u;
                aug = min(aug, g[i].cap);
                if(v == t)
                {
                    flow += aug;
                    while(v != s)
                    {
                        v = pre[v];
                        g[cur[v]].cap -= aug;
                        g[cur[v]^1].cap += aug;
                    }
                    aug = INF;
                }
                break;
            }
        }
        if(flag) continue;
        int minlevel = nv;
        for(int i = head[v]; i != -1; i = g[i].next)
        {
            int u = g[i].to;
            if(g[i].cap > 0 && minlevel > level[u])
                minlevel = level[u], cur[v] = i;
        }
        if(--gap[level[v]] == 0) break;
        level[v] = minlevel + 1;
        gap[level[v]]++;
        v = pre[v];
    }
    return flow;
}
int main()
{
    int t, n, m, a;
    scanf("%d", &t);
    while(t--)
    {
        cnt = 0;
        memset(head, -1, sizeof head);
        scanf("%d%d", &n, &m);
        int ss = 0, tt = 2 * (n+m-1) + 1;
        for(int i = 1; i <= n + m - 1; i++)
        {
            scanf("%d", &a);
            a -= min(n, i) - max(0, i-m);
            add_edge(ss, i, a);
        }
        for(int i = 1; i <= n + m - 1; i++)
        {
            scanf("%d", &a);
            a -= min(n, i) - max(0, i-m);
            add_edge(i + n+m-1, tt, a);
        }
        for(int i = 1; i <= n; i++)
            for(int j = 1; j <= m; j++)
                add_edge(i+j-1, m+i-j + n+m-1, 99), ans[(i-1)*m+j] = cnt - 1;
        nv = tt + 1;
        sap(ss, tt);
        printf("Case %d:
", ++cas); for(int i = 1; i <= n; i++) for(int j = 1; j <= m; j++) printf("%d%c", g[ans[(i-1)*m+j]].cap + 1, j == m ? '
' : ' '); } return 0; }