力扣 523. 连续的子数组和 hash+dp

https://leetcode-cn.com/problems/continuous-subarray-sum/

力扣 523. 连续的子数组和 hash+dp_第1张图片

思路: h a s h + d p hash+dp hash+dp。搞一个 h a s h hash hash表, d p [ s u m ] = i dp[sum]=i dp[sum]=i,表示 [ 0 … i ] [0…i] [0i]的前缀和 s i % k = s u m s_i\%k=sum si%k=sum,初始可以令 d p [ 0 ] = − 1 dp[0]=-1 dp[0]=1,然后遍历整个数组,记录前缀和 s i s_i si,如果 d p [ s i % k ] dp[s_i\%k] dp[si%k]之前出现过且 i − d p [ s i % k ] > = 2 i-dp[s_i\%k]>=2 idp[si%k]>=2,就说明有一组满足题意的解。

class Solution {
public:
    bool checkSubarraySum(vector<int>& nums, int k) {
        int siz=nums.size();
        if(siz<=1)
            return 0;
        k=abs(k);
        for(int i=1;i<siz;i++)
            if(nums[i]==0&&nums[i-1]==0)    
                 return 1;
        if(!k)
            return 0;
        unordered_map<int,int> dp;
        dp[0]=-1;
        int sum=0;
        for(int i=0;i<siz;i++){
            sum+=nums[i];
            sum%=k;
            if(dp.find(sum)==dp.end())
                dp[sum]=i;
            else if(i-dp[sum]>=2)
                return 1;
        }
        return 0;
    }
};

你可能感兴趣的:(面试题,哈希,dp,动态规划)