96 lines
2.7 KiB
C++
96 lines
2.7 KiB
C++
#include <bits/stdc++.h>
|
||
|
||
using namespace std;
|
||
// TODO 只过了两个测试点，因为没有测试数据，所以不知道为什么，两个样例都是可以过的
|
||
// 2023-03-20
|
||
// 对拍也不知道怎么做，一会问一下ChatGpt
|
||
|
||
// 明天研究一下对拍：https://www.cnblogs.com/EdisonBa/p/13509379.html
|
||
|
||
const int N = 2e5 + 10, M = N << 1;
|
||
|
||
int n, k; // 猫猫数量，房间总数
|
||
int x[N], y[N]; // i:第几个房间,x[i]:可以住哪只猫猫,y[i]:可以住哪只猫猫
|
||
|
||
// 链式前向星
|
||
int e[M], h[N], idx, w[M], ne[M];
|
||
void add(int a, int b, int c = 0) {
|
||
e[idx] = b, ne[idx] = h[a], w[idx] = c, h[a] = idx++;
|
||
}
|
||
|
||
int st[M]; // 房间是不是使用过
|
||
int flag; // 所有猫猫是不是都有了房间
|
||
int out;
|
||
vector<int> path;
|
||
// 树+dfs搜索
|
||
void dfs(int u, int m) {
|
||
if (u == n + 1) {
|
||
flag = 1;
|
||
if (!out) {
|
||
// 记录最新获得的结果
|
||
path.clear();
|
||
|
||
for (int i = 1; i <= m; i++) {
|
||
if (st[i] == x[i])
|
||
path.push_back(1);
|
||
else
|
||
path.push_back(2);
|
||
}
|
||
out = 1;
|
||
}
|
||
return;
|
||
}
|
||
for (int i = h[u]; ~i; i = ne[i]) {
|
||
int room = w[i]; // 哪个房间
|
||
if (st[room]) continue; // 如果这个房间用人用了，就不能用
|
||
// 记录
|
||
st[room] = u; // 记录这个房间谁在用
|
||
// 递归
|
||
dfs(u + 1, m);
|
||
// 回溯
|
||
st[room] = 0;
|
||
}
|
||
}
|
||
|
||
bool chk(int m) {
|
||
// 清空
|
||
memset(h, -1, sizeof h);
|
||
idx = 0;
|
||
flag = 0;
|
||
out = 0;
|
||
memset(st, 0, sizeof st);
|
||
|
||
for (int i = 1; i <= m; i++) { // 前m个房间
|
||
add(x[i], y[i], i); // 将房间号作为边权，理解为x[i]号小猫，可以用掉这个与它相连的房间，y[i]号小猫也可以用掉这个与它相边的房间
|
||
add(y[i], x[i], i);
|
||
}
|
||
// 检查一下现在可以满足每个小猫都有房间住吗？
|
||
dfs(1, m);
|
||
|
||
return flag;
|
||
}
|
||
|
||
int main() {
|
||
freopen("cats.in", "r", stdin);
|
||
// freopen("cats.out", "w", stdout);
|
||
|
||
memset(h, -1, sizeof h);
|
||
cin >> n >> k;
|
||
for (int i = 1; i <= k; i++) cin >> x[i];
|
||
for (int i = 1; i <= k; i++) cin >> y[i];
|
||
|
||
int L = 1, R = k, ans = L; // 二分答案
|
||
while (L <= R) {
|
||
int mid = (L + R) >> 1;
|
||
if (chk(mid)) // 如果每个猫猫都可以有房间，那么检查通过,说明房间的数量还是很宽松的
|
||
R = mid - 1, ans = mid;
|
||
else
|
||
L = mid + 1;
|
||
}
|
||
// 输出数量
|
||
cout << ans << endl;
|
||
|
||
for (int i = 0; i < path.size(); i++)
|
||
cout << path[i] << " ";
|
||
return 0;
|
||
} |