D. Little Pony and Harmony Chest
time limit per test
4 seconds
memory limit per test
256 megabytes
input
standard input
output
standard output
Princess Twilight went to Celestia and Luna's old castle to research the chest from the Elements of Harmony.
A sequence of positive integers bi is harmony if and only if for every two elements of the sequence their greatest common divisor equals 1. According to an ancient book, the key of the chest is a harmony sequence bi which minimizes the following expression:
You are given sequence ai, help Princess Twilight to find the key.
Input
The first line contains an integer n (1?≤?n?≤?100) ? the number of elements of the sequences a and b. The next line contains n integersa1,?a2,?...,?an (1?≤?ai?≤?30).
立即学习“前端免费学习笔记(深入)”;
Output
Output the key ? sequence bi that minimizes the sum described above. If there are multiple optimal sequences, you can output any of them.
Sample test(s)
input
51 1 1 1 1
output
1 1 1 1 1
input
51 6 4 2 8
output
1 5 3 1 8
给出N个数ai,求出另一个序列bi,要求sum |ai-bi|,最短,且所有的bi都互质。
这里题目给了几个很显眼的条件,ai限制在了1~30之间,由于可以bi无限选1这个数,那么|ai-bi| 最大就是29了,意味着bi
要求所有的bi互质,可以化为所有的bi分解出来的质因数均不相同,bi
用s表示当前阶段用了哪些质因数的状态,例如 s = 3 = 11 代表目前状态下使用了第一个和第二个质因数。
很快我们就可以写出状态转移方程:
f[i][s] = min(f[i-1][s^c[k]] + abs(a[i] - k))。 其中c[k]表示数字k使用了哪些质因数。
#include <cstdio>#include <cmath>#include <cstring>#define M_max 60#define N_max 123#define inf 0x3f3f3f3fusing namespace std;int p[N_max], c[M_max], a[N_max];int f[N_max][1<<16], pre[N_max][1<<16][2];int n, cnt, minnum, minpos;void prime() { for (int i = 2; i <= M_max; i++) { bool flag = false; for (int j = 2; j <= sqrt(i); j++) if (i%j == 0) { flag = true; break; } if (!flag) p[++cnt] = i; } for (int i = 1; i <= M_max; i++) for (int j = 1; j <= cnt; j++) if (i%p[j] == 0) c[i] |= 1 << (j-1);}void init() { prime(); scanf("%d", &n); for (int i = 1; i <= n; i++) scanf("%d", &a[i]);}void print(int x, int pos) { if (x == 0) return; print(x-1, pre[x][pos][0]); printf("%d ", pre[x][pos][1]);}void solve() { memset(f, inf, sizeof(f)); memset(f[0], 0, sizeof(f[0])); minnum = inf; for (int i = 1; i <= n; i++) for (int s = 0; s < (1<<16); s++) for (int k = 1; k <= M_max; k++) if ((s&c[k]) == c[k]) { int tmp = f[i-1][s^c[k]] + abs(a[i]-k); if (tmp < f[i][s]) { f[i][s] = tmp; pre[i][s][0] = s^c[k]; pre[i][s][1] = k; } } for (int s = 0; s < (1<<16); s++) if (f[n][s] < minnum) { minnum = f[n][s]; minpos = s; } print(n, minpos);}int main() { init(); solve();}
HTML怎么学习?HTML怎么入门?HTML在哪学?HTML怎么学才快?不用担心,这里为大家提供了HTML速学教程(入门课程),有需要的小伙伴保存下载就能学习啦!
Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号