GH-169: Solve codeforces 1788A, 1789A (#252)

Author: rain1024

collections/codeforces/README.md

@@ -17882,7 +17882,7 @@
 </tr>
 <tr>
 <tr>
-<td>1788A</td>
+<td>🟢&nbsp<a href='https://github.com/rain1024/datastructures-algorithms-competitive-programming/tree/main/problems/codeforces1788A'>1788A</a></td>
 <td>1788B</td>
 <td>1788C</td>
 <td>1788D</td>
@@ -17892,7 +17892,7 @@
 </tr>
 <tr>
 <tr>
-<td>1789A</td>
+<td>🟢&nbsp<a href='https://github.com/rain1024/datastructures-algorithms-competitive-programming/tree/main/problems/codeforces1789A'>1789A</a></td>
 <td>1789B</td>
 <td>1789C</td>
 <td>1789D</td>

collections/codeforces/data.yaml

@@ -71,6 +71,12 @@ problems:
 - name: 492B
   languages: cpp
   score: 1200
+- name: 1788A
+  languages: cpp
+  score: 800
+- name: 1789A
+  languages: cpp
+  score: 800
 - name: 1790A
   languages: cpp
   score: 800

collections/codeforces/generate_readme.py

@@ -17,7 +17,7 @@
     id = item['name']
     problems[id] = item
     solved_problems.add(id)
-print(f"Solve {len(solved_problems)} problems")
+print(f"Codeforces: {len(solved_problems)} problems solved.")
 
 def get_level(problem):
     if "score" not in problem:

problems/codeforces1788A/.gitignore

@@ -0,0 +1,2 @@
+solution
+*.dSYM

problems/codeforces1788A/README.md

@@ -0,0 +1,19 @@
+# Codeforces Problem 
+
+## Usage
+
+Run program with an example
+
+```
+bazel run cpp/main:solution < data/1.in
+```
+
+Test program
+
+```
+# Run all tests
+bazel test --test_output=all tests:solution_test
+bazel test --test_output=all --cache_test_results=no tests:solution_test
+# Run test with pecific test id
+bazel test --test_output=all tests:solution_test --test_arg=1
+```

problems/codeforces1788A/cpp/main/BUILD

@@ -0,0 +1,7 @@
+load("@rules_cc//cc:defs.bzl", "cc_binary")
+
+cc_binary(
+    name = "solution",
+    srcs = ["solution.cpp"],
+    visibility = ["//visibility:public"],
+)

problems/codeforces1788A/cpp/main/solution.cpp

@@ -0,0 +1,53 @@
+#include <algorithm>
+#include <cmath>
+#include <iostream>
+#include <map>
+#include <set>
+#include <string>
+#include <unordered_map>
+#include <vector>
+// #include <fstream>
+
+using namespace std;
+
+void solve(){
+  int n;
+  cin >> n;
+  vector<unsigned long long> a(n);
+  int c = 0;
+  for (int i = 0; i < n; i++) {
+    cin >> a[i];
+    c += (a[i] == 2);
+  }
+  if(c % 2 == 1){
+    cout << "-1" << endl;
+    return;
+  }
+  int ci = 0;
+  for(int i=0; i<n-1; i++){
+    ci += (a[i] == 2);
+    if(ci == c / 2){
+      cout << i+1 << endl; // 1-based index
+      return;
+    }
+  }
+  cout << "-1" << endl;
+}
+
+int main() {
+  ios::sync_with_stdio(false);
+  cin.tie(0);
+
+  // filesystem::path filepath =
+  //     filesystem::current_path().parent_path().parent_path() / "data" /
+  //     "1.in";
+  // ifstream file(filepath);
+  //   cin.rdbuf(file.rdbuf());  // redirect cin to file
+
+  int tt;
+  cin >> tt;
+  while (tt--) {
+    solve();
+  }
+  return 0;
+}

problems/codeforces1788A/cpp/tests/BUILD

@@ -0,0 +1,11 @@
+load("@rules_cc//cc:defs.bzl", "cc_binary", "cc_test")
+
+cc_test(
+    name = "solution_test",
+    srcs = ["solution_test.cpp"],
+    deps = [
+      "//problems/codeforces2A/src/main:solution",
+      "@com_google_googletest//:gtest_main",
+    ],
+    data = ["data"]
+)

problems/codeforces1788A/cpp/tests/solution_test.cpp

@@ -0,0 +1,97 @@
+#include <cstdlib>
+#include <filesystem>
+#include <fstream>
+#include <iostream>
+#include <string>
+#include <string_view>
+#include <vector>
+
+#include "gtest/gtest.h"
+
+namespace fs = std::filesystem;
+
+using namespace std;
+
+std::string TEST_ID = "";
+
+vector<string> split(string s, string delimiter) {
+  size_t pos_start = 0, pos_end, delim_len = delimiter.length();
+  string token;
+  vector<string> res;
+
+  while ((pos_end = s.find(delimiter, pos_start)) != string::npos) {
+    token = s.substr(pos_start, pos_end - pos_start);
+    pos_start = pos_end + delim_len;
+    res.push_back(token);
+  }
+
+  res.push_back(s.substr(pos_start));
+  return res;
+}
+
+std::string GetHello(std::string_view in) {
+  if (in.size() == 0) {
+    return std::string("hello, word");
+  } else {
+    return std::string("Hello, ") + in.data();
+  }
+}
+
+string ReadFile(const std::string &filename) {
+  std::ifstream f(filename);
+  string s((std::istreambuf_iterator<char>(f)),
+           std::istreambuf_iterator<char>());
+  return s;
+}
+
+vector<string> RunProgram(string fileid) {
+  vector<string> v;
+  string program = "problems/codeforces2A/src/main/solution";
+  string test_folder = "problems/codeforces2A/tests/data/";
+  string command = program + " < " + test_folder + fileid + ".in > output.txt";
+  std::system(command.c_str());
+  string actual = ReadFile("output.txt");
+  string expectedOutputFile = test_folder + fileid + ".out";
+  string expected = ReadFile(expectedOutputFile);
+  v.push_back(actual);
+  v.push_back(expected);
+  return v;
+}
+
+TEST(RunTest, AllTestCases) {
+  if (TEST_ID != "") {
+    std::cout << "TEST_ID = " << TEST_ID << std::endl;
+  } else {
+    std::cout << "RUN ALL TESTS" << std::endl;
+  }
+
+  std::string test_data_folder = "problems/codeforces2A/tests/data";
+  for (const auto &entry : fs::directory_iterator(test_data_folder)) {
+    string filename = split(entry.path(), "/").back();
+    vector<string> v = split(filename, ".");
+    string fileid = v[0];
+    string extension = v[1];
+    if (extension != "in"){
+      continue;
+    }
+    if (TEST_ID != ""){
+      if (fileid != TEST_ID){
+        continue;
+      }
+    }
+    // Run Test Case
+    vector<string> output = RunProgram(fileid);
+    string actual = output[0];
+    string expected = output[1];
+    string message = "❌ FAIL CASE: " + fileid;
+    ASSERT_EQ(actual, expected) << message;
+  }
+}
+
+int main(int argc, char **argv) {
+  ::testing::InitGoogleTest(&argc, argv);
+  if (argc == 2) {
+    TEST_ID = argv[1];
+  }
+  return RUN_ALL_TESTS();
+}

problems/codeforces1788A/data/1.in

@@ -0,0 +1,7 @@
+3
+6
+2 2 1 2 1 2
+3
+1 2 1
+4
+1 1 1 1

problems/codeforces1788A/data/1.out

@@ -0,0 +1,2 @@
+1
+-1

problems/codeforces1788A/data/2.in

@@ -0,0 +1,3 @@
+2
+-1
+1

problems/codeforces1788A/data/2.out

@@ -0,0 +1,10 @@
+9
+8
+7
+6
+5
+4
+3
+2
+1
+0

problems/codeforces1789A/.gitignore

@@ -0,0 +1,2 @@
+solution
+*.dSYM

problems/codeforces1789A/README.md

@@ -0,0 +1,19 @@
+# Codeforces Problem 
+
+## Usage
+
+Run program with an example
+
+```
+bazel run cpp/main:solution < data/1.in
+```
+
+Test program
+
+```
+# Run all tests
+bazel test --test_output=all tests:solution_test
+bazel test --test_output=all --cache_test_results=no tests:solution_test
+# Run test with pecific test id
+bazel test --test_output=all tests:solution_test --test_arg=1
+```

problems/codeforces1789A/cpp/main/BUILD

@@ -0,0 +1,7 @@
+load("@rules_cc//cc:defs.bzl", "cc_binary")
+
+cc_binary(
+    name = "solution",
+    srcs = ["solution.cpp"],
+    visibility = ["//visibility:public"],
+)

problems/codeforces1789A/cpp/main/solution.cpp

@@ -0,0 +1,53 @@
+#include <algorithm>
+#include <cmath>
+#include <iostream>
+#include <map>
+#include <set>
+#include <string>
+#include <unordered_map>
+#include <vector>
+#include <numeric>
+// #include <fstream>
+
+using namespace std;
+
+void solve(){
+  int n;
+  cin >> n;
+  vector<int> a(n);
+  for(int i=0; i<n; i++){
+    cin >> a[i];
+  }
+
+  for(int i=0; i<n-1; i++){
+    for(int j=i+1; j<n; j++){
+      // cout << "a[" << i << "]=" << a[i] << ", a[" << j << "]=" << a[j] << endl;
+      // cout << "gcd(a[" << i << "], a[" << j << "])=" << gcd(a[i], a[j]) << endl;
+      if(gcd(a[i], a[j]) <= 2){
+        cout << "Yes" << endl;
+        return;
+      }
+    }
+  }
+  cout << "No" << endl;
+  return;
+}
+
+
+int main() {
+  ios::sync_with_stdio(false);
+  cin.tie(0);
+
+  // filesystem::path filepath =
+  //     filesystem::current_path().parent_path().parent_path() / "data" /
+  //     "1.in";
+  // ifstream file(filepath);
+  //   cin.rdbuf(file.rdbuf());  // redirect cin to file
+
+  int tt;
+  cin >> tt;
+  while (tt--) {
+    solve();
+  }
+  return 0;
+}

problems/codeforces1789A/cpp/tests/BUILD

@@ -0,0 +1,11 @@
+load("@rules_cc//cc:defs.bzl", "cc_binary", "cc_test")
+
+cc_test(
+    name = "solution_test",
+    srcs = ["solution_test.cpp"],
+    deps = [
+      "//problems/codeforces2A/src/main:solution",
+      "@com_google_googletest//:gtest_main",
+    ],
+    data = ["data"]
+)