CF 1492E(暴搜)
Contents
题目链接
https://codeforces.com/contest/1492/problem/E
题意
给定 $n$ 个长度为 $m$ 的正整数array,其中 $n \geq 2, m \geq 1, n \times m \leq 250000$
问是否存在一个array,使得这个array 与 其他每个array的difference(不同元素的个数) $\leq 2$?
题解
暴搜,我们可以先令 $ans$ 等于第一个array,然后看一下其他array中,有没有 $diff > 2$ 的,如果有,尝试更改 $ans$ 中的一个元素,更改后再看一下其他array的difference情况,如果还是不行,就继续尝试更改其他元素。 直到所有array的 $diff \leq 2$
注意到,因为$ans$基于第一个array,所以最多只能更改两个元素,我们可以给在搜索的时候设置一个深度 left
,代表还可以更改几个元素。一开始就是dfs(2)
。
注1: 本题时限卡的比较紧,时限为2s,我们可以设定2s内如果搜索不出结果就直接返回"No”。
double start = clock(); double passed = clock() - start; if (passed > 1950.0) { // 1950ms cout << "No" << "\n" exit(0); }
注2: 暴搜的时候记得回溯!
代码
#include <bits/stdc++.h>
using namespace std;
int n,m;
vector< vector<int> > arr;
vector<int> ans;
vector<int> cnt;
double start;
inline bool check() {
for (int i = 0; i < n; i++) {
if (cnt[i] > 2) return 0;
}
return 1;
}
void dfs(int left) {
if (left < 0) return;
if (clock()-start > 1950) {
printf("No\n");
exit(0);
}
for (int i = 0; i < n; i++) {
cnt[i] = 0;
vector<int> pos;
for (int j = 0; j < m; j++) {
if (ans[j] != arr[i][j]) {
cnt[i] += 1;
pos.push_back(j);
}
}
if (cnt[i]-left > 2) {
return;
}
if (cnt[i] > 2) {
for (int j : pos) {
int pre = ans[j];
ans[j] = arr[i][j];
dfs(left-1);
ans[j] = pre; //记得回溯
}
}
}
if (check()) {
printf("Yes\n");
for (int j = 0; j < m; j++) printf("%d ", ans[j]);
printf("\n");
exit(0);
}
}
int main() {
scanf("%d%d",&n,&m);
start = clock();
arr = vector<vector<int> > (n, vector<int>(m,0));
for (int i = 0; i < n; i++) {
for (int j = 0; j < m; j++) {
scanf("%d", &arr[i][j]);
}
}
ans = arr[0];
cnt = vector<int>(n,0);
dfs(2);
printf("No\n");
}