Merge pull request #9406 from choroba/ech252

Solve 252: Special Numbers & Unique Sum Zero by E. Choroba

1 files changed, 41 insertions, 0 deletions

diff --git a/challenge-252/e-choroba/cpp/ch-2.cpp b/challenge-252/e-choroba/cpp/ch-2.cpp
new file mode 100644
index 0000000000..e10d27aae6
--- /dev/null
+++ b/challenge-252/e-choroba/cpp/ch-2.cpp
@@ -0,0 +1,41 @@
+#include <cassert>
+#include <iostream>
+#include <vector>
+
+std::vector<int> unique_sum_zero(int n) {
+    std::vector<int> v;
+    if (n % 2) {
+        v = {0};
+    }
+    for (int i = 1; i <= n / 2; ++i) {
+        v.insert(v.end(), {i, -i});
+    }
+    return v;
+}
+
+int main(int argc, char* argv[]) {
+
+    auto u1 = unique_sum_zero(1);
+    assert(u1.size() == 1);
+    assert(u1[0] == 0);
+
+    auto u2 = unique_sum_zero(2);
+    assert(u2.size() == 2);
+    assert(u2[0] == 1);
+    assert(u2[1] == -1);
+
+    auto u3 = unique_sum_zero(3);
+    assert(u3.size() == 3);
+    assert(u3[0] == 0);
+    assert(u3[1] == 1);
+    assert(u3[2] == -1);
+
+    auto u4 = unique_sum_zero(4);
+    assert(u4.size() == 4);
+    assert(u4[0] == 1);
+    assert(u4[1] == -1);
+    assert(u4[2] == 2);
+    assert(u4[3] == -2);
+
+    std::cout << "Ok." << std::endl;
+}