题解 : 首先发现这个题目中的 b 数组中每一个元素都不会超过 58 为什么呢 ? 自己想去 ….
然后我们发现,两个数互质就是没有相同的素因子,然后 1 -57 只有 18 个素数,先将这 18 个数打进一个表然后,用
state[i] 存下每个数包含的所有的素因子。
剩下的就是转移了
dp[i][j] 表示前 i 个 用了 j 这些素数 (j 是一个二维的压缩状态)。
dp[i + 1][j | state[k]] = min (dp[i][j] + abs (k - a[i]),dp[i + 1][j | k]) ;
由于这个题目要求保留路径,那么就还需要开一个数组去储存路径就好了
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
using namespace std;
const int INF = 1e9 + 7;
const int maxn = (1 << 18);
int prime[] = {2,3,5,7,11,13,17,19,23,29,31,37,39,41,43,47,53,57};
int a[105] = {0};
int b[105] = {0};
int state[60] = {0};
int dp[105][maxn] = {0};
int pos[105][maxn] = {0};
int main() {
ios_base :: sync_with_stdio(false);
int n;
cin >> n;
for (int i = 1;i <= n; ++ i) cin >> a[i];
for (int i = 2;i < 60; ++ i) {
for (int j = 0;j < 18; ++ j) {
if (i % prime[j] == 0) {
state[i] |= (1 << j);
}
}
}
for (int i = 0;i < 105; ++ i) {
for (int j = 0; j < maxn; ++ j) dp[i][j] = INF;
}
dp[0][0] = 0;
for (int i = 0;i < n; ++ i) {
for (int j = 0;j < maxn; ++ j) {
if (dp[i][j] == INF) continue;
if (dp[i + 1][j] > dp[i][j] + a[i + 1] - 1) {
dp[i + 1][j] = dp[i][j] + a[i + 1] - 1;
pos[i + 1][j] = 1;
}
for (int k = 2;k < 60; ++ k) {
if (state[k] & j) continue;
else {
int u = abs (a[i + 1] - k);
if (u + dp[i][j] < dp[i + 1][j | state[k]]) {
dp[i + 1][j | state[k]] = dp[i][j] + u;
pos[i + 1][j | state[k]] = k;
}
}
}
}
}
int ans = INF;
int tem = 0;
for (int i = 0;i < maxn; ++ i) {
if (dp[n][i] < ans) {
ans = dp[n][i];
tem = i;
}
}
int cnt = 0;
for (int i = n;i > 0;-- i) {
b[cnt ++] = pos[i][tem];
tem -= state[pos[i][tem]];
}
for (int i = cnt - 1;i >= 0 ; -- i) {
cout << b[i] << ' ';
}
return 0;
}