Merge pull request #7258 from deadmarshal/TWC194

TWC194

1 files changed, 24 insertions, 0 deletions

diff --git a/challenge-194/deadmarshal/cpp/ch-2.cpp b/challenge-194/deadmarshal/cpp/ch-2.cpp
new file mode 100644
index 0000000000..8eaf52a1d8
--- /dev/null
+++ b/challenge-194/deadmarshal/cpp/ch-2.cpp
@@ -0,0 +1,24 @@
+#include<iostream>
+#include<vector>
+#include<string>
+#include<unordered_map>
+#include<algorithm>
+
+bool frequency_equalizer(const std::string &str)
+{
+  std::unordered_map<char,unsigned> m{};
+  std::vector<unsigned> vals{};
+  for(const auto &c : str) m[c]++;
+  for(const auto& kv : m) vals.push_back(kv.second);
+  std::sort(vals.begin(),vals.end(),std::greater<>());
+  if((vals[0] == vals[1]+1) && (vals.back() == vals[1])) return true;
+  return false;
+}
+
+int main()
+{
+  std::cout << std::boolalpha << frequency_equalizer("abbc")<< '\n';
+  std::cout << std::boolalpha << frequency_equalizer("xyzyyxz")<<'\n';
+  std::cout << std::boolalpha << frequency_equalizer("xzxz")<< '\n';
+  return 0;
+}