feat: add solutions to lc problem: No.1426 (doocs#2059)

No.1426.Counting Elements

Author: yanglbme

solution/1400-1499/1426.Counting Elements/README.md

@@ -40,17 +40,11 @@
 
 <!-- 这里可写通用的实现逻辑 -->
 
-**方法一：暴力枚举**
+**方法一：计数**
 
-枚举 `arr` 的每个元素 `x`，判断 `x+1` 是否在 `arr` 中，是则累加答案。
+我们可以用一个哈希表或数组 $cnt$ 记录数组 $arr$ 中的每个数出现的次数，然后遍历 $cnt$ 中的每个数 $x$，如果 $x+1$ 也在 $cnt$ 中，那么就将 $cnt[x]$ 加到答案中。
 
-时间复杂度 $O(n^2)$，空间复杂度 $O(1)$。
-
-**方法二：哈希表**
-
-将 `arr` 所有元素放入哈希表 `s` 中。然后遍历 `arr` 的每个元素 `x`，判断 `x+1` 是否在 `s` 中，是则累加答案。
-
-时间复杂度 $O(n)$，空间复杂度 $O(n)$。
+时间复杂度 $O(n)$，空间复杂度 $O(n)$。其中 $n$ 是数组 $arr$ 的长度。
 
 <!-- tabs:start -->
 
@@ -61,14 +55,8 @@
 ```python
 class Solution:
     def countElements(self, arr: List[int]) -> int:
-        return sum(x + 1 in arr for x in arr)
-```
-
-```python
-class Solution:
-    def countElements(self, arr: List[int]) -> int:
-        s = set(arr)
-        return sum(x + 1 in s for x in arr)
+        cnt = Counter(arr)
+        return sum(v for x, v in cnt.items() if cnt[x + 1])
 ```
 
 ### **Java**
@@ -78,34 +66,17 @@ class Solution:
 ```java
 class Solution {
     public int countElements(int[] arr) {
-        int ans = 0;
+        int[] cnt = new int[1001];
         for (int x : arr) {
-            for (int v : arr) {
-                if (x + 1 == v) {
-                    ++ans;
-                    break;
-                }
-            }
+            ++cnt[x];
         }
-        return ans;
-    }
-}
-```
-
-```java
-class Solution {
-    public int countElements(int[] arr) {
-        Set<Integer> s = new HashSet<>();
-        for (int num : arr) {
-            s.add(num);
-        }
-        int res = 0;
-        for (int num : arr) {
-            if (s.contains(num + 1)) {
-                ++res;
+        int ans = 0;
+        for (int x = 0; x < 1000; ++x) {
+            if (cnt[x + 1] > 0) {
+                ans += cnt[x];
             }
         }
-        return res;
+        return ans;
     }
 }
 ```
@@ -116,28 +87,15 @@ class Solution {
 class Solution {
 public:
     int countElements(vector<int>& arr) {
-        int ans = 0;
+        int cnt[1001]{};
         for (int x : arr) {
-            for (int v : arr) {
-                if (x + 1 == v) {
-                    ++ans;
-                    break;
-                }
-            }
+            ++cnt[x];
         }
-        return ans;
-    }
-};
-```
-
-```cpp
-class Solution {
-public:
-    int countElements(vector<int>& arr) {
-        unordered_set<int> s(arr.begin(), arr.end());
         int ans = 0;
-        for (int x : arr) {
-            ans += s.count(x + 1);
+        for (int x = 0; x < 1000; ++x) {
+            if (cnt[x + 1]) {
+                ans += cnt[x];
+            }
         }
         return ans;
     }
@@ -147,72 +105,82 @@ public:
 ### **Go**
 
 ```go
-func countElements(arr []int) int {
-	ans := 0
-	for _, x := range arr {
-		for _, v := range arr {
-			if x+1 == v {
-				ans++
-				break
-			}
-		}
-	}
-	return ans
-}
-```
-
-```go
-func countElements(arr []int) int {
-	s := map[int]bool{}
+func countElements(arr []int) (ans int) {
+	mx := slices.Max(arr)
+	cnt := make([]int, mx+1)
 	for _, x := range arr {
-		s[x] = true
+		cnt[x]++
 	}
-	ans := 0
-	for _, x := range arr {
-		if s[x+1] {
-			ans++
+	for x := 0; x < mx; x++ {
+		if cnt[x+1] > 0 {
+			ans += cnt[x]
 		}
 	}
-	return ans
+	return
 }
 ```
 
-### **JavaScript**
+### **TypeScript**
 
-```js
-/**
- * @param {number[]} arr
- * @return {number}
- */
-var countElements = function (arr) {
-    let ans = 0;
+```ts
+function countElements(arr: number[]): number {
+    const mx = Math.max(...arr);
+    const cnt = Array(mx + 1).fill(0);
     for (const x of arr) {
-        ans += arr.includes(x + 1);
+        ++cnt[x];
+    }
+    let ans = 0;
+    for (let i = 0; i < mx; ++i) {
+        if (cnt[i + 1] > 0) {
+            ans += cnt[i];
+        }
     }
     return ans;
-};
+}
 ```
 
+### **JavaScript**
+
 ```js
 /**
  * @param {number[]} arr
  * @return {number}
  */
 var countElements = function (arr) {
-    const s = new Set();
+    const mx = Math.max(...arr);
+    const cnt = Array(mx + 1).fill(0);
     for (const x of arr) {
-        s.add(x);
+        ++cnt[x];
     }
     let ans = 0;
-    for (const x of arr) {
-        if (s.has(x + 1)) {
-            ++ans;
+    for (let i = 0; i < mx; ++i) {
+        if (cnt[i + 1] > 0) {
+            ans += cnt[i];
         }
     }
     return ans;
 };
 ```
 
+### **Rust**
+
+```rust
+use std::collections::HashMap;
+
+impl Solution {
+    pub fn count_elements(arr: Vec<i32>) -> i32 {
+        let mut cnt = HashMap::new();
+        for &num in &arr {
+            *cnt.entry(num).or_insert(0) += 1;
+        }
+        cnt.iter()
+            .filter(|(&x, _)| cnt.contains_key(&(x + 1)))
+            .map(|(_, &v)| v)
+            .sum()
+    }
+}
+```
+
 ### **PHP**
 
 ```php
@@ -222,13 +190,14 @@ class Solution {
      * @return Integer
      */
     function countElements($arr) {
-        $cnt = 0;
-        for ($i = 0; $i < count($arr); $i++) {
-            if (in_array($arr[$i] + 1, $arr)) {
-                $cnt++;
+        $cnt = array_count_values($arr);
+        $ans = 0;
+        foreach ($cnt as $x => $v) {
+            if (isset($cnt[$x + 1])) {
+                $ans += $v;
             }
         }
-        return $cnt++;
+        return $ans;
     }
 }
 ```