Runtime: 26 ms (Top 90.8%) | Memory: 26.10 MB (Top 93.66%)

Author: AnasImloul

scripts/algorithms/T/Teemo Attacking/Teemo Attacking.cpp

@@ -1,11 +1,17 @@
+// Runtime: 26 ms (Top 90.8%) | Memory: 26.10 MB (Top 93.66%)
+
 class Solution {
 public:
     int findPoisonedDuration(vector<int>& timeSeries, int duration) {
-        int ans=0;
-        int n=timeSeries.size();
-        for(int i=0;i<n;i++){
-            ans+=min(duration,(i==n-1?duration:timeSeries[i+1]-timeSeries[i]));
+        if (timeSeries.size() == 0)
+            return 0;
+        int res = 0;
+        for(int i=0; i<timeSeries.size()-1; i++) {
+            if (timeSeries[i+1] - timeSeries[i] < duration)
+                res += timeSeries[i+1] - timeSeries[i];
+            else
+                res += duration;
         }
-        return ans;
+        return res+duration;
     }
-};
+};