Codeforces 358D. Dima and Hares (dp)
题目链接:
http://codeforces.com/problemset/problem/358/D
题意:
给出n个数,每个数取走的贡献与相邻的数有关,如果取这个数的时候,左右的数都还没被取,那么权值为a,如果左右两个数有一个被取走了,那么权值为b,如果左右两个数都被取走了,那么权值为c,求取取走全部数的最大值。
思路:
from: http://blog.csdn.net/qq_24451605/article/details/48896657
定义状态:
- dp[i][1][0]表示选第i时,相邻的数有一个被取走,这个数在当前数左侧。
- dp[i][1][1]表示选第i时,相邻的数有一个被取走,这个数在当前数右侧。
- dp[i][2][0]表示选第i时,相邻的数都被取走了
- dp[i][0][0]表示选第i时,相邻的数都没被取走
代码:
#include<bits/stdc++.h> using namespace std; typedef long long ll; #define MS(a) memset(a,0,sizeof(a)) #define MP make_pair #define PB push_back const int INF = 0x3f3f3f3f; const ll INFLL = 0x3f3f3f3f3f3f3f3fLL; inline ll read(){ ll x=0,f=1;char ch=getchar(); while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();} while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();} return x*f; } ////////////////////////////////////////////////////////////////////////// const int maxn = 3e3+10; int n,a[maxn],b[maxn],c[maxn],dp[maxn][3][2]; int main(){ cin >> n; for(int i=1; i<=n; i++) cin >> a[i]; for(int i=1; i<=n; i++) cin >> b[i]; for(int i=1; i<=n; i++) cin >> c[i]; memset(dp,-INF,sizeof(dp)); dp[1][0][0] = a[1]; dp[1][1][1] = b[1]; dp[1][1][0] = dp[1][2][0] = -INF; for(int i=2; i<=n; i++){ dp[i][0][0] = max(dp[i-1][1][1],dp[i-1][2][0])+a[i]; dp[i][1][0] = max(dp[i-1][1][0],dp[i-1][0][0])+b[i]; dp[i][1][1] = max(dp[i-1][1][1],dp[i-1][2][0])+b[i]; dp[i][2][0] = max(dp[i-1][1][0],dp[i-1][0][0])+c[i]; } cout << max(dp[n][0][0],dp[n][1][0]) << endl; return 0; }