牛客-华为机试题库-困难 两道题

HJ28 素数伴侣 - 二分图匹配（匈牙利）

链接：HJ28 素数伴侣
题目并没有说数据没有重复，但是根据提交情况来看，数据确实没有重复。
经典算法，贴个板子。
vis 很关键。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
#include <cmath>
#include <map>
using namespace std;
const int MAXN = 105;
int n;
int a[MAXN];
int to[30005];
bool vis[MAXN];
bool judge(int x) {
    if(x < 2) return false;
    int z = sqrt(x);
    for(int i = 2; i <= z; i++) {
        if(x%i == 0) return false;
    }
    return true;
}
bool match(int x) {
    for(int i = 0; i < n; i++) {
        if(!vis[i] && ((a[i]+x)&1) && judge(x+a[i])) {
            //  cout << x << " " << a[i] << endl;
            // cout << mp.count(make_pair(i,a[i])) << endl;
            vis[i] = true;
            if(to[a[i]] == 0 || match(to[a[i]])) {
                to[a[i]] = x;
                // cout << "yes" << endl;
                return true;
            }
        }
    }
    return false;
}
int main() {
    cin >> n;
    for(int i = 0; i < n; i++) {
        cin >> a[i];
    }
    int ans = 0;
    for(int i = 0; i < n; i++) {
        memset(vis,0,sizeof(vis));
        if(a[i]&1) { // 一边集合为奇数
            if(match(a[i])) {
                ans++;
            }
        }
    }
    cout << ans << endl;
    
}
// 64 位输出请用 printf("%lld")

HJ44 Sudoku

链接：HJ44 Sudoku

数独游戏

暴力 dfs
直接从第一个格子开始遍历所有的格子，
遇到0，枚举1-9能不能放在这里，能放，继续，回溯
全部都放完，打个标记， 返回。

#include <iostream>
#include <cstring>
#include <vector>
using namespace std;
int n = 9;
int a[10][10];
bool flag = false;
bool check(int i, int j, int val) {
    // 行
    for (int k = 0; k < n; k++) {
        if(a[i][k] == val) return false;
    }
    // 列
    for (int k = 0; k < n; k++) {
        if(a[k][j] == val) return false;
    }
    // 小方块
    for(int l = i/3*3; l < i/3*3+3; l++) {
        for(int r = j/3*3; r < j/3*3+3; r++) {
            if(a[l][r] == val) return false;
            // cout << l << " " << r << " l,r" << endl;
        }
    }
    return true;
}

void dfs(int i, int j) {
    if(j == 9) {
        i++;
        j = 0;
    }
    if(i == 9 && j == 0) {
        flag = true;
        return;
    }
    if (a[i][j] == 0) {
        for(int k = 1; k <= 9; k++) {
            if(check(i, j, k)) {
                a[i][j] = k;
                dfs(i, j+1);
                if(flag) return;
                a[i][j] = 0;
            }
        }
    }
    else {
        dfs(i, j+1);
    }
}
int main() {
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++) {
            cin >> a[i][j];
        }
    }
    dfs(0, 0);
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < n; j++) {
            cout << a[i][j];
            if(j < n-1) cout << " ";
        }
        cout << endl;
    }   
}

还有一个困难，题面太长了，告辞。