题目链接

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");
}