线性规划与网络流24题の7 试题库问题（二分图匹配）

有n道题和k种类型，每道题可以属于多种类型。给出试卷中每种题目的数量vali，val1+...+valk就是试卷的总题数。输出试卷的方案。

这道题也是方案不唯一的。。。

从源点向每个类型连一条容量为vali的边；

从每到题目向汇点连一条容量为1的边（每道题的数目不能重复计算）

如果题目i属于类型j，就从j向i连一条容量为1的边。

然后求最大流，如果最大流等于试卷总题数就有解。

输出方案的时候，从每个类型出发，满流的边所对应的题目就是属于该类型的。

//#pragma comment(linker,"/STACK:1024000000,1024000000")
#include <map>
#include <set>
#include <cmath>
#include <queue>
#include <stack>
#include <cstdio>
#include <string>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;

typedef double DB;
typedef long long ll;
typedef pair<int, int> PII;

#define pb push_back
#define MP make_pair
#define lson l, m, rt << 1
#define rson m + 1, r, rt << 1 | 1

const DB eps = 1e-6;
const int inf = ~0U>>1;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const int mod = 1000000007;
const int maxn = 40000 + 10;


///init是初始化要在加边之前初始化，然后调用max_flow(顶点数，边数，源点， 汇点)
const int maxv = 40000 + 10;///顶点个数
const int maxe = 1000000 + 10;///边数

int c[maxe];

struct node{
    int v, cap, next;
}edge[maxe];
int head[maxv], cnt;
int n;///n是节点个数，m是边数
int st, ed;///st是源点，ed是汇点
int gap[maxv], h[maxv];
void addedge(int u, int v, ll w){///有向图加边
    edge[cnt].v = v; edge[cnt].cap = w; edge[cnt].next = head[u]; head[u] = cnt++;///正向边
    edge[cnt].v = u; edge[cnt].cap = 0; edge[cnt].next = head[v]; head[v] = cnt++;///反向边
}
int dfs(int x, int cost){
    if(x == ed) return cost; ///当前节点是汇点，直接返回cost

    int can = cost, d, minh = n - 1;
    for(int i=head[x]; ~i; i=edge[i].next){
        int v = edge[i].v, w = edge[i].cap;
        if(w > 0){///如果这条边的容量大于0
            if(h[v] + 1 == h[x]){///如果这是允许弧
                if(can > w) d = w;///如果当前弧的容量小于之前可增广的容量
                else d = can;

                d = dfs(v, d);///从v开始可增广的容量为d

                ///更新弧的容量和可增广的容量
                edge[i].cap -= d;
                edge[i ^ 1].cap += d;
                can -= d;

                if(h[st] >= n) return cost - can;
                if(!can) break;///不能再继续增广
            }
            if(h[v] < minh) minh = h[v];///更新最小标号
        }
    }
    if(can == cost){///如果没有增广...GAP
        gap[h[x]]--;
        if(gap[h[x]] == 0) h[st] = n;///存在断层，没有增广路了
        h[x] = minh + 1;///重新标记，保证下次再访问的时候有流量
        gap[h[x]]++;
    }
    return cost - can;///在这个点之前可以增广的 - 访问这个点之后可以增广的 = 在这个点增广的容量
}
int max_flow(int N){///SAP+GAP优化
    n = N;//m = M;
    for(int i=0; i<cnt; i++) c[i] = edge[i].cap;
    memset(h, 0, sizeof(h));///h[i]表示i节点的标号
    memset(gap, 0, sizeof(gap));///gap[i]表示标号为i的节点个数
    gap[0] = n;///初始有n个节点标号为0
    int ret = 0;
    while(h[st] < n){
        ret += dfs(st, inf);
    }
    return ret;
}
void init(int source, int sink){
    memset(head, -1, sizeof(head)); cnt = 0;
    st = source; ed = sink;
}
///求割集
bool vis[maxn];
void Dfs(int x){
    vis[x] = true;
    for(int i=head[x]; ~i; i=edge[i].next){
        if(edge[i].cap > 0 && !vis[edge[i].v]) Dfs(edge[i].v);
    }
}
vector<int> V[22];
int N, k;
void Find(){
    for(int i=1; i<=k; i++) {
        V[i].clear();
        for(int j=head[i]; ~j; j=edge[j].next)
        if(edge[j].v != st && edge[j].cap == 0)
            V[i].pb(edge[j].v - k);
    }
}
int main(){
    cin >> k >> N;
    int sum = 0;
    init(0, N + k + 1);
    for(int i=1; i<=k; i++){
        int val;
        scanf("%d", &val);
        sum += val;
        addedge(st, i, val);
    }
    for(int i=1; i<=N; i++){
        int num, v;
        addedge(i + k, ed, 1);
        scanf("%d", &num);
        while(num--){
            scanf("%d", &v);
            addedge(v, i + k, 1);
        }
    }
    int ans = max_flow(N + k + 2);
    if(ans != sum) {puts("No Solution!");  return 0;}
    Find();
    for(int i=1; i<=k; i++){
        printf("%d:", i);
        for(unsigned j=0; j<V[i].size(); j++)
            printf(" %d", V[i][j]);
        puts("");
    }
    return 0;
}