力扣974.和可被K整除的子数组

力扣974.和可被K整除的子数组

• 将余数相同的做差

  • 若为负数要翻正再存入哈希表
  • 若为正数要存入哈希表
  • 统一操作 (sj % k + k ) % k

  class Solution {
  public:
      int subarraysDivByK(vector<int>& nums, int k) {
          int n = nums.size();
          vector<long> s(n+1);
          for(int i=0;i<n;i++) s[i+1] = s[i] + nums[i];
  
          int res=0;
          unordered_map<int,int> cnt;
          for(long sj:s)
          {
              //重要操作
              int t = (sj % k + k) % k;
              res += cnt[t];
              cnt[t] ++;
          }
          return res;
      }
  };