CSES - Aalto Competitive Programming 2025 - wk1 - Wed

Submission details
Task:	Aquarium
Sender:	aalto25a_001
Submission time:	2025-09-03 17:44:34 +0300
Language:	C++ (C++17)
Status:	READY
Result:	ACCEPTED

Test results
test	verdict
#1	ACCEPTED	details
#2	ACCEPTED	details
#3	ACCEPTED	details
#4	ACCEPTED	details
#5	ACCEPTED	details
#6	ACCEPTED	details
#7	ACCEPTED	details
#8	ACCEPTED	details
#9	ACCEPTED	details
#10	ACCEPTED	details
#11	ACCEPTED	details
#12	ACCEPTED	details
#13	ACCEPTED	details
#14	ACCEPTED	details
#15	ACCEPTED	details
#16	ACCEPTED	details
#17	ACCEPTED	details
#18	ACCEPTED	details
#19	ACCEPTED	details
#20	ACCEPTED	details
#21	ACCEPTED	details
#22	ACCEPTED	details
#23	ACCEPTED	details
#24	ACCEPTED	details
#25	ACCEPTED	details
#26	ACCEPTED	details
#27	ACCEPTED	details
#28	ACCEPTED	details
#29	ACCEPTED	details
#30	ACCEPTED	details
#31	ACCEPTED	details
#32	ACCEPTED	details
#33	ACCEPTED	details
#34	ACCEPTED	details
#35	ACCEPTED	details
#36	ACCEPTED	details
#37	ACCEPTED	details
#38	ACCEPTED	details
#39	ACCEPTED	details
#40	ACCEPTED	details
#41	ACCEPTED	details
#42	ACCEPTED	details
#43	ACCEPTED	details
#44	ACCEPTED	details

Code

#include <iostream>
#include <stack>
#include <iomanip>

using namespace std;

#define int long long

const int N = 1e4 + 5;

int n;
int h[N], l[N], r[N];
int q;
long long V;

signed main() {
  ios::sync_with_stdio(false);
  cin.tie(0);
  cin >> n;
  for (int i = 1; i <= n; i++) {
    cin >> h[i];
  }
  cin >> q >> V;
  stack<int> st;
  for (int i = 1; i <= n; i++) {
    while (!st.empty() && h[i] >= h[st.top()]) {
      st.pop();
    }
    l[i] = (st.empty() ? 0 : st.top());
    st.push(i);
  }
  while (!st.empty()) {
    st.pop();
  }
  for (int i = n; i >= 1; i--) {
    while (!st.empty() && h[i] >= h[st.top()]) {
      st.pop();
    }
    r[i] = (st.empty() ? n + 1 : st.top());
    st.push(i);
  }
  if (V <= h[q]) {
    cout << V << '\n';
    return 0;
  }
  V -= h[q];
  int cur = h[q];
  int lf = q, rt = q;
  long double ans = cur;
  while (true) {
    int old_l = lf, old_r = rt;
    lf = l[lf];
    rt = r[rt];
    if (lf == 0 || rt == n + 1) {
      break;
    }
    long long nd = (rt - 1 - (lf + 1) + 1) - (old_r - old_l + 1);
    nd *= cur;
    if (V <= nd) {
      break;
    } else {
      V -= nd;
    }
    int new_height = min(h[lf], h[rt]);
    if (1LL * (new_height - cur) * ((rt - 1) - (lf + 1) + 1) > V) {
      ans += (long double)(V) / ((rt - 1) - (lf + 1) + 1);
      cout << fixed << setprecision(6) << ans << '\n';
      return 0;
    } else {
      V -= 1LL * (new_height - cur) * ((rt - 1) - (lf + 1) + 1);
      cur = new_height;
      if (h[lf] < h[rt]) {
        if (V < h[lf]) {
          break;
        } else {
          V -= h[lf];
          rt--;
          cur = h[lf];
        }
      } else if (h[rt] < h[lf]) {
        if (V < h[rt]) {
          break;
        } else {
          V -= h[rt];
          lf++;
          cur = h[rt];
        }
      } else {
        if (V < h[lf] + h[rt]) {
          break;
        } else {
          V -= (h[lf] + h[rt]);
          cur = h[lf];
        }
      }
    }
    ans = cur;
  }
  cout << fixed << setprecision(6) << cur << '\n';
  return 0;
}

Test details

Test 1

Verdict: ACCEPTED

input
`1 1 1 1` view save

correct output
`1` view save

user output
`1` view save

Test 2

Verdict: ACCEPTED

input
`2 1 2 1 1` view save

correct output
`1` view save

user output
`1` view save

Test 3

Verdict: ACCEPTED

input
`3 1 1 3 2 7` view save

correct output
`1` view save

user output
`1` view save

Test 4

Verdict: ACCEPTED

input
`5 1 1 5 3 5 3 6` view save

correct output
`5` view save

user output
`5` view save

Test 5

Verdict: ACCEPTED

input
`6 5 3 1 2 4 6 3 4` view save

correct output
`2` view save

user output
`2` view save

Test 6

Verdict: ACCEPTED

input
`8 7 5 3 1 2 4 6 8 4 7` view save

correct output
`3` view save

user output
`3` view save

Test 7

Verdict: ACCEPTED

input
`9 2 1 9 5 9 4 5 4 3 4 5` view save

correct output
`5` view save

user output
`5` view save

Test 8

Verdict: ACCEPTED

input
`10 9 7 5 3 1 2 4 6 8 10 5 11` view save

correct output
`3.66667` view save

user output
`3.666667` view save

Test 9

Verdict: ACCEPTED

input
`12 11 9 7 5 3 1 2 4 6 8 10 12 6 16` view save

correct output
`4` view save

user output
`4` view save

Test 10

Verdict: ACCEPTED

input
`14 13 11 9 7 5 3 1 2 4 6 8 10 12 ...` view save

correct output
`5` view save

user output
`5` view save

Test 11

Verdict: ACCEPTED

input
`16 15 13 11 9 7 5 3 1 2 4 6 8 10 ...` view save

correct output
`5.6` view save

user output
`5.600000` view save

Test 12

Verdict: ACCEPTED

input
`18 17 15 13 11 9 7 5 3 1 2 4 6 8 ...` view save

correct output
`6` view save

user output
`6` view save

Test 13

Verdict: ACCEPTED

input
`20 19 17 15 13 11 9 7 5 3 1 2 4 6...` view save

correct output
`7` view save

user output
`7` view save

Test 14

Verdict: ACCEPTED

input
`25 5 1 24 14 24 11 13 11 9 9 4 6 ...` view save

correct output
`6` view save

user output
`6` view save

Test 15

Verdict: ACCEPTED

input
`30 29 27 25 23 21 19 17 15 13 11 ...` view save

correct output
`10` view save

user output
`10` view save

Test 16

Verdict: ACCEPTED

input
`40 39 37 35 33 31 29 27 25 23 21 ...` view save

correct output
`13.4615` view save

user output
`13.461538` view save

Test 17

Verdict: ACCEPTED

input
`50 49 47 45 43 41 39 37 35 33 31 ...` view save

correct output
`17` view save

user output
`17` view save

Test 18

Verdict: ACCEPTED

input
`70 69 67 65 63 61 59 57 55 53 51 ...` view save

correct output
`23.2609` view save

user output
`23.260870` view save

Test 19

Verdict: ACCEPTED

input
`1 1 1 1` view save

correct output
`1` view save

user output
`1` view save

Test 20

Verdict: ACCEPTED

input
`2 3 2 1 16` view save

correct output
`3` view save

user output
`3` view save

Test 21

Verdict: ACCEPTED

input
`3 6 4 3 3 81` view save

correct output
`3` view save

user output
`3` view save

Test 22

Verdict: ACCEPTED

input
`5 14 8 5 24 16 5 625` view save

correct output
`16` view save

user output
`16` view save

Test 23

Verdict: ACCEPTED

input
`6 19 11 6 34 23 35 3 1296` view save

correct output
`19` view save

user output
`19` view save

Test 24

Verdict: ACCEPTED

input
`8 32 18 9 61 39 62 32 35 4 4096` view save

correct output
`61` view save

user output
`61` view save

Test 25

Verdict: ACCEPTED

input
`9 40 22 10 77 49 78 40 44 39 3 6561` view save

correct output
`40` view save

user output
`40` view save

Test 26

Verdict: ACCEPTED

input
`10 49 26 12 94 60 96 49 54 48 39 4 10000` view save

correct output
`94` view save

user output
`94` view save

Test 27

Verdict: ACCEPTED

input
`12 69 36 15 135 85 138 69 76 67 5...` view save

correct output
`69` view save

user output
`69` view save

Test 28

Verdict: ACCEPTED

input
`14 93 47 18 184 114 187 93 102 90...` view save

correct output
`141` view save

user output
`141` view save

Test 29

Verdict: ACCEPTED

input
`16 121 60 22 240 148 244 120 132 ...` view save

correct output
`240` view save

user output
`240` view save

Test 30

Verdict: ACCEPTED

input
`18 151 74 25 303 186 308 151 166 ...` view save

correct output
`303` view save

user output
`303` view save

Test 31

Verdict: ACCEPTED

input
`20 186 90 29 374 229 381 185 204 ...` view save

correct output
`286` view save

user output
`286` view save

Test 32

Verdict: ACCEPTED

input
`25 287 136 40 584 355 594 286 316...` view save

correct output
`516` view save

user output
`516` view save

Test 33

Verdict: ACCEPTED

input
`30 409 191 52 841 508 855 409 452...` view save

correct output
`656` view save

user output
`656` view save

Test 34

Verdict: ACCEPTED

input
`40 720 328 80 1494 898 1519 719 7...` view save

correct output
`1519` view save

user output
`1519` view save

Test 35

Verdict: ACCEPTED

input
`50 1118 503 113 2333 1397 2372 11...` view save

correct output
`2372` view save

user output
`2372` view save

Test 36

Verdict: ACCEPTED

input
`70 2176 964 195 4570 2725 4648 21...` view save

correct output
`4648` view save

user output
`4648` view save

Test 37

Verdict: ACCEPTED

input
`100 4416 1932 356 9323 5542 9483 4...` view save

correct output
`9700` view save

user output
`9700` view save

Test 38

Verdict: ACCEPTED

input
`500 104548 249301 180224 233177 52...` view save

correct output
`104548` view save

user output
`104548` view save

Test 39

Verdict: ACCEPTED

input
`600 150482 358999 259492 335771 64...` view save

correct output
`353835` view save

user output
`353835` view save

Test 40

Verdict: ACCEPTED

input
`900 446578 58127 573897 680544 236...` view save

correct output
`790626` view save

user output
`790626` view save

Test 41

Verdict: ACCEPTED

input
`900 783372 729637 443692 140636 78...` view save

correct output
`805592` view save

user output
`805592` view save

Test 42

Verdict: ACCEPTED

input
`100 4416 1932 356 9323 5542 9483 4...` view save

correct output
`9700` view save

user output
`9700` view save

Test 43

Verdict: ACCEPTED

input
`500 104548 249301 180224 233177 52...` view save

correct output
`104548` view save

user output
`104548` view save

Test 44

Verdict: ACCEPTED

input
`600 150482 358999 259492 335771 64...` view save

correct output
`353835` view save

user output
`353835` view save

Aalto Competitive Programming 2025 - wk1 - Wed

Aquarium

Code

Test details

Test 1

Test 2

Test 3

Test 4

Test 5

Test 6

Test 7

Test 8

Test 9

Test 10

Test 11

Test 12

Test 13

Test 14

Test 15

Test 16

Test 17

Test 18

Test 19

Test 20

Test 21

Test 22

Test 23

Test 24

Test 25

Test 26

Test 27

Test 28

Test 29

Test 30

Test 31

Test 32

Test 33

Test 34

Test 35

Test 36

Test 37

Test 38

Test 39

Test 40

Test 41

Test 42

Test 43

Test 44

Tasks

Submissions (aalto25a_001)