feat: add solutions to lc problem: No.1773

No.1773.Count Items Matching a Rule

Author: yanglbme

solution/0900-0999/0907.Sum of Subarray Minimums/README.md

@@ -45,11 +45,33 @@
 
 **方法一：单调栈**
 
-枚举 $arr[i]$ 作为子数组的最小值，找出左侧第一个比 $arr[i]$ 小的位置 $left[i]$，右侧第一个不大于 $arr[i]$ 的位置 $right[i]$。由此可以算出最小值 $arr[i]$ 可以出现在多少个子数组中。
+题目要求的是每个子数组的最小值之和，实际上相当于，对于每个元素 $arr[i]$，求以 $arr[i]$ 为最小值的子数组的个数，然后乘以 $arr[i]$，最后求和。
 
-计算每个 $arr[i]$ 的贡献 $(i-left[i])*(right[i]-i)*arr[i]$，累加得到结果。
+因此，题目的重点转换为：求以 $arr[i]$ 为最小值的子数组的个数。对于 $arr[i]$，我们找出其左边第一个小于 $arr[i]$ 的位置 $left[i]$，右侧第一个小于等于 $arr[i]$ 的位置 $right[i]$，则以 $arr[i]$ 为最小值的子数组的个数为 $(i - left[i]) \times (right[i] - i)$。
 
-注意数据溢出与取模操作。
+注意，这里为什么要求右侧第一个小于等于 $arr[i]$ 的位置 $right[i]$，而不是小于 $arr[i]$ 的位置呢？这是因为，如果是右侧第一个小于 $arr[i]$ 的位置 $right[i]$，则会导致重复计算。
+
+我们可以举个例子来说明，对于以下数组：
+
+下标为 $3$ 的元素大小为 $2$，左侧第一个小于 $2$ 的元素下标为 $0$，如果我们求右侧第一个小于 $2$ 的元素下标，可以得到下标为 $7$。也即是说，子数组区间为 $(0, 7)$。注意，这里是开区间。
+
+```
+0 4 3 2 5 3 2 1
+*     ^       *
+```
+
+按照同样的方法，我们可以求出下标为 $6$ 的元素的子数组区间，可以发现，其子数组区间也为 $(0, 7)$，也即是说，下标为 $3$ 和下标为 $6$ 的元素的子数组区间是重复的。这样就造成了重复计算。
+
+```
+0 4 3 2 5 3 2 1
+*           ^ *
+```
+
+如果我们求的是右侧第一个小于等于其值的下标，就不会有重复问题，因为下标为 $3$ 的子数组区间变为 $(0, 6)$，下标为 $6$ 的子数组区间为 $(0, 7)$，两者不重复。
+
+回到这道题上，我们只需要遍历数组，对于每个元素 $arr[i]$，利用单调栈求出其左侧第一个小于 $arr[i]$ 的位置 $left[i]$，右侧第一个小于等于 $arr[i]$ 的位置 $right[i]$，则以 $arr[i]$ 为最小值的子数组的个数为 $(i - left[i]) \times (right[i] - i)$，然后乘以 $arr[i]$，最后求和即可。
+
+注意数据的溢出以及取模操作。
 
 时间复杂度 $O(n)$，其中 $n$ 表示数组 $arr$ 的长度。
 
@@ -72,14 +94,15 @@ class Solution:
             if stk:
                 left[i] = stk[-1]
             stk.append(i)
+
         stk = []
         for i in range(n - 1, -1, -1):
             while stk and arr[stk[-1]] > arr[i]:
                 stk.pop()
             if stk:
                 right[i] = stk[-1]
             stk.append(i)
-        mod = int(1e9) + 7
+        mod = 10**9 + 7
         return sum((i - left[i]) * (right[i] - i) * v for i, v in enumerate(arr)) % mod
 ```
 

solution/0900-0999/0907.Sum of Subarray Minimums/README_EN.md

@@ -52,14 +52,15 @@ class Solution:
             if stk:
                 left[i] = stk[-1]
             stk.append(i)
+
         stk = []
         for i in range(n - 1, -1, -1):
             while stk and arr[stk[-1]] > arr[i]:
                 stk.pop()
             if stk:
                 right[i] = stk[-1]
             stk.append(i)
-        mod = int(1e9) + 7
+        mod = 10**9 + 7
         return sum((i - left[i]) * (right[i] - i) * v for i, v in enumerate(arr)) % mod
 ```
 

solution/0900-0999/0907.Sum of Subarray Minimums/Solution.py

@@ -10,12 +10,13 @@ def sumSubarrayMins(self, arr: List[int]) -> int:
             if stk:
                 left[i] = stk[-1]
             stk.append(i)
+        
         stk = []
         for i in range(n - 1, -1, -1):
             while stk and arr[stk[-1]] > arr[i]:
                 stk.pop()
             if stk:
                 right[i] = stk[-1]
             stk.append(i)
-        mod = int(1e9) + 7
-        return sum((i - left[i]) * (right[i] - i) * v for i, v in enumerate(arr)) % mod
+        mod = 10**9 + 7
+        return sum((i - left[i]) * (right[i] - i) * v for i, v in enumerate(arr)) % mod

solution/1700-1799/1773.Count Items Matching a Rule/README.md

@@ -52,6 +52,12 @@
 
 <!-- 这里可写通用的实现逻辑 -->
 
+**方法一：直接模拟**
+
+由于 `ruleKey` 只可能是 `"type"`、`"color"` 或 `"name"`，我们可以直接取 `ruleKey` 的第一个字符 $x$ 来确定 `item` 的下标 $i$。然后遍历 `items` 数组，统计 `item[i] == ruleValue` 的个数即可。
+
+时间复杂度 $O(n)$，空间复杂度 $O(1)$。其中 $n$ 为 `items` 的长度。
+
 <!-- tabs:start -->
 
 ### **Python3**
@@ -61,9 +67,8 @@
 ```python
 class Solution:
     def countMatches(self, items: List[List[str]], ruleKey: str, ruleValue: str) -> int:
-        count = 0
-        m = {'type': 0, 'color': 1, 'name': 2}
-        return sum([item[m[ruleKey]] == ruleValue for item in items])
+        i = 0 if ruleKey[0] == 't' else (1 if ruleKey[0] == 'c' else 2)
+        return sum(v[i] == ruleValue for v in items)
 ```
 
 ### **Java**
@@ -73,22 +78,46 @@ class Solution:
 ```java
 class Solution {
     public int countMatches(List<List<String>> items, String ruleKey, String ruleValue) {
-        int count = 0;
-        for (List<String> item : items) {
-            String t = item.get(0), c = item.get(1), n = item.get(2);
-            if ("type".equals(ruleKey) && t.equals(ruleValue)) {
-                ++count;
-            } else if ("color".equals(ruleKey) && c.equals(ruleValue)) {
-                ++count;
-            } else if ("name".equals(ruleKey) && n.equals(ruleValue)) {
-                ++count;
+        char x = ruleKey.charAt(0);
+        int i = x == 't' ? 0 : (x == 'c' ? 1 : 2);
+        int ans = 0;
+        for (var v : items) {
+            if (v.get(i).equals(ruleValue)) {
+                ++ans;
             }
         }
-        return count;
+        return ans;
     }
 }
 ```
 
+### **C++**
+
+```cpp
+class Solution {
+public:
+    int countMatches(vector<vector<string>>& items, string ruleKey, string ruleValue) {
+        char x = ruleKey[0];
+        int i = x == 't' ? 0 : (x == 'c' ? 1 : 2);
+        return count_if(items.begin(), items.end(), [&](auto& v) { return v[i] == ruleValue; });
+    }
+};
+```
+
+### **Go**
+
+```go
+func countMatches(items [][]string, ruleKey string, ruleValue string) (ans int) {
+	i := map[byte]int{'t': 0, 'c': 1, 'n': 2}[ruleKey[0]]
+	for _, v := range items {
+		if v[i] == ruleValue {
+			ans++
+		}
+	}
+	return
+}
+```
+
 ### **...**
 
 ```

solution/1700-1799/1773.Count Items Matching a Rule/README_EN.md

@@ -51,32 +51,55 @@
 ```python
 class Solution:
     def countMatches(self, items: List[List[str]], ruleKey: str, ruleValue: str) -> int:
-        count = 0
-        m = {'type': 0, 'color': 1, 'name': 2}
-        return sum([item[m[ruleKey]] == ruleValue for item in items])
+        i = 0 if ruleKey[0] == 't' else (1 if ruleKey[0] == 'c' else 2)
+        return sum(v[i] == ruleValue for v in items)
 ```
 
 ### **Java**
 
 ```java
 class Solution {
     public int countMatches(List<List<String>> items, String ruleKey, String ruleValue) {
-        int count = 0;
-        for (List<String> item : items) {
-            String t = item.get(0), c = item.get(1), n = item.get(2);
-            if ("type".equals(ruleKey) && t.equals(ruleValue)) {
-                ++count;
-            } else if ("color".equals(ruleKey) && c.equals(ruleValue)) {
-                ++count;
-            } else if ("name".equals(ruleKey) && n.equals(ruleValue)) {
-                ++count;
+        char x = ruleKey.charAt(0);
+        int i = x == 't' ? 0 : (x == 'c' ? 1 : 2);
+        int ans = 0;
+        for (var v : items) {
+            if (v.get(i).equals(ruleValue)) {
+                ++ans;
             }
         }
-        return count;
+        return ans;
     }
 }
 ```
 
+### **C++**
+
+```cpp
+class Solution {
+public:
+    int countMatches(vector<vector<string>>& items, string ruleKey, string ruleValue) {
+        char x = ruleKey[0];
+        int i = x == 't' ? 0 : (x == 'c' ? 1 : 2);
+        return count_if(items.begin(), items.end(), [&](auto& v) { return v[i] == ruleValue; });
+    }
+};
+```
+
+### **Go**
+
+```go
+func countMatches(items [][]string, ruleKey string, ruleValue string) (ans int) {
+	i := map[byte]int{'t': 0, 'c': 1, 'n': 2}[ruleKey[0]]
+	for _, v := range items {
+		if v[i] == ruleValue {
+			ans++
+		}
+	}
+	return
+}
+```
+
 ### **...**
 
 ```

solution/1700-1799/1773.Count Items Matching a Rule/Solution.cpp

@@ -0,0 +1,8 @@
+class Solution {
+public:
+    int countMatches(vector<vector<string>>& items, string ruleKey, string ruleValue) {
+        char x = ruleKey[0];
+        int i = x == 't' ? 0 : (x == 'c' ? 1 : 2);
+        return count_if(items.begin(), items.end(), [&](auto& v) { return v[i] == ruleValue; });
+    }
+};

solution/1700-1799/1773.Count Items Matching a Rule/Solution.go

@@ -0,0 +1,9 @@
+func countMatches(items [][]string, ruleKey string, ruleValue string) (ans int) {
+	i := map[byte]int{'t': 0, 'c': 1, 'n': 2}[ruleKey[0]]
+	for _, v := range items {
+		if v[i] == ruleValue {
+			ans++
+		}
+	}
+	return
+}

solution/1700-1799/1773.Count Items Matching a Rule/Solution.java

@@ -1,16 +1,13 @@
 class Solution {
     public int countMatches(List<List<String>> items, String ruleKey, String ruleValue) {
-        int count = 0;
-        for (List<String> item : items) {
-            String t = item.get(0), c = item.get(1), n = item.get(2);
-            if ("type".equals(ruleKey) && t.equals(ruleValue)) {
-                ++count;
-            } else if ("color".equals(ruleKey) && c.equals(ruleValue)) {
-                ++count;
-            } else if ("name".equals(ruleKey) && n.equals(ruleValue)) {
-                ++count;
+        char x = ruleKey.charAt(0);
+        int i = x == 't' ? 0 : (x == 'c' ? 1 : 2);
+        int ans = 0;
+        for (var v : items) {
+            if (v.get(i).equals(ruleValue)) {
+                ++ans;
             }
         }
-        return count;
+        return ans;
     }
 }

solution/1700-1799/1773.Count Items Matching a Rule/Solution.py

@@ -1,5 +1,4 @@
 class Solution:
     def countMatches(self, items: List[List[str]], ruleKey: str, ruleValue: str) -> int:
-        count = 0
-        m = {'type': 0, 'color': 1, 'name': 2}
-        return sum([item[m[ruleKey]] == ruleValue for item in items])
+        i = 0 if ruleKey[0] == 't' else (1 if ruleKey[0] == 'c' else 2)
+        return sum(v[i] == ruleValue for v in items)