LINK: http://codeforces.com/contest/454/problem/D
一道状态压缩DP题目,感觉不错的题目。
题意是,给你一个数列a[n],n<=100, 1<=a[i]<=30,让你求b[n], 满足b中两两互质,且sigma(abs(a[i] - b[i])) 最小。
可以发现,b[i]的最大值不会过59,因为a[i]最大为 30 ,如果b[i]>=59,不如令b[i] = 1,因为全1的数列互质。
一道状态压缩DP题目,感觉不错的题目。
题意是,给你一个数列a[n],n<=100, 1<=a[i]<=30,让你求b[n], 满足b中两两互质,且sigma(abs(a[i] - b[i])) 最小。
可以发现,b[i]的最大值不会过59,因为a[i]最大为 30 ,如果b[i]>=59,不如令b[i] = 1,因为全1的数列互质。
这样就可以DP了,60内的质数不过16个(状态压缩), dp[i][j], i表示处理到第i个数,j表示此时前i个数用到的质因数, 它的值表示sigma值。
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<string>
#include<vector>
#include<cmath>
#include<queue>
#include<map>
#include<set>
using namespace std;
#define INF 1000000000
//typedef __int64 LL;
#define N 105
int dp[N][1<<16], n, num[N], dat[N][1<<16];
int prime[50] = {2,3,5,7,11,13,17,19,23,29,31,37,41,43,47,53};//size 16
int c[66],bit[66], out[N];
int main()
{
#ifndef ONLINE_JUDGE
freopen("in.txt", "r", stdin);
#endif // ONLINE_JUDGE
scanf("%d", &n);
for(int i=1; i<=n; i++)
{
scanf("%d", &num[i]);
}
c[0] = 1;
for(int i=1; i<=16; i++) c[i] = c[i-1] <<1;
for(int i=1; i<=60; i++)
{
int tmp = 0;
for(int j=0; j<16; j++)
{
if(i%prime[j] == 0) tmp |= c[j];
}
bit[i] = tmp;
}
memset(dp, -1, sizeof(dp));
dp[0][0]= 0;
for(int i=0; i<n; i++)
{
for(int j=0; j<(1<<16); j++)
{
if(dp[i][j] != -1)
{
for(int k=1; k<59; k++)
{
int tmp = bit[k];
if(tmp & j) continue;
int next = tmp | j;
int ss = abs(k - num[i+1]);
if(dp[i][j] + ss<dp[i+1][next] || dp[i+1][next]==-1) dp[i+1][next] =dp[i][j] + ss, dat[i+1][next] = k;
}
}
}
}
int id = -1, mi=INF;
for(int i=0; i<(1<<16); i++)
{
if(dp[n][i]!=-1 && dp[n][i]<mi)
{
mi = dp[n][i]; id = i;
}
}
out[n] = dat[n][id];
for(int i=n;i>=1; i--)
{
out[i] = dat[i][id];
id = id - bit[out[i]];
}
for(int i=1; i<=n; i++) printf("%d ", out[i]);
puts("");
return 0;
}