招商银行-FinTech精英训练营初赛-编程题

第一题是经典dp拼凑钱币的改版,看这里

第二题

题目化简为,一个数字环,才不能选择相邻的数字,问能选择的数字之和最大为多少

类似于leetcode上的股票买卖问题,用一个二维数组表示第i个数字选或者不选的最大和

因为是环,所以要分情况讨论第一个数字选不选的情况

#include <iostream>
#include <cstdio>
#include <cstring>
using namespace std;
int main(){
    int t;
    int n,num[1111];
    int dp[1111][3];
    cin>>t;
    while(t--){
        cin>>n;
        for(int i=0;i<n;i++){
            cin>>num[i];
        }
        if(n==2){
            cout<<max(num[0],num[1])<<endl;
        }
        else{
            int ans=0;
            memset(dp,0,sizeof(dp));
            dp[1][0]=0;
            dp[1][1]=num[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]+num[i];
            }

            ans = max(ans,dp[n-1][0]);
            ans = max(ans,dp[n-1][1]);

            memset(dp,0,sizeof(dp));
            dp[2][0]=num[0];
            dp[2][1]=num[0]+num[2];
            for(int i=3;i<n-1;i++){
                dp[i][0]=max(dp[i-1][0],dp[i-1][1]);
                dp[i][1]=dp[i-1][0]+num[i];
            }

            ans = max(ans,dp[n-2][0]);
            ans = max(ans,dp[n-2][1]);
            cout<<ans<<endl;
        }

    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/q295657451/article/details/80044037