Codeforces 454D Little Pony and Harmony Chest 状压DP

点击打开链接

题意:n个数a[i],找到一组序列B,B中任意两个元素互质,最小值为?  n<=100,a[i]<=30

一定有解 并且b[i]不超过2*a[i]-1 若超过,b[i]用1来代替显然更优
b[i]范围被限制在60内,又要满足两两互质,用二进制记录前i个数中素因子出现的状态
设dp[i][s] 前i个数 素因子出现状态为s 则新添加的x要满足num[x]&s==0 否则有大于1的公约数  num[i]为i包含素因子的二进制状态
pre保存当前决策  复杂度为O(n*2^16*60)

#include <bits/stdc++.h>using namespace std;typedef long long ll;const int N=2e2+20;const int M=1<<17;const int inf=0x3f;int a[N],b[N],n,pri[N],pn,num[N];//num[i] i°üº¬ËØÒò×ӵĶþ½øÖÆ״̬ int pre[N][M],dp[N][M];void init(){memset(num,0,sizeof(num));pn=0;int vis[110]={0};for(int i=2;i<60;i++){if(vis[i]==0){pri[pn++]=i;for(int j=i+i;j<60;j+=i)vis[j]=1;}}for(int i=2;i<60;i++){for(int j=0;j<pn;j++){if(i%pri[j]==0)num[i]|=1<<j;}}}int main(){init();while(cin>>n){for(int i=1;i<=n;i++)scanf("%d",&a[i]);memset(dp,0x3f,sizeof(dp));dp[0][0]=0;for(int i=1;i<=n;i++){for(int s=0;s<M;s++){for(int x=1;x<60;x++){//Ç°Ò»¸ö״̬ҪºÏ·¨ if(dp[i-1][s]==inf||(num[x]&s))continue;int res=dp[i-1][s]+abs(x-a[i]);if(dp[i][s|num[x]]>res){dp[i][s|num[x]]=res;pre[i][s|num[x]]=x;}}}}int pos,ans=2e8;for(int s=0;s<M;s++){if(ans>dp[n][s])ans=dp[n][s],pos=s;}for(int i=n;i>=1;i--){b[i]=pre[i][pos];pos=pos^num[b[i]];}for(int i=1;i<=n;i++)printf("%d%c",b[i],i==n?'\n':' ');}return 0;}