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

Author: kamyu104

Python/number-of-unequal-triplets-in-array.py

@@ -15,9 +15,9 @@ def unequalTriplets(self, nums):
         cnt = collections.Counter()
         dp = [0]*K  # dp[i]: number of unequal (i+1)-plets
         for x in nums:
+            cnt[x] += 1
             other_cnt = 1
             for i in xrange(K):
                 dp[i] = dp[i]+other_cnt
-                other_cnt = dp[i]-(cnt[x]+1)*other_cnt
-            cnt[x] += 1
+                other_cnt = dp[i]-cnt[x]*other_cnt
         return dp[K-1]