题意:给定一个2*n的矩阵,每个元素都是[1,n]的整数,现要选取某些列,使这些列的上下两行交换,得到的矩阵每行都是一个排列,问最少要选多少列。
分析:
- 对于每列,只有选和不选两种状态。
- 如果两个相同的数位于同一行,那么所在的两列的状态必须相异(一个选 一个不选)
- 如果位于不同行,那么所在的两列的状态必须相同(同时选或不选)
- 也就是说,列的状态是具有传递性的,可以用(带权)并查集来维护传递性关系,当然,这题用虚点并查集的写法更简洁 , i表示选取,代价为1, i+n表示不选取,代价为0.
(此处的传递性是无向的,也就是两列可以互相传递,所以可以用并查集,如果是有向的,那需要用求强连通分量的算法比如tarjan)
实现:
①:每个数有且只能出现2次,如果不是2次,那么肯定输出-1
②:枚举每个数i,(所在列记为col1[i]
,col2[i]
),
所在的两个位置如果同行,
就把col1[i]
,col2[i+n]
合并,col2[i]
,col1[i+n]
合并。
如果不同行,就把col1[i]
,col2[i]
合并,col2[i+n]
,col1[i+n]
合并。
③:合并完后,检查是否是i i+n在同一个并查集当中的情况,如果有,那么肯定输出-1
④:枚举每个连通块,选取i 、i+n所在并查集当中代价更小的那个。
#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pii pair<int,int>
const int maxn = 4e2+10;
const int mx = 40;
const int mod = 1e9+7;
const ll inf = 34359738370;
const int INF = 1e5;
//给定一个2*n的矩阵 由1~n数字构成 问最少选定多少列 行交换 才能使得两行都成为一个排列
//2-sat问题的并查集做法
//对于2个数 如果在同一行 那么必须不同时换 在不同行 必须同时换或不换
//开两个并查集 i 换 i+n 不换
//对于列i 有i所在的并查集 i+1所在并查集 cost[find(i)] 表示i列交换的情况下的代价
//如果i代价小于i+n 则选i
int loca1[maxn][2],loca2[maxn][2];//第一次出现的位置
int fa[maxn];//并查集
int g[3][maxn];
int cost[maxn];//代价
int n;
int cnt[maxn];//出现次数 有不等于2的直接-1
int flag[maxn];//判断i选了哪个
vector<int> ans;
void init()
{
ans.clear();
for(int i=1;i<=2*n