原题链接: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];
}