test: 3541 solution

py, c++, go, java

Author: QuBenhao

problems/problems_3541/Solution.cpp

@@ -1,28 +1,41 @@
 //go:build ignore
 #include "cpp/common/Solution.h"
-
+#include <algorithm>
 
 using namespace std;
 using json = nlohmann::json;
 
+constexpr string VOWELS = "aeiou";
+
 class Solution {
 public:
-    int maxFreqSum(string s) {
-        
+  int maxFreqSum(const string &s) {
+    array<int, 5> vowels_count = {};
+    array<int, 26> non_vowels_count = {};
+    for (const auto &c : s) {
+      size_t pos = VOWELS.find(c);
+      if (pos != std::string::npos) {
+        ++vowels_count[pos];
+      } else {
+        ++non_vowels_count[c - 'a'];
+      }
     }
+    return *std::max_element(vowels_count.begin(), vowels_count.end()) +
+           *std::max_element(non_vowels_count.begin(), non_vowels_count.end());
+  }
 };
 
 json leetcode::qubh::Solve(string input_json_values) {
-	vector<string> inputArray;
-	size_t pos = input_json_values.find('\n');
-	while (pos != string::npos) {
-		inputArray.push_back(input_json_values.substr(0, pos));
-		input_json_values = input_json_values.substr(pos + 1);
-		pos = input_json_values.find('\n');
-	}
-	inputArray.push_back(input_json_values);
+  vector<string> inputArray;
+  size_t pos = input_json_values.find('\n');
+  while (pos != string::npos) {
+    inputArray.push_back(input_json_values.substr(0, pos));
+    input_json_values = input_json_values.substr(pos + 1);
+    pos = input_json_values.find('\n');
+  }
+  inputArray.push_back(input_json_values);
 
-	Solution solution;
-	string s = json::parse(inputArray.at(0));
-	return solution.maxFreqSum(s);
+  Solution solution;
+  string s = json::parse(inputArray.at(0));
+  return solution.maxFreqSum(s);
 }

problems/problems_3541/Solution.java

@@ -6,8 +6,27 @@
 
 
 public class Solution extends BaseSolution {
+    private static final String VOWELS = "aeiou";
+
     public int maxFreqSum(String s) {
-        
+        int[] vowelsCount = new int[5], nonVowesCount = new int[26];
+        for (char c: s.toCharArray()) {
+            int idx = VOWELS.indexOf(c);
+            if (idx != -1) {
+                ++vowelsCount[idx];
+            } else {
+                ++nonVowesCount[c - 'a'];
+            }
+        }
+        return arrayMax(vowelsCount) + arrayMax(nonVowesCount);
+    }
+
+    private int arrayMax(int[] arr) {
+        int ans = 0;
+        for (int v: arr) {
+            ans = Math.max(ans, v);
+        }
+        return ans;
     }
 
     @Override

problems/problems_3541/solution.go

@@ -6,8 +6,26 @@ import (
 	"strings"
 )
 
+const VOWELS = "aeiou"
+
 func maxFreqSum(s string) int {
+	vowelsCount := make([]int, 5)
+	nonVowelsCount := make([]int, 26)
+	for _, r := range s {
+		if idx := strings.IndexRune(VOWELS, r); idx >= 0 {
+			vowelsCount[idx]++
+		} else {
+			nonVowelsCount[r-'a']++
+		}
+	}
+	return maxElement(vowelsCount) + maxElement(nonVowelsCount)
+}
 
+func maxElement(arr []int) (ans int) {
+	for _, v := range arr {
+		ans = max(ans, v)
+	}
+	return
 }
 
 func Solve(inputJsonValues string) any {