（C++/Python）LeetCode 1262 可被三整除的最大和

最新推荐文章于 2024-06-30 08:30:00 发布

原创最新推荐文章于 2024-06-30 08:30:00 发布 · 602 阅读

1 ·

本内容遵循CC 4.0 BY-SA版权协议

标签

#c++ #leetcode #python #动态规划 #被三整除的最大和

leetcode/剑指offer 专栏收录该内容

209 篇文章

订阅专栏

本文介绍了一种解决整数数组中寻找能被三整除的最大元素和的问题的方法。通过动态规划，利用三个状态dp[0], dp[1], dp[2]分别记录对3取余为0, 1, 2的最大和，最终得到数组中能被三整除的最大和。

题目描述：
给你一个整数数组 nums，请你找出并返回能被三整除的元素最大和。

思路：
对于数组中的一个数nums[i]，对三取余后可能是0，1，2
定义dp[0]表示数nums[i]之前的对3取余等于0的最大和
dp[1]表示数nums[i]之前的对3取余等于1的最大和
dp[2]表示数nums[i]之前的对3取余等于2的最大和
当nums[i]%30，则dp[0]，dp[1]，dp[2]都加上这个数，他们的余数不变
当nums[i]%31，dp[0]=max（dp[0]，dp[2]+nums[i]）
（因为dp[2]的余数是2，nums[i]的余数是1，则他们相加后的余数是0）
dp[1]=max（dp[1]，dp[0]+nums[i]）
（因为dp[0]的余数是0，nums[i]的余数是1，则他们相加后的余数是1）
dp[2]=max（dp[2]，dp[1]+nums[i]）
（因为dp[1]的余数是1，nums[i]的余数是1，则他们相加后的余数是1）

同理得
当nums[i]%3==2，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]）

代码如下：
C++：

class Solution {
public:
    int maxSumDivThree(vector<int>& nums) {
        vector<int>dp(3,0),temp(3,0);
        for(auto n:nums){
            temp=dp;
            if(n%3==0){
                dp[0]+=n;
                dp[1]+=n;
                dp[2]+=n;
            }
            if(n%3==1){
                if((n+temp[2])%3==0) dp[0]=max(dp[0],n+temp[2]);
                if((n+temp[0])%3==1) dp[1]=max(dp[1],n+temp[0]);
                if((n+temp[1])%3==2) dp[2]=max(dp[2],n+temp[1]);
            }
            if(n%3==2){
                if((n+temp[1])%3==0) dp[0]=max(dp[0],n+temp[1]);
                if((n+temp[2])%3==1) dp[1]=max(dp[1],n+temp[2]);
                if((n+temp[0])%3==2) dp[2]=max(dp[2],n+temp[0]);
            }
        }
        return dp[0];
    }
};

python:
（这个Python仿照上面C++写的，但是有bug，在下Python渣渣，哪位有发现问题在哪里，欢迎指出）

class Solution:
    def maxSumDivThree(self, nums: List[int]) -> int:
        # dp=[0,0,0]
        # temp=[0,0,0]
        # for i in range(len(nums)):
        #     temp=dp
        #     if nums[i]%3==0:
        #         dp[0]+=nums[i]
        #         dp[1]+=nums[i]
        #         dp[2]+=nums[i]
        #     elif nums[i]%3==1:
        #         if (nums[i]+temp[2])%3==0:
        #             dp[0]=max(dp[0],nums[i]+temp[2])
        #         elif (nums[i]+temp[0])%3==1:
        #             dp[1]=max(dp[1],nums[i]+temp[0])
        #         elif (nums[i]+temp[1])%3==2:
        #             dp[2]=max(dp[2],nums[i]+temp[1])
        #     elif nums[i]%3==2:
        #         if (nums[i]+temp[1])%3==0:
        #             dp[0]=max(dp[0],nums[i]+temp[1])
        #         elif (nums[i]+temp[2])%3==1:
        #             dp[1]=max(dp[1],nums[i]+temp[2])
        #         elif (nums[i]+temp[0])%3==2:
        #             dp[2]=max(dp[2],nums[i]+temp[0])
        #    return dp[0]

class Solution:
    def maxSumDivThree(self, nums: List[int]) -> int:
        dp = [0, float('-inf'), float('-inf')]
        for n in nums:
            dp = [max(dp[i], dp[(i+n)%3]+n) for i in range(3)]
        return dp[0]