Codeforces Round #259 (Div. 2) D. Little Pony and Harmony Chest

最新推荐文章于 2022-11-22 23:15:14 发布

ACM_Napoleon

最新推荐文章于 2022-11-22 23:15:14 发布

阅读量727

点赞数

分类专栏： DP

本文链接：https://blog.csdn.net/napoleon_acm/article/details/38351575

版权

DP 专栏收录该内容

39 篇文章 0 订阅

订阅专栏

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了，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;
}