[ZJOI2009]假期的宿舍

OJ题号：

洛谷2055、BZOJ1433

思路：

将需要住宿的学生与超极源点$S$连一条边，将所有的床与超极汇点$T$连一条边。
将每个人与可以睡的床连一条边。
最大流跑二分图最大匹配即可。
连边时注意细节。

#include<queue>
#include<cstdio>
#include<cctype>
#include<vector>
#include<cstring>
inline int getint() {
    char ch;
    while(!isdigit(ch=getchar()));
    int x=ch^'0';
    while(isdigit(ch=getchar())) x=(((x<<2)+x)<<1)+(ch^'0');
    return x;
}
const int inf=0x7fffffff;
int s,t;
struct Edge {
    int from,to;
    bool remain;
};
const int E=5200,V=202;
Edge e[E];
std::vector<int> g[V];
int sz;
inline void add_edge(const int u,const int v,const bool w) {
    e[sz]=(Edge){u,v,w};
    g[u].push_back(sz);
    sz++;
}
int p[V];
bool a[V];
inline bool Augment() {
    memset(a,0,sizeof a);
    a[s]=inf;
    std::queue<int> q;
    q.push(s);
    while(!q.empty()&&!a[t]) {
        int x=q.front();
        q.pop();
        for(unsigned i=0;i<g[x].size();i++) {
            Edge &y=e[g[x][i]];
            if(!a[y.to]&&y.remain) {
                p[y.to]=g[x][i];
                a[y.to]=a[x]&&y.remain;
                q.push(y.to);
            }
        }
    }
    return a[t];
}
inline int EdmondsKarp() {
    int maxflow=0;
    while(Augment()) {
        for(int i=t;i!=s;i=e[p[i]].from) {
            e[p[i]].remain^=true;
            e[p[i]^1].remain^=true;
        }
        maxflow++;
    }
    return maxflow;
}
inline void reset() {
    sz=0;
    for(int i=0;i<V;i++) {
        g[i].clear();
    }
}
int main() {
    for(int T=getint();T;T--) {
        reset();
        int n=getint();
        s=0,t=n<<1|1;
        bool isStudent[n+1];
        for(int i=1;i<=n;i++) {
            if(isStudent[i]=(bool)getint()) {
                add_edge(n+i,t,true);
                add_edge(t,n+i,false);
            }
        }
        int cnt=0;
        for(int i=1;i<=n;i++) {
            if(!getint()||!isStudent[i]) {
                add_edge(s,i,true);
                add_edge(i,s,false);
                cnt++;
            }
        }
        for(int i=1;i<=n;i++) {
            for(int j=1;j<=n;j++) {
                if(getint()||(i==j)) {
                    add_edge(i,n+j,true);
                    add_edge(n+j,i,false);
                }
            }
        }
        puts(EdmondsKarp()==cnt?"^_^":"T_T");
    }
    return 0;
}