题目描述
阿福是一名经验丰富的大盗。趁着月黑风高,阿福打算今晚洗劫一条街上的店铺。
这条街上一共有 N 家店铺,每家店中都有一些现金。阿福事先调查得知,只有当他同时洗劫了两家相邻的店铺时,街上的报警系统才会启动,然后警察就会蜂拥而至。
作为一向谨慎作案的大盗,阿福不愿意冒着被警察追捕的风险行窃。他想知道,在不惊动警察的情况下,他今晚最多可以得到多少现金?
输入
输入的第一行是一个整数 T (T <= 50) ,表示一共有 T 组数据。 接下来的每组数据,第一行是一个整数 N (1 <= N <= 100, 000) ,表示一共有 N 家店铺。第二行是 N 个被空格分开的正整数,表示每一家店铺中的现金数量。每家店铺中的现金数量均不超过 1000 。
输出
对于每组数据,输出一行。该行包含一个整数,表示阿福在不惊动警察的情况下可以得到的现金数量。
样例输入
2 3 1 8 2 4 10 7 6 14
样例输出
8 24
提示
对于第一组样例,阿福选择第 2 家店铺行窃,获得的现金数量为 8 。 对于第二组样例,阿福选择第 1 和 4 家店铺行窃,获得的现金数量为 10 + 14 = 24 。
分析
对于每一家店,有两种状态,一是不偷,二是不偷前面那家,偷当前店铺(相当于偷)
方法1:
思路:不连续取数(一维DP)
代码如下:
#include<bits/stdc++.h>
using namespace std;
//dp[i]:抢a[i]结尾的店铺可以得到的最大现金
long long n,a[100001],dp[100001];
int main(){
long long t;
cin>>t;
while(t--){
cin>>n;
for(int i=1;i<=n;i++){
cin>>a[i];
dp[i]=a[i];//自己一定要抢
}
dp[3]=dp[1]+dp[3];//dp[3]就是抢1+3
long long mt=max(dp[1],dp[2]);//最大值
for(int i=4;i<=n;i++){
dp[i]=dp[i]+mt;
mt=max(mt,dp[i-1]);//更新第1~i-1的dp最大值
}
long long ok=0;
for(int i=1;i<=n;i++){
ok=max(ok,dp[i]);
}
cout<<ok<<"\n";
}
return 0;
}
方法2:
思路:二维DP(动态规划)
dp[i][0]:表示不偷第i家可以获得的最大值 0表示不偷
dp[i][1]:表示偷第i家,不偷第i-1家可以获得的最大值 1表示偷
状态转移方程:
dp[i][0]=max(dp[i-1][0],dp[i-1][1]); 不偷当前店铺,最大值为前一家两种状态中的最大值
dp[i][1]=dp[i-1][0]+a[i]; 偷当前店铺,就是不偷上一家店铺的值加上偷这家的值
代码如下:
#include<bits/stdc++.h>
using namespace std;
long long n,a[100001],dp[100001][1];
int main(){
long long t;
scanf("%lld",&t);//与cin>>t;差不多,不过scanf输入的值可以更大
while(t--){
scanf("%lld",&n);
for(int i=1;i<=n;i++){
scanf("%lld",&a[i]);
}
dp[1][0]=0;
dp[1][1]=a[1];
for(int i=2;i<=n;i++){
dp[i][0]=max(dp[i-1][0],dp[i-1][1]);
dp[i][1]=dp[i-1][0]+a[i];
}
printf("%lld\n",max(dp[n][0],dp[n][1]));
}
return 0;
}