leetcode算法题--可被三整除的最大和

原题链接：https://leetcode-cn.com/problems/greatest-sum-divisible-by-three/

动态规划

dp[0]存储余数为0的最大和
dp[1]存储余数为1的最大和
dp[2]存储余数为2的最大和

状态转移：

dp[0]+=nums[i];dp[1]+=nums[i];dp[2]+=nums[i];													if num[i]%3 == 0
dp[0]=max(dp[0],dp[2]+nums[i]);dp[1]=max(dp[1],dp[0]+nums[i]);dp[2]=max(dp[2],dp[1]+nums[i]);	if nums[i]%3 == 1
dp[0]=max(dp[0],dp[1]+nums[i]);dp[1]=max(dp[1],dp[2]+nums[i]);dp[2]=max(dp[2],dp[0]+nums[i]); 	if nums[i]%3 == 2

代码：

int maxSumDivThree(vector<int>& nums) {
    int n=nums.size();
    vector<int> dp(3,INT_MIN);
    dp[0]=0;//dp[0]初始化为0,dp[1]dp[2]初始化为INT_MIN
    for(int i=0;i<n;i++){
        int x=nums[i]%3;
        //保留上一个状态
        int a=dp[0];
        int b=dp[1];
        int c=dp[2];
        if(x==0){
            dp[0]+=nums[i];
            dp[1]+=nums[i];
            dp[2]+=nums[i];
        } 
        if(x==1){
            dp[0]=max(a,c+nums[i]);
            dp[1]=max(b,a+nums[i]);
            dp[2]=max(c,b+nums[i]);
        } 
        if(x==2){
            dp[0]=max(a,b+nums[i]);
            dp[1]=max(b,c+nums[i]);
            dp[2]=max(c,a+nums[i]);
        } 
    }
    return dp[0];
}