(状态机dp)大盗阿福


题目描述:

阿福是一名经验丰富的大盗。趁着月黑风高,阿福打算今晚洗劫一条街上的店铺。
这条街上一共有 N 家店铺,每家店中都有一些现金。
阿福事先调查得知,只有当他同时洗劫了两家相邻的店铺时,街上的报警系统才会启动,然后警察就会蜂拥而至。
作为一向谨慎作案的大盗,阿福不愿意冒着被警察追捕的风险行窃。
他想知道,在不惊动警察的情况下,他今晚最多可以得到多少现金?
输入格式:
输入的第一行是一个整数 T,表示一共有 T 组数据。
接下来的每组数据,第一行是一个整数 N ,表示一共有 N 家店铺。
第二行是 N 个被空格分开的正整数,表示每一家店铺中的现金数量。
每家店铺中的现金数量均不超过1000。
输出格式:
对于每组数据,输出一行。
该行包含一个整数,表示阿福在不惊动警察的情况下可以得到的现金数量。
数据范围
1≤T≤50,
1≤N≤1e5
输入样例

2
3
1 8 2
4
10 7 6 14

输出样例

8
24

样例解释
对于第一组样例,阿福选择第2家店铺行窃,获得的现金数量为8。
对于第二组样例,阿福选择第1和4家店铺行窃,获得的现金数量为10+14=24。

思路:

  这道题可能是我除了背包问题外,写得最顺的dp了。阿福对于一个店铺有两种选择:1-盗窃,0-不盗窃,
那么我们的dp数组就这样定义了:dp[n][2](对于dp[i][1]表示前i个店铺,并盗窃第i个店铺的最大收益)。
既然这样,对于dp[i][0] 和 dp[i][1]的最优解怎么求呢?
  让我们先来想一下对于dp[i][0]这个状态该怎么求出来,当第i个店铺不被盗窃的时
候,那我们就能盗窃第i-1个店铺,也可以不盗窃第i - 1个店铺,所以dp[i][0] = max(dp[i - 1][1], dp[i - 1][0]),
那么为什么前面的第i-2,i-3,...1个店铺的状态不用考虑呢,因为dp[i-1][1]和dp[i-1][0]已经是前面状态的最优解了.
  对于dp[i][1]这个状态也是一样的思考方式,就能顺理成章地得出另一个状态转移方程:dp[i][1] = dp[i - 1][0] + w[i];
  最后让我们愉快地AC吧!

Code

#include
using namespace std;
#define endl '\n'
typedef long long ll;
typedef pair PII;

const int mod = 1e9 + 7;
const int inf = 0x3f3f3f3f;
const int N = 1e5 + 10;
int n, m, w[N];
int dp[N][2];

void solve(){
	cin >> n;
	for(int i = 1; i <= n; i++)
		cin >> w[i];
	dp[1][0] = 0; dp[1][1] = w[1];
	for(int i = 2; i <= N; i ++){
		dp[i][0] = max(dp[i - 1][1], dp[i - 1][0]);
		dp[i][1] = dp[i - 1][0] + w[i];
	}
	cout << max(dp[n][1], dp[n][0]) << endl;
}
int main(void){
	ios::sync_with_stdio(false);
	cin.tie(0);
    int _; cin >> _;
    while(_--) solve();
	return 0;
}