solved 3 problems from cses

6 files changed, 54 insertions, 0 deletions

diff --git a/cses/TwoSets/main b/cses/TwoSets/main
new file mode 100755
index 0000000..61c6ec3
--- /dev/null
+++ b/cses/TwoSets/main
diff --git a/cses/TwoSets/main.cpp b/cses/TwoSets/main.cpp
new file mode 100644
index 0000000..8c1aed0
--- /dev/null
+++ b/cses/TwoSets/main.cpp
@@ -0,0 +1,46 @@
+#include<bits/stdc++.h>
+
+using namespace std;
+
+typedef long long ll;
+
+int main () {
+	ios_base::sync_with_stdio(false);
+	cin.tie(NULL);
+  ll n;
+  cin >> n;
+  ll sum = (n * (n + 1)) / 2;
+  ll s = sum / 2;
+  set<int> vis;
+  vector<int> st2;
+  if (sum & 1) {
+    cout << "NO" << '\n';
+  } else {
+    cout << "YES" << '\n';
+    for (int i = n; i >= 1; --i) {
+      if (s - i > 0) {
+        s -= i;
+        vis.insert(i);
+      } else {
+        vis.insert(s);
+        s -= i;
+        break;
+      }
+    }
+    for (int i = 1; i <= n; ++i) {
+      if (!vis.count(i)) {
+        st2.push_back(i);
+      }
+    }
+    cout << vis.size() << '\n';
+    for (auto x : vis) {
+      cout << x << " ";
+    }
+    cout << '\n';
+    cout << st2.size() << '\n';
+    for (auto x : st2) {
+      cout << x << " ";
+    }
+    cout << '\n';
+  }
+}
diff --git a/cses/TwoSets/main_input0.txt b/cses/TwoSets/main_input0.txt
new file mode 100644
index 0000000..7f8f011
--- /dev/null
+++ b/cses/TwoSets/main_input0.txt
@@ -0,0 +1 @@
+7
diff --git a/cses/TwoSets/main_input1.txt b/cses/TwoSets/main_input1.txt
new file mode 100644
index 0000000..1e8b314
--- /dev/null
+++ b/cses/TwoSets/main_input1.txt
@@ -0,0 +1 @@
+6
diff --git a/cses/TwoSets/main_output0.txt b/cses/TwoSets/main_output0.txt
new file mode 100644
index 0000000..90d1101
--- /dev/null
+++ b/cses/TwoSets/main_output0.txt
@@ -0,0 +1,5 @@
+YES
+4
+1 2 4 7
+3
+3 5 6
diff --git a/cses/TwoSets/main_output1.txt b/cses/TwoSets/main_output1.txt
new file mode 100644
index 0000000..5e35d1b
--- /dev/null
+++ b/cses/TwoSets/main_output1.txt
@@ -0,0 +1 @@
+NO