状压DP——彩虹小马与魔法数列

最新推荐文章于 2017-05-26 14:46:04 发布

原创最新推荐文章于 2017-05-26 14:46:04 发布 · 519 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

动态规划同时被 2 个专栏收录

41 篇文章

订阅专栏

位运算

10 篇文章

订阅专栏

本文介绍了一道关于魔法数列的问题，通过使用DP算法解决暮光闪闪如何以最小魔力值消耗达到任意两数最大公约数为1的目标。文章详细解析了问题背景、输入输出格式、样例及DP算法实现。

P3728彩虹小马与魔法数列
时间限制 : - MS 空间限制 : 65536 KB
评测说明 : 时限3000ms
问题描述
暮光闪闪想要快快长高，但她学习的有关身高的魔法都会一下子让她长高到100米以上，这显然是她不愿意看到的。在查阅了许多魔法典籍后，她终于发现长高的速度与一个魔法数列a有关。她知道，当魔法数列中的任意两个数最大公约数都为1时，自己的长高速度会变得更快。非常幸运，暮光闪闪拥有修改魔法序列的能力，她可以消耗abs(ai-bi)点魔力值将ai改为bi。可是暮光闪闪还很小，负担不起过多的魔力值消耗，所以她来请求你的帮助，帮她算出她加快自己长高速度所需要的消耗的最小魔力值。

输入格式

第一行，一个整数n表示数列a的长度；
第二行，n个整数，分别描述a1,a2,a3,…,an。

输出格式

一个整数，表示暮光闪闪完成修改的最小魔力值。

样例输入 1

3
2 2 15

样例输出 1

1
样例输入 2

5
1 6 4 2 8

样例输出 2

3
提示
1<=n<=60

1<=ai<=30

分析：

注意倒 1<=n<=60,而1~60中的质数个数只有16个，每一个数包含哪几个质因数可以看成一个集合，于是互质也就对应了集合中的不相交。

将1~60之间的质数按0~16编号，一个数包含的质因数的集合就可以用一个整数表示出来。

于是愉快的开始DP啦！！！

状态：f[i][s]表示前i个数使用的质因数集合为s的最小花费。

决策：第i个数变成什么。

方程：f[i][s]=min{ f[i-1][s^id[k] ]+ abs(k-a[i]) } (id[k]表示k这个数对应的质因数集合，id[k]是s的子集)

因为规定1与任何数互质，所以id[1]=0.

代码如下：

#include<cstdio>
#include<iostream>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<queue>
#include<vector>
#include<algorithm>
#define LL long long
#define CLEAR(xxx) memset(xxx,0,sizeof(xxx))
using namespace std;
const int maxn=17,inf=1e9;

int All,n,m,f[62][(1<<maxn)+5],s[65],id[65];
bool mark[100];
void id_table(){
	for(int i=2;i<=60;i++)
		if(!mark[i]){
			id[i]|=(1<<m);
			for(int j=i+i;j<=60;j+=i)
				mark[j]=true,id[j]|=(1<<m);	
			m++;
		}
}

int main(){
	int i,j,k,x,s0;
	cin>>n;
	for(i=1;i<=n;i++)cin>>s[i];
	id_table();
	m--;
	All=(1<<m)-1;
	for(i=1;i<=n;i++)
		for(j=0;j<=All;j++)f[i][j]=inf;
	for(i=1;i<=n;i++)
		for(k=1;k<(s[i]<<1);k++)
			for(s0=id[k];s0<=All;s0++)
				if((s0&id[k])==id[k])   //id[k]是s0的子集 
				//位运算优先级低，不能写成 s0&id[k] ==id[k] 
					f[i][s0]=min(f[i][s0],f[i-1][s0^id[k]]+abs(s[i]-k));
	int ans=inf;
	for(s0=0;s0<=All;s0++)
		ans=min(ans,f[n][s0]);;
	cout<<ans<<endl;
	return 0;
}