UVA - 1459 Flowers Placement(二分图＋dfs)

题目大意：给出M行N列的花的摆放形式，要求每行每列的任意的两种花的编号不同。现在问，如果新添加一行，如何添加，输入第k个字典序的摆放

解题思路：找出每列可以放的花的编号，连边，接着跑一下二分图匹配，看能否完美匹配，如果不能的话，表示无法摆放
跑这个二分图匹配不是白跑的，后面用得上
因为要第k个字典序，所以要枚举每一列所能摆放的花。
这里需要一个剪枝，如果当前枚举的列为第j列，往后的j+1到第N列如
果无法进行匹配的话，那么就可以剪掉了

#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;
#define N 210

int n, m, k;
int link[N], ans[N];
bool map[N][N], vis[N], used[N];
vector<int> G[N];

void init() {
    scanf("%d%d%d", &n, &m, &k);
    for (int i = 1; i <= n; i++)
        G[i].clear();
    memset(map, 0, sizeof(map));

    int t;
    for (int i = 1; i <= m; i++)
        for (int j = 1; j <= n; j++) {
            scanf("%d", &t);
            map[j][t] = true;
        }

    for (int i = 1; i <= n; i++)
        for (int j = 1; j <= n; j++)  
            if (!map[i][j]) G[i].push_back(j);
}

bool dfs(int cur, int pos) {
    //因为前面的列的花已经订好了，不可再修改
    if (cur <= pos) return false;
    for (int i = 0; i < G[cur].size(); i++) {
        int v = G[cur][i];
        if (!vis[v]) {
            vis[v] = true;
            if (link[v] == -1 || dfs(link[v], pos)) {
                link[v] = cur;
                return true;
            }
        }
    }
    return false;
}

int hungary() {
    memset(link, -1, sizeof(link));
    int cnt = 0;
    for (int i = 1; i <= n; i++) {
        memset(vis, 0, sizeof(vis));
        if (dfs(i,-1))
            cnt++;
    }
    return cnt;
}

int tot;
int tmp[N];

bool ok(int pos, int flower) {
    //如果flower刚好在第pos列
    if (link[flower] == pos) return true;
    int t = 0;
    //找出pos列上所放的花，并记录一下上一个完美匹配
    for (int i = 1; i <= n; i++) {
        tmp[i] = link[i];
        vis[i] = false;
        if (link[i] == pos) t = i;
    }
    //因为要将flower放在第pos列，所以以前放的那列就没花了,第noflower列没有放花
    int noFlower = link[flower];
    link[flower] = pos;
    //第t朵花本来是放在pos列的，因为pos列放了flower，所以第t朵花没地方放了
    link[t] = -1;

    if (dfs(noFlower,pos)) return true;
    else {
        for (int i = 1; i <= n; i++)
            link[i] = tmp[i];
        return false;
    }
}

bool find(int pos) {
    if (pos == n + 1) {
        if (++tot == k) return true;
        return false;
    }

    for (int i = 0; i < G[pos].size(); i++) {
        int v = G[pos][i];
        if (!used[v] && ok(pos, v)) {
            used[v] = true;
            ans[pos] = v;
            if (find(pos + 1)) return true;
            used[v] = false;
        }
    }
    return false;
}

int cas = 1;
void solve() {
    printf("Case #%d: ", cas++);
    if (hungary() != n) {
        printf("-1\n");
        return ;
    }

    memset(used, 0, sizeof(used));
    tot = 0;
    if (find(1)) {
        printf("%d", ans[1]);
        for (int i = 2; i <= n; i++)
            printf(" %d", ans[i]);
        printf("\n");
    }
    else printf("-1\n");
}

int main() {
    int test;
    scanf("%d", &test);
    while (test--) {
        init();
        solve();
    }
    return 0;
}