From ff7585043c5f48cf94bc7ace4ad1b8fe9b4df9ef Mon Sep 17 00:00:00 2001 From: omagdy7 Date: Sat, 13 Apr 2024 16:15:35 +0200 Subject: Solved 4 problems in last Div.3 --- contests/Round938/C/main.cpp | 109 +++++++++++++++++++++++++++++++++++ contests/Round938/C/main_input0.txt | 13 +++++ contests/Round938/C/main_input1.txt | 3 + contests/Round938/C/main_input2.txt | 3 + contests/Round938/C/main_input3.txt | 3 + contests/Round938/C/main_input4.txt | 3 + contests/Round938/C/main_input5.txt | 3 + contests/Round938/C/main_output0.txt | 6 ++ contests/Round938/C/main_output1.txt | 1 + contests/Round938/C/main_output2.txt | 1 + contests/Round938/C/main_output3.txt | 1 + contests/Round938/C/main_output4.txt | 1 + contests/Round938/C/main_output5.txt | 1 + 13 files changed, 148 insertions(+) create mode 100755 contests/Round938/C/main.cpp create mode 100644 contests/Round938/C/main_input0.txt create mode 100644 contests/Round938/C/main_input1.txt create mode 100644 contests/Round938/C/main_input2.txt create mode 100644 contests/Round938/C/main_input3.txt create mode 100644 contests/Round938/C/main_input4.txt create mode 100644 contests/Round938/C/main_input5.txt create mode 100644 contests/Round938/C/main_output0.txt create mode 100644 contests/Round938/C/main_output1.txt create mode 100644 contests/Round938/C/main_output2.txt create mode 100644 contests/Round938/C/main_output3.txt create mode 100644 contests/Round938/C/main_output4.txt create mode 100644 contests/Round938/C/main_output5.txt (limited to 'contests/Round938/C') diff --git a/contests/Round938/C/main.cpp b/contests/Round938/C/main.cpp new file mode 100755 index 0000000..3ce56ff --- /dev/null +++ b/contests/Round938/C/main.cpp @@ -0,0 +1,109 @@ +#include +#include + +using namespace std; + +using ll = long long; +using pii = pair; +using vpii = vector; +using vi = vector; +using vll = vector; +using mpii = map; +using mpll = map; +using db = long double; + +#define pb push_back +#define all(x) (x).begin(), (x).end() +#define rall(x) (x).rbegin(), (x).rend() +#define lb lower_bound +#define ub upper_bound +#define make_unique(x) \ + sort(all((x))); \ + (x).resize(unique(all((x))) - (x).begin()) +#define ceil(a, b) ((a) + (b)-1) / (b) + +const int mod = (int)1e9 + 7; +const db pi = acos((db)-1); +const int dx[4]{1, 0, -1, 0}; +const int dy[4]{0, 1, 0, -1}; + +#ifdef LOCAL +#include "debug.h" +#else +#define dbg(...) 42 +#endif + +/* stuff you should look for: + --------------------------- + * special cases (n=1?) + * int overflow, array bounds + * do smth instead of nothing and stay organized + * WRITE STUFF DOWN + * DON'T GET STUCK ON ONE APPROACH + */ + +// 11 +// 6 4 2 + +void solve() { + int n; + ll k; + cin >> n >> k; + vi v(n); + ll sum = 0; + for (auto &x : v) { + cin >> x; + sum += x; + } + vi tmp = v; + reverse(all(tmp)); + vll prefix(n + 1); + vll suffix(n + 1); + prefix[0] = 0; + suffix[0] = 0; + for (int i = 1; i < n + 1; i++) { + prefix[i] = prefix[i - 1] + v[i - 1]; + suffix[i] = suffix[i - 1] + tmp[i - 1]; + } + if (k >= sum) { + cout << n << '\n'; + } else { + int ans = 0; + if (k % 2 == 0) { + ll first = k / 2; + ll last = k / 2; + auto it_first = lower_bound(all(prefix), first); + auto it_last = lower_bound(all(suffix), last); + ans = (it_first - prefix.begin()) + (it_last - suffix.begin()); + if (*it_first > first) { + ans--; + } + if (*it_last > last) { + ans--; + } + } else { + ll first = (k / 2) + 1; + ll last = k / 2; + auto it_first = lower_bound(all(prefix), first); + auto it_last = lower_bound(all(suffix), last); + ans = (it_first - prefix.begin()) + (it_last - suffix.begin()); + if (*it_first > first) { + ans--; + } + if (*it_last > last) { + ans--; + } + } + cout << ans << '\n'; + } +} + +int main() { + ios_base::sync_with_stdio(false); + cin.tie(NULL); + int tt; + cin >> tt; + while (tt--) { + solve(); + } +} diff --git a/contests/Round938/C/main_input0.txt b/contests/Round938/C/main_input0.txt new file mode 100644 index 0000000..b411a1f --- /dev/null +++ b/contests/Round938/C/main_input0.txt @@ -0,0 +1,13 @@ +6 +4 5 +1 2 4 3 +4 6 +1 2 4 3 +5 20 +2 7 1 8 2 +2 2 +3 2 +2 15 +1 5 +2 7 +5 2 diff --git a/contests/Round938/C/main_input1.txt b/contests/Round938/C/main_input1.txt new file mode 100644 index 0000000..6c38806 --- /dev/null +++ b/contests/Round938/C/main_input1.txt @@ -0,0 +1,3 @@ +1 +5 11 +5 1 2 3 2 \ No newline at end of file diff --git a/contests/Round938/C/main_input2.txt b/contests/Round938/C/main_input2.txt new file mode 100644 index 0000000..dd544a9 --- /dev/null +++ b/contests/Round938/C/main_input2.txt @@ -0,0 +1,3 @@ +1 +3 5 +1 2 2 1 \ No newline at end of file diff --git a/contests/Round938/C/main_input3.txt b/contests/Round938/C/main_input3.txt new file mode 100644 index 0000000..eef0104 --- /dev/null +++ b/contests/Round938/C/main_input3.txt @@ -0,0 +1,3 @@ +1 +4 6 +1 3 1 1 \ No newline at end of file diff --git a/contests/Round938/C/main_input4.txt b/contests/Round938/C/main_input4.txt new file mode 100644 index 0000000..60c0ced --- /dev/null +++ b/contests/Round938/C/main_input4.txt @@ -0,0 +1,3 @@ +1 +1 1 +1 \ No newline at end of file diff --git a/contests/Round938/C/main_input5.txt b/contests/Round938/C/main_input5.txt new file mode 100644 index 0000000..4d01be5 --- /dev/null +++ b/contests/Round938/C/main_input5.txt @@ -0,0 +1,3 @@ +1 +2 1 +1 1 \ No newline at end of file diff --git a/contests/Round938/C/main_output0.txt b/contests/Round938/C/main_output0.txt new file mode 100644 index 0000000..8a5cc99 --- /dev/null +++ b/contests/Round938/C/main_output0.txt @@ -0,0 +1,6 @@ +2 +3 +5 +0 +2 +2 diff --git a/contests/Round938/C/main_output1.txt b/contests/Round938/C/main_output1.txt new file mode 100644 index 0000000..bf0d87a --- /dev/null +++ b/contests/Round938/C/main_output1.txt @@ -0,0 +1 @@ +4 \ No newline at end of file diff --git a/contests/Round938/C/main_output2.txt b/contests/Round938/C/main_output2.txt new file mode 100644 index 0000000..e440e5c --- /dev/null +++ b/contests/Round938/C/main_output2.txt @@ -0,0 +1 @@ +3 \ No newline at end of file diff --git a/contests/Round938/C/main_output3.txt b/contests/Round938/C/main_output3.txt new file mode 100644 index 0000000..bf0d87a --- /dev/null +++ b/contests/Round938/C/main_output3.txt @@ -0,0 +1 @@ +4 \ No newline at end of file diff --git a/contests/Round938/C/main_output4.txt b/contests/Round938/C/main_output4.txt new file mode 100644 index 0000000..56a6051 --- /dev/null +++ b/contests/Round938/C/main_output4.txt @@ -0,0 +1 @@ +1 \ No newline at end of file diff --git a/contests/Round938/C/main_output5.txt b/contests/Round938/C/main_output5.txt new file mode 100644 index 0000000..56a6051 --- /dev/null +++ b/contests/Round938/C/main_output5.txt @@ -0,0 +1 @@ +1 \ No newline at end of file -- cgit v1.2.3