답안 #863222

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
863222	2023-10-19T20:06:43 Z	arashMLG	The short shank; Redemption (BOI21_prison)	C++17	100 / 100	353 ms	303784 KB

prison

#include<bits/stdc++.h>
#ifdef LOCAL
#include "Essentials/algo/debug.h"
#else
#define debug(...) 69
#endif
using namespace std;

typedef long long     ll;
typedef pair<int,int> pii;

const int N = 2e6 + 23;
const ll inf = 1e18;

#define F           first
#define S           second
#define pb          push_back
#define kill(x)     cout<<x<<endl, exit(0);
#define all(x)      x.begin(),x.end()
#define sz(x)       (int)x.size()
#define lc          (v << 1)
#define rc          ((v<<1) |1)
// https://oj.uz/problem/view/BOI21_prison

int n,d,T;
int a[N];
stack<int> st;
vector<int> G[N];
int h[N],par[N];
pii mh[N];
vector<int> hs[N];
int l[N];
int ans;

void calcl() {
	for(int i = 1; i<= n ; i++) {
		if(a[i] <= T) {
			l[i] =i;
			st.push(i);
			continue;;
		}
		debug(sz(st));
		while(sz(st) && (i - st.top()) + a[st.top()] > T) debug(st.top()),st.pop();
		if(sz(st)) l[i] = st.top();
		st.push(i);
	}
}

bool isroot[N];

void buildg() {
	ans = n;
	while(sz(st)) st.pop();
	for(int i = n; i>=1 ; i--) {
		debug(i,l[i]);
		if(a[i] <= T) continue;
		if(l[i] == 0) {
			ans--;
			continue;
		}
		while(sz(st) && l[st.top()] > i) st.pop();
		if(sz(st)) G[st.top()].pb(i);
		else isroot[i] = true;
		st.push(i);
	}
}

void dfs(int v) {
	mh[v] = {h[v],v};
	debug(v,G[v]);
	for(int u : G[v]) {
		h[u] = h[v] +1;
		par[u] = v;
		dfs(u);
		mh[v] = max(mh[v] ,mh[u]);
	}
}

bool mark[N];

int32_t main() {
    cin.tie(nullptr)->sync_with_stdio(false);
	cin>>n>>d>>T;
	for(int i = 1; i<= n ; i ++) cin>>a[i];
	calcl();
	buildg();
	for(int i = 1; i<= n ; i ++) {
		if(isroot[i]) {
			par[i] = i;
			debug(i);
			dfs(i);
			hs[mh[i].F+1].pb(i);
		}
	}
	for(int i = N-1; i >= 0; i --) {
		for(int sex : hs[i]) {
			if(d == 0) {
				cout<<ans << '\n';
				return 0;
			}
			ans -= i;
			d--;
			int x = mh[sex].S;
			while(!mark[x]) {
				mark[x] = true;
				for(int u : G[x]) if(!mark[u]) {
					hs[mh[u].F-h[u]+1].pb(u);
				}
				x= par[x];
			}
		}
	}
	cout<<ans<<'\n';
	return 0;
}

Compilation message

prison.cpp: In function 'void calcl()':
prison.cpp:5:20: warning: statement has no effect [-Wunused-value]
    5 | #define debug(...) 69
      |                    ^~
prison.cpp:42:3: note: in expansion of macro 'debug'
   42 |   debug(sz(st));
      |   ^~~~~
prison.cpp:5:20: warning: left operand of comma operator has no effect [-Wunused-value]
    5 | #define debug(...) 69
      |                    ^~
prison.cpp:43:53: note: in expansion of macro 'debug'
   43 |   while(sz(st) && (i - st.top()) + a[st.top()] > T) debug(st.top()),st.pop();
      |                                                     ^~~~~
prison.cpp: In function 'void buildg()':
prison.cpp:5:20: warning: statement has no effect [-Wunused-value]
    5 | #define debug(...) 69
      |                    ^~
prison.cpp:55:3: note: in expansion of macro 'debug'
   55 |   debug(i,l[i]);
      |   ^~~~~
prison.cpp: In function 'void dfs(int)':
prison.cpp:5:20: warning: statement has no effect [-Wunused-value]
    5 | #define debug(...) 69
      |                    ^~
prison.cpp:70:2: note: in expansion of macro 'debug'
   70 |  debug(v,G[v]);
      |  ^~~~~
prison.cpp: In function 'int32_t main()':
prison.cpp:5:20: warning: statement has no effect [-Wunused-value]
    5 | #define debug(...) 69
      |                    ^~
prison.cpp:90:4: note: in expansion of macro 'debug'
   90 |    debug(i);
      |    ^~~~~

Subtask #1
15.0 / 15.0

#	결과	실행 시간	메모리	Grader output
1	Correct	26 ms	105048 KB	Output is correct
2	Correct	26 ms	105048 KB	Output is correct
3	Correct	26 ms	105052 KB	Output is correct
4	Correct	26 ms	105052 KB	Output is correct
5	Correct	26 ms	105052 KB	Output is correct
6	Correct	26 ms	105052 KB	Output is correct
7	Correct	26 ms	105052 KB	Output is correct
8	Correct	26 ms	105052 KB	Output is correct
9	Correct	26 ms	105136 KB	Output is correct
10	Correct	26 ms	105052 KB	Output is correct
11	Correct	26 ms	105052 KB	Output is correct

Subtask #2
10.0 / 10.0

#	결과	실행 시간	메모리	Grader output
1	Correct	25 ms	105048 KB	Output is correct
2	Correct	82 ms	124952 KB	Output is correct
3	Correct	78 ms	122764 KB	Output is correct
4	Correct	79 ms	126944 KB	Output is correct
5	Correct	80 ms	126544 KB	Output is correct
6	Correct	79 ms	127220 KB	Output is correct
7	Correct	118 ms	176780 KB	Output is correct
8	Correct	83 ms	130776 KB	Output is correct

Subtask #3
20.0 / 20.0

#	결과	실행 시간	메모리	Grader output
1	Correct	26 ms	105048 KB	Output is correct
2	Correct	26 ms	105048 KB	Output is correct
3	Correct	26 ms	105052 KB	Output is correct
4	Correct	26 ms	105052 KB	Output is correct
5	Correct	26 ms	105052 KB	Output is correct
6	Correct	26 ms	105052 KB	Output is correct
7	Correct	26 ms	105052 KB	Output is correct
8	Correct	26 ms	105052 KB	Output is correct
9	Correct	26 ms	105136 KB	Output is correct
10	Correct	26 ms	105052 KB	Output is correct
11	Correct	26 ms	105052 KB	Output is correct
12	Correct	25 ms	105048 KB	Output is correct
13	Correct	26 ms	105048 KB	Output is correct
14	Correct	26 ms	105120 KB	Output is correct
15	Correct	26 ms	105052 KB	Output is correct
16	Correct	26 ms	105064 KB	Output is correct
17	Correct	26 ms	105052 KB	Output is correct
18	Correct	26 ms	105052 KB	Output is correct
19	Correct	26 ms	105052 KB	Output is correct
20	Correct	26 ms	105304 KB	Output is correct
21	Correct	26 ms	105052 KB	Output is correct
22	Correct	27 ms	105132 KB	Output is correct
23	Correct	27 ms	105308 KB	Output is correct
24	Correct	26 ms	105052 KB	Output is correct
25	Correct	27 ms	105020 KB	Output is correct
26	Correct	26 ms	105052 KB	Output is correct
27	Correct	26 ms	105052 KB	Output is correct
28	Correct	26 ms	105052 KB	Output is correct
29	Correct	26 ms	105052 KB	Output is correct
30	Correct	26 ms	105052 KB	Output is correct
31	Correct	26 ms	105020 KB	Output is correct
32	Correct	27 ms	105052 KB	Output is correct
33	Correct	26 ms	105052 KB	Output is correct
34	Correct	26 ms	105220 KB	Output is correct
35	Correct	27 ms	105052 KB	Output is correct

Subtask #4
10.0 / 10.0

#	결과	실행 시간	메모리	Grader output
1	Correct	26 ms	105052 KB	Output is correct
2	Correct	35 ms	106844 KB	Output is correct
3	Correct	35 ms	106588 KB	Output is correct
4	Correct	37 ms	111956 KB	Output is correct
5	Correct	42 ms	111444 KB	Output is correct
6	Correct	37 ms	111444 KB	Output is correct
7	Correct	33 ms	109136 KB	Output is correct
8	Correct	33 ms	106832 KB	Output is correct
9	Correct	39 ms	116228 KB	Output is correct
10	Correct	32 ms	105820 KB	Output is correct

Subtask #5
25.0 / 25.0

#	결과	실행 시간	메모리	Grader output
1	Correct	26 ms	105048 KB	Output is correct
2	Correct	26 ms	105048 KB	Output is correct
3	Correct	26 ms	105052 KB	Output is correct
4	Correct	26 ms	105052 KB	Output is correct
5	Correct	26 ms	105052 KB	Output is correct
6	Correct	26 ms	105052 KB	Output is correct
7	Correct	26 ms	105052 KB	Output is correct
8	Correct	26 ms	105052 KB	Output is correct
9	Correct	26 ms	105136 KB	Output is correct
10	Correct	26 ms	105052 KB	Output is correct
11	Correct	26 ms	105052 KB	Output is correct
12	Correct	25 ms	105048 KB	Output is correct
13	Correct	26 ms	105048 KB	Output is correct
14	Correct	26 ms	105120 KB	Output is correct
15	Correct	26 ms	105052 KB	Output is correct
16	Correct	26 ms	105064 KB	Output is correct
17	Correct	26 ms	105052 KB	Output is correct
18	Correct	26 ms	105052 KB	Output is correct
19	Correct	26 ms	105052 KB	Output is correct
20	Correct	26 ms	105304 KB	Output is correct
21	Correct	26 ms	105052 KB	Output is correct
22	Correct	27 ms	105132 KB	Output is correct
23	Correct	27 ms	105308 KB	Output is correct
24	Correct	26 ms	105052 KB	Output is correct
25	Correct	27 ms	105020 KB	Output is correct
26	Correct	26 ms	105052 KB	Output is correct
27	Correct	26 ms	105052 KB	Output is correct
28	Correct	26 ms	105052 KB	Output is correct
29	Correct	26 ms	105052 KB	Output is correct
30	Correct	26 ms	105052 KB	Output is correct
31	Correct	26 ms	105020 KB	Output is correct
32	Correct	27 ms	105052 KB	Output is correct
33	Correct	26 ms	105052 KB	Output is correct
34	Correct	26 ms	105220 KB	Output is correct
35	Correct	27 ms	105052 KB	Output is correct
36	Correct	26 ms	105052 KB	Output is correct
37	Correct	35 ms	106844 KB	Output is correct
38	Correct	35 ms	106588 KB	Output is correct
39	Correct	37 ms	111956 KB	Output is correct
40	Correct	42 ms	111444 KB	Output is correct
41	Correct	37 ms	111444 KB	Output is correct
42	Correct	33 ms	109136 KB	Output is correct
43	Correct	33 ms	106832 KB	Output is correct
44	Correct	39 ms	116228 KB	Output is correct
45	Correct	32 ms	105820 KB	Output is correct
46	Correct	26 ms	105208 KB	Output is correct
47	Correct	26 ms	105052 KB	Output is correct
48	Correct	26 ms	105108 KB	Output is correct
49	Correct	26 ms	105052 KB	Output is correct
50	Correct	26 ms	105052 KB	Output is correct
51	Correct	26 ms	104940 KB	Output is correct
52	Correct	25 ms	105048 KB	Output is correct
53	Correct	26 ms	105104 KB	Output is correct
54	Correct	26 ms	105052 KB	Output is correct
55	Correct	25 ms	105092 KB	Output is correct
56	Correct	26 ms	105052 KB	Output is correct
57	Correct	26 ms	105052 KB	Output is correct
58	Correct	27 ms	105052 KB	Output is correct
59	Correct	26 ms	105052 KB	Output is correct
60	Correct	27 ms	105052 KB	Output is correct
61	Correct	26 ms	105052 KB	Output is correct
62	Correct	26 ms	105040 KB	Output is correct
63	Correct	26 ms	105044 KB	Output is correct
64	Correct	26 ms	105208 KB	Output is correct
65	Correct	26 ms	105136 KB	Output is correct
66	Correct	26 ms	105052 KB	Output is correct
67	Correct	26 ms	105052 KB	Output is correct
68	Correct	26 ms	105052 KB	Output is correct
69	Correct	26 ms	105048 KB	Output is correct
70	Correct	25 ms	105048 KB	Output is correct
71	Correct	35 ms	106836 KB	Output is correct
72	Correct	34 ms	106580 KB	Output is correct
73	Correct	36 ms	111952 KB	Output is correct
74	Correct	37 ms	111448 KB	Output is correct
75	Correct	37 ms	111472 KB	Output is correct
76	Correct	33 ms	109140 KB	Output is correct
77	Correct	33 ms	106960 KB	Output is correct
78	Correct	38 ms	116316 KB	Output is correct
79	Correct	32 ms	105812 KB	Output is correct
80	Correct	35 ms	106488 KB	Output is correct
81	Correct	35 ms	106988 KB	Output is correct
82	Correct	37 ms	106524 KB	Output is correct
83	Correct	36 ms	108376 KB	Output is correct
84	Correct	33 ms	106848 KB	Output is correct
85	Correct	38 ms	113756 KB	Output is correct
86	Correct	35 ms	110160 KB	Output is correct
87	Correct	34 ms	106548 KB	Output is correct
88	Correct	41 ms	114892 KB	Output is correct
89	Correct	38 ms	115816 KB	Output is correct
90	Correct	35 ms	113296 KB	Output is correct

Subtask #6
20.0 / 20.0

#	결과	실행 시간	메모리	Grader output
1	Correct	26 ms	105048 KB	Output is correct
2	Correct	26 ms	105048 KB	Output is correct
3	Correct	26 ms	105052 KB	Output is correct
4	Correct	26 ms	105052 KB	Output is correct
5	Correct	26 ms	105052 KB	Output is correct
6	Correct	26 ms	105052 KB	Output is correct
7	Correct	26 ms	105052 KB	Output is correct
8	Correct	26 ms	105052 KB	Output is correct
9	Correct	26 ms	105136 KB	Output is correct
10	Correct	26 ms	105052 KB	Output is correct
11	Correct	26 ms	105052 KB	Output is correct
12	Correct	25 ms	105048 KB	Output is correct
13	Correct	82 ms	124952 KB	Output is correct
14	Correct	78 ms	122764 KB	Output is correct
15	Correct	79 ms	126944 KB	Output is correct
16	Correct	80 ms	126544 KB	Output is correct
17	Correct	79 ms	127220 KB	Output is correct
18	Correct	118 ms	176780 KB	Output is correct
19	Correct	83 ms	130776 KB	Output is correct
20	Correct	25 ms	105048 KB	Output is correct
21	Correct	26 ms	105048 KB	Output is correct
22	Correct	26 ms	105120 KB	Output is correct
23	Correct	26 ms	105052 KB	Output is correct
24	Correct	26 ms	105064 KB	Output is correct
25	Correct	26 ms	105052 KB	Output is correct
26	Correct	26 ms	105052 KB	Output is correct
27	Correct	26 ms	105052 KB	Output is correct
28	Correct	26 ms	105304 KB	Output is correct
29	Correct	26 ms	105052 KB	Output is correct
30	Correct	27 ms	105132 KB	Output is correct
31	Correct	27 ms	105308 KB	Output is correct
32	Correct	26 ms	105052 KB	Output is correct
33	Correct	27 ms	105020 KB	Output is correct
34	Correct	26 ms	105052 KB	Output is correct
35	Correct	26 ms	105052 KB	Output is correct
36	Correct	26 ms	105052 KB	Output is correct
37	Correct	26 ms	105052 KB	Output is correct
38	Correct	26 ms	105052 KB	Output is correct
39	Correct	26 ms	105020 KB	Output is correct
40	Correct	27 ms	105052 KB	Output is correct
41	Correct	26 ms	105052 KB	Output is correct
42	Correct	26 ms	105220 KB	Output is correct
43	Correct	27 ms	105052 KB	Output is correct
44	Correct	26 ms	105052 KB	Output is correct
45	Correct	35 ms	106844 KB	Output is correct
46	Correct	35 ms	106588 KB	Output is correct
47	Correct	37 ms	111956 KB	Output is correct
48	Correct	42 ms	111444 KB	Output is correct
49	Correct	37 ms	111444 KB	Output is correct
50	Correct	33 ms	109136 KB	Output is correct
51	Correct	33 ms	106832 KB	Output is correct
52	Correct	39 ms	116228 KB	Output is correct
53	Correct	32 ms	105820 KB	Output is correct
54	Correct	26 ms	105208 KB	Output is correct
55	Correct	26 ms	105052 KB	Output is correct
56	Correct	26 ms	105108 KB	Output is correct
57	Correct	26 ms	105052 KB	Output is correct
58	Correct	26 ms	105052 KB	Output is correct
59	Correct	26 ms	104940 KB	Output is correct
60	Correct	25 ms	105048 KB	Output is correct
61	Correct	26 ms	105104 KB	Output is correct
62	Correct	26 ms	105052 KB	Output is correct
63	Correct	25 ms	105092 KB	Output is correct
64	Correct	26 ms	105052 KB	Output is correct
65	Correct	26 ms	105052 KB	Output is correct
66	Correct	27 ms	105052 KB	Output is correct
67	Correct	26 ms	105052 KB	Output is correct
68	Correct	27 ms	105052 KB	Output is correct
69	Correct	26 ms	105052 KB	Output is correct
70	Correct	26 ms	105040 KB	Output is correct
71	Correct	26 ms	105044 KB	Output is correct
72	Correct	26 ms	105208 KB	Output is correct
73	Correct	26 ms	105136 KB	Output is correct
74	Correct	26 ms	105052 KB	Output is correct
75	Correct	26 ms	105052 KB	Output is correct
76	Correct	26 ms	105052 KB	Output is correct
77	Correct	26 ms	105048 KB	Output is correct
78	Correct	25 ms	105048 KB	Output is correct
79	Correct	35 ms	106836 KB	Output is correct
80	Correct	34 ms	106580 KB	Output is correct
81	Correct	36 ms	111952 KB	Output is correct
82	Correct	37 ms	111448 KB	Output is correct
83	Correct	37 ms	111472 KB	Output is correct
84	Correct	33 ms	109140 KB	Output is correct
85	Correct	33 ms	106960 KB	Output is correct
86	Correct	38 ms	116316 KB	Output is correct
87	Correct	32 ms	105812 KB	Output is correct
88	Correct	35 ms	106488 KB	Output is correct
89	Correct	35 ms	106988 KB	Output is correct
90	Correct	37 ms	106524 KB	Output is correct
91	Correct	36 ms	108376 KB	Output is correct
92	Correct	33 ms	106848 KB	Output is correct
93	Correct	38 ms	113756 KB	Output is correct
94	Correct	35 ms	110160 KB	Output is correct
95	Correct	34 ms	106548 KB	Output is correct
96	Correct	41 ms	114892 KB	Output is correct
97	Correct	38 ms	115816 KB	Output is correct
98	Correct	35 ms	113296 KB	Output is correct
99	Correct	26 ms	105116 KB	Output is correct
100	Correct	26 ms	105052 KB	Output is correct
101	Correct	26 ms	105064 KB	Output is correct
102	Correct	27 ms	105052 KB	Output is correct
103	Correct	27 ms	105052 KB	Output is correct
104	Correct	26 ms	105052 KB	Output is correct
105	Correct	26 ms	105104 KB	Output is correct
106	Correct	25 ms	105052 KB	Output is correct
107	Correct	25 ms	105048 KB	Output is correct
108	Correct	26 ms	105112 KB	Output is correct
109	Correct	26 ms	105052 KB	Output is correct
110	Correct	26 ms	105052 KB	Output is correct
111	Correct	81 ms	124712 KB	Output is correct
112	Correct	79 ms	122684 KB	Output is correct
113	Correct	78 ms	126844 KB	Output is correct
114	Correct	78 ms	126700 KB	Output is correct
115	Correct	78 ms	127184 KB	Output is correct
116	Correct	118 ms	176832 KB	Output is correct
117	Correct	81 ms	130808 KB	Output is correct
118	Correct	27 ms	105368 KB	Output is correct
119	Correct	26 ms	105052 KB	Output is correct
120	Correct	26 ms	105052 KB	Output is correct
121	Correct	27 ms	105048 KB	Output is correct
122	Correct	26 ms	105052 KB	Output is correct
123	Correct	26 ms	105048 KB	Output is correct
124	Correct	26 ms	105052 KB	Output is correct
125	Correct	26 ms	105052 KB	Output is correct
126	Correct	26 ms	105052 KB	Output is correct
127	Correct	26 ms	105052 KB	Output is correct
128	Correct	27 ms	105052 KB	Output is correct
129	Correct	27 ms	105052 KB	Output is correct
130	Correct	26 ms	105052 KB	Output is correct
131	Correct	26 ms	105052 KB	Output is correct
132	Correct	35 ms	106836 KB	Output is correct
133	Correct	35 ms	106588 KB	Output is correct
134	Correct	37 ms	111956 KB	Output is correct
135	Correct	37 ms	111896 KB	Output is correct
136	Correct	38 ms	111444 KB	Output is correct
137	Correct	35 ms	109144 KB	Output is correct
138	Correct	33 ms	106844 KB	Output is correct
139	Correct	38 ms	116476 KB	Output is correct
140	Correct	32 ms	105820 KB	Output is correct
141	Correct	34 ms	106588 KB	Output is correct
142	Correct	35 ms	106832 KB	Output is correct
143	Correct	34 ms	106708 KB	Output is correct
144	Correct	36 ms	108372 KB	Output is correct
145	Correct	34 ms	106844 KB	Output is correct
146	Correct	38 ms	113756 KB	Output is correct
147	Correct	37 ms	110424 KB	Output is correct
148	Correct	33 ms	106584 KB	Output is correct
149	Correct	37 ms	114768 KB	Output is correct
150	Correct	37 ms	115868 KB	Output is correct
151	Correct	36 ms	113236 KB	Output is correct
152	Correct	252 ms	173704 KB	Output is correct
153	Correct	255 ms	166464 KB	Output is correct
154	Correct	246 ms	161220 KB	Output is correct
155	Correct	301 ms	208988 KB	Output is correct
156	Correct	249 ms	177904 KB	Output is correct
157	Correct	223 ms	145096 KB	Output is correct
158	Correct	253 ms	187080 KB	Output is correct
159	Correct	353 ms	303784 KB	Output is correct
160	Correct	343 ms	299980 KB	Output is correct
161	Correct	200 ms	161892 KB	Output is correct
162	Correct	226 ms	169664 KB	Output is correct

답안 #863222

Compilation message

Subtask #1 15.0 / 15.0

Subtask #2 10.0 / 10.0

Subtask #3 20.0 / 20.0

Subtask #4 10.0 / 10.0

Subtask #5 25.0 / 25.0

Subtask #6 20.0 / 20.0

Subtask #1
15.0 / 15.0

Subtask #2
10.0 / 10.0

Subtask #3
20.0 / 20.0

Subtask #4
10.0 / 10.0

Subtask #5
25.0 / 25.0

Subtask #6
20.0 / 20.0