一道动态规划的题,设一个dp数组。
有三种状态,对0取余为0,1,2 。
求每个数组元素对0取余的值,并动态更新dp数组。
例如:0的dp数组+取余为0的还是0
0的dp数组+取余为1的是1
0的dp数组+取余为2的是2.
这里有一点需要注意,即0,1,2三种情况的初始化。
0的数组可以直接初始化为0,而1和2的数组都需要初始化为一个很小的值(即一个绝对值比较大的负数),这是因为不能让1和2的数组加上前面的可被3整除的数而对0的数组产生影响
如果1和2的数组初始化为0或者比较小的负数,就会在后面max判断中对0数组产生一些影响,使结果错误。所以一定要把他们赋很小的初值,使得直到数组中出现余数为1或2的元素,1和2的数组才会被更新为正数。
class Solution {
public:
int maxSumDivThree(vector& nums) {
int n=nums.size();
int result=0;
vector<vector> dp(n+1,vector(3,0));
dp[0][0]=0,dp[0][1]=-9999,dp[0][2]=-9999;
for(int i=0;i<n;i++){
// cout<<“dp[0]:”<<dp[i][0]<<endl;
//cout<<“dp[1]:”<<dp[i][1]<<endl;
//cout<<“dp[2]:”<<dp[i][2]<<endl;
if(nums[i]%30){
dp[i+1][0]+=max(dp[i][0],dp[i][0]+nums[i]);
dp[i+1][1]+=max(dp[i][1],dp[i][1]+nums[i]);
dp[i+1][2]+=max(dp[i][2],dp[i][2]+nums[i]);
}
else if(nums[i]%31){
dp[i+1][0]=max(dp[i][0],dp[i][2]+nums[i]);
dp[i+1][1]=max(dp[i][1],dp[i][0]+nums[i]);
dp[i+1][2]=max(dp[i][2],dp[i][1]+nums[i]);
}
else{
dp[i+1][0]=max(dp[i][0],dp[i][1]+nums[i]);
dp[i+1][1]=max(dp[i][1],dp[i][2]+nums[i]);
dp[i+1][2]=max(dp[i][2],dp[i][0]+nums[i]);
}
}
result=dp[n][0];
return result;
}
};