题解 | 打家劫舍(二)

#include <iostream>
#include <vector>
using namespace std;
int rob_linear(vector<int>& nums);
int main() {
    int n;
    cin>>n;
    vector<int> arr(n,0);
    for(int i=0;i<n;i++){
        cin>>arr[i];
    }
    if(n==0) return 0;
    vector<int> temp1=vector<int>(arr.begin(),arr.end()-1);
    vector<int> temp2=vector<int>(arr.begin()+1,arr.end());
    int res1=rob_linear(temp1);
    int res2=rob_linear(temp2);
    cout<<(res1>res2?res1:res2);
    return 0;
    // int res=max(,)
}
int rob_linear(vector<int>& nums){
    int n=nums.size();
    vector<int> dp(n+1,0);
    if(n==0) return 0;
    if(n==1) return nums[0];
    else{
        dp[0]=0;
        dp[1]=nums[0];
        for(int i=2;i<n+1;i++){
            dp[i]=max(dp[i-1],dp[i-2]+nums[i-1]);
        }
    }
    return dp[n];
}
// 64 位输出请用 printf("%lld")