1. 程式人生 > 其它 >BZOJ#1059 [ZJOI2007]矩陣遊戲

BZOJ#1059 [ZJOI2007]矩陣遊戲

[ZJOI2007]矩陣遊戲

思路:

假設只考慮變換列,不會影響每一行得相對位置,那麼要使得每一行都有一個1,就相當於每一個x有一個匹配的y,可以使用匈牙利演算法

程式碼:

#include <bits/stdc++.h>
#define int long long
int _ = 0, Case = 1;
using namespace std;
#define all(v) begin(v),end(v)
#define nline '\n'

const int N = 220;
vector<int> h[N];
int a[N][N];
int match[N], st[N];
bool find(int u) {
    for (auto i : h[u]) {
        if (!st[i]) {
            st[i] = true;
            if (!match[i] or find(match[i])) {
                match[i] = u;
                return true;
            }
        }
    }
    return false;
}
void solve(int Case) {
    int n;
    cin >> n;
    for(int i=1;i<=n;i++) match[i]=0,h[i].clear();
    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= n; j++) {
            cin >> a[i][j];
            if (a[i][j]) {
                h[i].push_back(j);
            }
        }
    }
    for (int i = 1; i <= n; i++) {
        for (int j = 1; j <= n; j++) st[j] = false;
        if (!find(i)) {
            cout << "No" << nline;
            return;
        }
    }
    cout << "Yes" << nline;
}

signed main() {
    ios::sync_with_stdio(false); cin.tie(nullptr);
    cin >> _; for (Case = 1; Case <= _; Case++)
        solve(Case);

    return 0;
}