Added some problems

6 files changed, 49 insertions, 0 deletions

diff --git a/cses/IntroductoryProblems/TwoSets/main b/cses/IntroductoryProblems/TwoSets/main
new file mode 100755
index 0000000..ed99e71
--- /dev/null
+++ b/cses/IntroductoryProblems/TwoSets/main
diff --git a/cses/IntroductoryProblems/TwoSets/main.cpp b/cses/IntroductoryProblems/TwoSets/main.cpp
new file mode 100644
index 0000000..bec99d9
--- /dev/null
+++ b/cses/IntroductoryProblems/TwoSets/main.cpp
@@ -0,0 +1,41 @@
+#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;
+  if (sum & 1) {
+    cout << "NO" << '\n';
+  } else {
+    cout << "YES" << '\n';
+    vector<bool> a(n + 1);
+    for (int i = n; i >= 1; i--) {
+      if (s <= i) {
+        a[s] = true;
+        break;
+      }
+      s -= i;
+      a[i] = true;
+    }
+    int c1 = count(a.begin() + 1, a.end(), false);
+    cout << c1 << '\n';
+    for (int i = 1; i <= n; i++) {
+      if (a[i] == false) {
+        cout << i << ' ';
+      }
+    }
+    cout << '\n' << n - c1 << '\n';
+    for (int i = 1; i <= n; i++) {
+      if (a[i] == true) {
+        cout << i << ' ';
+      }
+    }
+  }
+}
diff --git a/cses/IntroductoryProblems/TwoSets/main_input0.txt b/cses/IntroductoryProblems/TwoSets/main_input0.txt
new file mode 100644
index 0000000..7f8f011
--- /dev/null
+++ b/cses/IntroductoryProblems/TwoSets/main_input0.txt
@@ -0,0 +1 @@
+7
diff --git a/cses/IntroductoryProblems/TwoSets/main_input1.txt b/cses/IntroductoryProblems/TwoSets/main_input1.txt
new file mode 100644
index 0000000..1e8b314
--- /dev/null
+++ b/cses/IntroductoryProblems/TwoSets/main_input1.txt
@@ -0,0 +1 @@
+6
diff --git a/cses/IntroductoryProblems/TwoSets/main_output0.txt b/cses/IntroductoryProblems/TwoSets/main_output0.txt
new file mode 100644
index 0000000..90d1101
--- /dev/null
+++ b/cses/IntroductoryProblems/TwoSets/main_output0.txt
@@ -0,0 +1,5 @@
+YES
+4
+1 2 4 7
+3
+3 5 6
diff --git a/cses/IntroductoryProblems/TwoSets/main_output1.txt b/cses/IntroductoryProblems/TwoSets/main_output1.txt
new file mode 100644
index 0000000..5e35d1b
--- /dev/null
+++ b/cses/IntroductoryProblems/TwoSets/main_output1.txt
@@ -0,0 +1 @@
+NO