Update number-of-unequal-triplets-in-array.cpp

Author: kamyu104

C++/number-of-unequal-triplets-in-array.cpp

@@ -10,12 +10,12 @@ class Solution {
         unordered_map<int, int> cnt;
         vector<int> dp(K);  // dp[i]: number of unequal (i+1)-plets
         for (const auto& x : nums) {
+            ++cnt[x];
             int other_cnt = 1;
             for (int i = 0; i < K; ++i) {
                 dp[i] += other_cnt;
-                other_cnt = dp[i] - (cnt[x] + 1) * other_cnt;
+                other_cnt = dp[i] - cnt[x] * other_cnt;
             }
-            ++cnt[x];
         }
         return dp[K - 1];
     }