codeforces D. Make The Fence Great Again(线性二维dp)

传送:https://codeforces.com/contest/1221/problem/D

题目:给出一个长度为n的序列,a[1]...a[n],以及各值高度加1的代价,cost[1]...cost[n],要使得序列任意相邻的两个数a[i-1]!=a[i]。

求代价最小是多少?

思路:通过模拟观察可知每个数的增加值最多为2,用二维dp枚举递推。

#include
#define LL long long
#define INF 0x3f3f3f3f3f3f3f3f
using namespace std;
const int maxn=300005;
const int mod=1e9+7;
LL h[maxn],cost[maxn];
LL dp[maxn][3];
int main()
{
	ios::sync_with_stdio(false);
	int q;
	cin>>q;
	while(q--)
	{
		int n;
		cin>>n;
		for(int i=1;i<=n;i++)
		{
			cin>>h[i];
			cin>>cost[i];
			dp[i][0]=dp[i][1]=dp[i][2]=INF;
		}
		dp[1][0]=0;
		dp[1][1]=cost[1];
		dp[1][2]=2*cost[1];
		for(int i=2;i<=n;i++)
		{
			for(int j=0;j<3;j++)
			{
				for(int k=0;k<3;k++)
				{
					if(h[i-1]+j!=h[i]+k)
					{
						dp[i][k]=min(dp[i][k],dp[i-1][j]+k*cost[i]);
					}
				}
			}
		}
		cout << min(dp[n][0],min(dp[n][1],dp[n][2])) << endl;
	} 

	return 0;
}

 

你可能感兴趣的:(acm,dp)