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;
}
}
还有一个困难,题面太长了,告辞。