change testcase

rainbou-kpr · Nov 10, 2023 · 8ccc0ab · 8ccc0ab
1 parent 3f55cfb
commit 8ccc0ab
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 27 deletions.
diff --git a/test/atcoder-abc126-d.test.cpp b/test/atcoder-abc126-d.test.cpp
diff --git a/test/atcoder-code_festival_2017_qualb_c.test.cpp b/test/atcoder-code_festival_2017_qualb_c.test.cpp
@@ -0,0 +1,22 @@
+#define PROBLEM "https://atcoder.jp/contests/code-festival-2017-qualb/tasks/code_festival_2017_qualb_c"
+
+#include <array>
+
+#include "../cpp/graph_util.hpp"
+
+int main() {
+    long long N, M;
+    std::cin >> N >> M;
+    Graph<int> graph(N);
+    graph.read(M, -1);
+    std::vector<int> color = bipartite_coloring(graph);
+
+    if (color.empty()) {
+        std::cout << N * (N - 1) / 2 - M << std::endl;
+    } else {
+        std::array<long long, 2> cnt = {0, 0};
+        cnt[0] = count(color.begin(), color.end(), 0);
+        cnt[1] = count(color.begin(), color.end(), 1);
+        std::cout << cnt[0] * cnt[1] - M << std::endl;
+    }
+}