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

mac2022-08-06  13

传送: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<bits/stdc++.h> #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; }

 

最新回复(0)