diff options
Diffstat (limited to 'contests/Round938/C')
| -rwxr-xr-x | contests/Round938/C/main.cpp | 109 | ||||
| -rw-r--r-- | contests/Round938/C/main_input0.txt | 13 | ||||
| -rw-r--r-- | contests/Round938/C/main_input1.txt | 3 | ||||
| -rw-r--r-- | contests/Round938/C/main_input2.txt | 3 | ||||
| -rw-r--r-- | contests/Round938/C/main_input3.txt | 3 | ||||
| -rw-r--r-- | contests/Round938/C/main_input4.txt | 3 | ||||
| -rw-r--r-- | contests/Round938/C/main_input5.txt | 3 | ||||
| -rw-r--r-- | contests/Round938/C/main_output0.txt | 6 | ||||
| -rw-r--r-- | contests/Round938/C/main_output1.txt | 1 | ||||
| -rw-r--r-- | contests/Round938/C/main_output2.txt | 1 | ||||
| -rw-r--r-- | contests/Round938/C/main_output3.txt | 1 | ||||
| -rw-r--r-- | contests/Round938/C/main_output4.txt | 1 | ||||
| -rw-r--r-- | contests/Round938/C/main_output5.txt | 1 |
13 files changed, 148 insertions, 0 deletions
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 <algorithm> +#include <bits/stdc++.h> + +using namespace std; + +using ll = long long; +using pii = pair<int, int>; +using vpii = vector<pii>; +using vi = vector<int>; +using vll = vector<long long>; +using mpii = map<int, int>; +using mpll = map<ll, ll>; +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 |
