我们考虑什么样的棋盘是符合题意的,我们会发现同一行和同一列如果有多个黑格子,那么他们之中最多有一个能成为最后的对角线
进一步发现,我们如果能在棋盘上找到n个黑格子,使得他们的行号,列号两两不同,那么就能拼出对角线,这个条件等价于每个行和列都被一枚棋子匹配了一次
于是想到我们对每个行建一个点,对每一个列建一个点,形成一个二分图,对于在a[i][j]的黑格子,就在行的i和列的j之间连一条边,最后跑一个二分图最大匹配,如果是n说明有解
这个思路还是挺棒的
#include <cstdio>
#include <iostream>
#include <cstring>
#include <string>
#include <cstdlib>
#include <utility>
#include <cctype>
#include <algorithm>
#include <bitset>
#include <set>
#include <map>
#include <vector>
#include <queue>
#include <deque>
#include <stack>
#include <cmath>
#define LL long long
#define LB long double
#define x first
#define y second
#define Pair pair<int,LL>
#define pb push_back
#define pf p