Advertisement
maycod23

subarray_questions.cpp

Mar 11th, 2022
1,058
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 0.56 KB | None | 0 0
  1. class Solution {
  2. public:
  3.     int subarraysDivByK(vector<int>& nums, int k)
  4.     {
  5.         unordered_map<int,int> m; m[0]++;
  6.         int sum=0,count=0;
  7.         for(int i=0;i<nums.size();i++)
  8.         {
  9.             //ith index->as end
  10.             sum+=nums[i];
  11.             sum=sum%k;
  12.             if(sum<0) sum+=k;
  13.             if(m.find(sum%k)!=m.end())
  14.             {
  15.                 count+=m[sum%k];
  16.                 m[sum%k]++;
  17.             }
  18.             else
  19.             {
  20.                 m[sum%k]++;
  21.             }
  22.         }
  23.         return count;
  24.     }
  25. };
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement