Update 1354.Construct-Target-Array-With-Multiple-Sums.cpp

wisdompeak · web-flow · commit df93126e65da · 2021-11-10T01:29:23.000-08:00
diff --git a/Greedy/1354.Construct-Target-Array-With-Multiple-Sums/1354.Construct-Target-Array-With-Multiple-Sums.cpp b/Greedy/1354.Construct-Target-Array-With-Multiple-Sums/1354.Construct-Target-Array-With-Multiple-Sums.cpp
@@ -2,27 +2,27 @@ class Solution {
 public:
     bool isPossible(vector<int>& target) 
     {
-        long long sum = 0;
         priority_queue<int>pq;
-        for (int i=0; i<target.size(); i++)
-        {
-            pq.push(target[i]);
-            sum+=target[i];
-        }
+        for (auto x: target) pq.push(x);
+        long long sum = 0;
+        for (auto x: target) 
+            sum += x;
         
         while (pq.top()!=1)
         {
             long long a = pq.top();
-            long long others = sum - a;
             pq.pop();
-
-            if (others ==0) return false;
-            if (a - others <= 0) return false;
+            long long others = sum - a;
+            
+            if (others == 0) return false;      
+            if (a<=others) return false;
             long long b = a % others;
+            if (b==0) b = others;
             
-            sum = others + b;
-            pq.push(b);            
-        }     
-        return true;
+            sum = others+b;
+            pq.push(b);
+        }
+        
+        return true;        
     }
 };
