From b79a55ceee2f4532ef6621f9d86c6c61543ce6c9 Mon Sep 17 00:00:00 2001
From: omagdy7 <omar.professional8777@gmail.com>
Date: Thu, 3 Nov 2022 19:21:07 +0200
Subject: Added some problems

---
 cses/IntroductoryProblems/TwoSets/main.cpp | 41 ++++++++++++++++++++++++++++++
 1 file changed, 41 insertions(+)
 create mode 100644 cses/IntroductoryProblems/TwoSets/main.cpp

(limited to 'cses/IntroductoryProblems/TwoSets/main.cpp')

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 << ' ';
+      }
+    }
+  }
+}
-- 
cgit v1.2.3