1550: Simple String 最大流解法

http://acm.csu.edu.cn/csuoj/problemset/problem?pid=1550

很久以前做的一题，当时队友用最大流做，现在我也是

这个转化为二分图多重匹配，就是一样的意思了。

设出一个原点S，两个人1,和2，S-->1的流量是n表明只能流出n个字母，s-->2的流量是n，一样含义。

然后再设一个汇点T，表明需要有多少流进来，跑一发就好。

#include <bits/stdc++.h>
#define IOS ios::sync_with_stdio(false)
using namespace std;
#define inf (0x3f3f3f3f)
typedef long long int LL;
const int maxn =  * ;
struct Edge {
    int u, v, w, tonext;
    int id;
}e[maxn * ];
int first[maxn], num;
void addEdge(int u, int v, int w) {
    e[num].u = u, e[num].v = v, e[num].w = w, e[num].tonext = first[u], e[num].id = num;
    first[u] = num++;
}
int flow[maxn], pre[maxn];
int bfs(int be, int en) { //O(n * E * E)
    queue<int> que;
    memset(flow, false, sizeof flow);
    pre[be] = -, flow[be] = inf;
    que.push(be);
    while (!que.empty()) {
        int cur = que.front();
        que.pop();
        for (int i = first[cur]; ~i; i = e[i].tonext) {
            int v = e[i].v;
            if (flow[v] ==  && e[i].w > ) {
                pre[v] = e[i].id;
                flow[v] = min(flow[cur], e[i].w);
                que.push(v);
            }
        }
        if (flow[en]) break; //找到了增广路
    }
    if (flow[en] == ) return -;
    else return flow[en];
}

int maxFlow(int be, int en) {
    int sumFlow = ;
    while (true) {
        int res = bfs(be, en);
        if (res == -) { //找不到增广路
            break;
        }
        int edgeID = pre[en];
        while (edgeID != -) {
            e[edgeID].w -= res;
            e[edgeID ^ ].w += res;
            edgeID = pre[e[edgeID].u];
        }
        sumFlow += res;
    }
    return sumFlow;
}
char s1[maxn], s2[maxn], s3[maxn];
int cnt1[], cnt2[], cnt3[];
void work() {
    memset(cnt1, , sizeof cnt1);
    memset(cnt2, , sizeof cnt2);
    memset(cnt3, , sizeof cnt3);
    memset(first, -, sizeof first);
    num = ;
    int lenstr = strlen(s1 + );
    addEdge(, , lenstr / );
    addEdge(, , );
    addEdge(, , lenstr / );
    addEdge(, , );
    for (int i = ; i <= lenstr; ++i) {
        cnt1[s1[i]]++;
        cnt2[s2[i]]++;
        cnt3[s3[i]]++;
    }
    for (int i = 'A'; i <= 'Z'; ++i) {
        addEdge(, i - 'A' + , cnt1[i]);
        addEdge(i - 'A' + , , );
        addEdge(, i - 'A' + , cnt2[i]);
        addEdge(i - 'A' + , , cnt2[i]);
    }
    for (int i = 'A'; i <= 'Z'; ++i) {
        addEdge(i - 'A' + , , cnt3[i]);
        addEdge(, i - 'A' + , );
    }
    int res = maxFlow(, );
    if (res == lenstr) {
        printf("YES\n");
    } else printf("NO\n");
}

int main() {
#ifdef local
    freopen("data.txt", "r", stdin);
//    freopen("data.txt", "w", stdout);
#endif
    while (scanf("%s%s%s", s1 + , s2 + , s3 + ) > ) work();
    return ;
}