#include <bits/stdc++.h>
using namespace std;
#define int long long
#define INF (int)1e18
#define f first
#define s second
const int N=2e5+69;
int ans=0;
mt19937_64 RNG(chrono::steady_clock::now().time_since_epoch().count());
class DisjointSets {
private:
vector<int> parents;
vector<int> sizes;
public:
DisjointSets(int size) : parents(size), sizes(size, 1) {
for (int i = 0; i < size; i++) { parents[i] = i; }
}
/** @return the "representative" node in x's component */
int find(int x) {
return parents[x] == x ? x : (parents[x] = find(parents[x]));
}
/** @return whether the merge changed connectivity */
bool unite(int x, int y) {
int x_root = find(x);
int y_root = find(y);
if (x_root == y_root)
{ return false;
}
ans+=sizes[x_root]*sizes[y_root];
if (sizes[x_root] < sizes[y_root]) { swap(x_root, y_root); }
sizes[x_root] += sizes[y_root];
parents[y_root] = x_root;
return true;
}
/** @return whether x and y are in the same connected component */
bool connected(int x, int y) { return find(x) == find(y); }
};
void Solve(){
int n,q;
cin>>n>>q;
int H[n];
int A[q];
map <int,int> mp;
for(int i=0;i<n;i++)
cin>>H[i];
for(int i=0;i<q;i++)
cin>>A[i];
vector <pair<int,int>> vec;
for(int i=0;i<n;i++)
vec.push_back({H[i],i});
DisjointSets ds(n);
sort(vec.begin(),vec.end());
set <int> st;
int cnt = 0;
for(auto [x,i]: vec){
if(i!=0&&H[i-1]<=H[i]){
ds.unite(i-1,i);
}
if(i!=n-1 && H[i+1]<=H[i])
ds.unite(i,i+1);
cnt++;
mp[x]=ans + cnt;
st.insert(x);
}
for(int i=0;i<q;i++){
// we need to find the largest number <= x
auto it = st.upper_bound(A[i]);
// this finds the first place with a number > x
// --it
if (it == st.begin()){
cout << 0 << "\n";
} else {
--it;
cout << mp[*it] << "\n";
}
}
}
int32_t main()
{
auto begin = std::chrono::high_resolution_clock::now();
ios_base::sync_with_stdio(0);
cin.tie(0);
int t = 1;
// freopen("in", "r", stdin);
// freopen("out", "w", stdout);
//cin >> t;
for(int i = 1; i <= t; i++)
{
//cout << "Case #" << i << ": ";
Solve();
}
auto end = std::chrono::high_resolution_clock::now();
auto elapsed = std::chrono::duration_cast<std::chrono::nanoseconds>(end - begin);
cerr << "Time measured: " << elapsed.count() * 1e-9 << " seconds.\n";
return 0;
}
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
348 KB |
Output is correct |
2 |
Correct |
0 ms |
348 KB |
Output is correct |
3 |
Correct |
0 ms |
348 KB |
Output is correct |
4 |
Correct |
1 ms |
348 KB |
Output is correct |
5 |
Correct |
0 ms |
348 KB |
Output is correct |
6 |
Correct |
0 ms |
348 KB |
Output is correct |
7 |
Correct |
0 ms |
348 KB |
Output is correct |
8 |
Correct |
0 ms |
348 KB |
Output is correct |
9 |
Correct |
0 ms |
344 KB |
Output is correct |
10 |
Correct |
0 ms |
348 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
348 KB |
Output is correct |
2 |
Correct |
0 ms |
348 KB |
Output is correct |
3 |
Correct |
0 ms |
348 KB |
Output is correct |
4 |
Correct |
1 ms |
348 KB |
Output is correct |
5 |
Correct |
0 ms |
348 KB |
Output is correct |
6 |
Correct |
0 ms |
348 KB |
Output is correct |
7 |
Correct |
0 ms |
348 KB |
Output is correct |
8 |
Correct |
0 ms |
348 KB |
Output is correct |
9 |
Correct |
0 ms |
344 KB |
Output is correct |
10 |
Correct |
0 ms |
348 KB |
Output is correct |
11 |
Correct |
0 ms |
348 KB |
Output is correct |
12 |
Correct |
1 ms |
348 KB |
Output is correct |
13 |
Correct |
0 ms |
344 KB |
Output is correct |
14 |
Correct |
0 ms |
348 KB |
Output is correct |
15 |
Correct |
0 ms |
344 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
348 KB |
Output is correct |
2 |
Correct |
0 ms |
348 KB |
Output is correct |
3 |
Correct |
0 ms |
348 KB |
Output is correct |
4 |
Correct |
1 ms |
348 KB |
Output is correct |
5 |
Correct |
0 ms |
348 KB |
Output is correct |
6 |
Correct |
0 ms |
348 KB |
Output is correct |
7 |
Correct |
0 ms |
348 KB |
Output is correct |
8 |
Correct |
0 ms |
348 KB |
Output is correct |
9 |
Correct |
0 ms |
344 KB |
Output is correct |
10 |
Correct |
0 ms |
348 KB |
Output is correct |
11 |
Correct |
0 ms |
348 KB |
Output is correct |
12 |
Correct |
1 ms |
348 KB |
Output is correct |
13 |
Correct |
0 ms |
344 KB |
Output is correct |
14 |
Correct |
0 ms |
348 KB |
Output is correct |
15 |
Correct |
0 ms |
344 KB |
Output is correct |
16 |
Correct |
0 ms |
348 KB |
Output is correct |
17 |
Correct |
0 ms |
456 KB |
Output is correct |
18 |
Correct |
0 ms |
460 KB |
Output is correct |
19 |
Correct |
0 ms |
348 KB |
Output is correct |
20 |
Correct |
0 ms |
348 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
348 KB |
Output is correct |
2 |
Correct |
0 ms |
348 KB |
Output is correct |
3 |
Correct |
0 ms |
348 KB |
Output is correct |
4 |
Correct |
1 ms |
348 KB |
Output is correct |
5 |
Correct |
0 ms |
348 KB |
Output is correct |
6 |
Correct |
0 ms |
348 KB |
Output is correct |
7 |
Correct |
0 ms |
348 KB |
Output is correct |
8 |
Correct |
0 ms |
348 KB |
Output is correct |
9 |
Correct |
0 ms |
344 KB |
Output is correct |
10 |
Correct |
0 ms |
348 KB |
Output is correct |
11 |
Correct |
0 ms |
348 KB |
Output is correct |
12 |
Correct |
1 ms |
348 KB |
Output is correct |
13 |
Correct |
0 ms |
344 KB |
Output is correct |
14 |
Correct |
0 ms |
348 KB |
Output is correct |
15 |
Correct |
0 ms |
344 KB |
Output is correct |
16 |
Correct |
0 ms |
348 KB |
Output is correct |
17 |
Correct |
0 ms |
456 KB |
Output is correct |
18 |
Correct |
0 ms |
460 KB |
Output is correct |
19 |
Correct |
0 ms |
348 KB |
Output is correct |
20 |
Correct |
0 ms |
348 KB |
Output is correct |
21 |
Correct |
1 ms |
348 KB |
Output is correct |
22 |
Correct |
1 ms |
604 KB |
Output is correct |
23 |
Correct |
1 ms |
348 KB |
Output is correct |
24 |
Correct |
1 ms |
604 KB |
Output is correct |
25 |
Correct |
1 ms |
348 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
42 ms |
11460 KB |
Output is correct |
2 |
Correct |
48 ms |
15148 KB |
Output is correct |
3 |
Correct |
39 ms |
10952 KB |
Output is correct |
4 |
Correct |
44 ms |
13976 KB |
Output is correct |
5 |
Correct |
42 ms |
10956 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
77 ms |
17092 KB |
Output is correct |
2 |
Correct |
75 ms |
17092 KB |
Output is correct |
3 |
Correct |
78 ms |
16840 KB |
Output is correct |
4 |
Correct |
76 ms |
17608 KB |
Output is correct |
5 |
Correct |
75 ms |
16688 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
96 ms |
17452 KB |
Output is correct |
2 |
Correct |
80 ms |
17092 KB |
Output is correct |
3 |
Correct |
86 ms |
16840 KB |
Output is correct |
4 |
Correct |
101 ms |
18140 KB |
Output is correct |
5 |
Correct |
91 ms |
17288 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
348 KB |
Output is correct |
2 |
Correct |
0 ms |
348 KB |
Output is correct |
3 |
Correct |
0 ms |
348 KB |
Output is correct |
4 |
Correct |
1 ms |
348 KB |
Output is correct |
5 |
Correct |
0 ms |
348 KB |
Output is correct |
6 |
Correct |
0 ms |
348 KB |
Output is correct |
7 |
Correct |
0 ms |
348 KB |
Output is correct |
8 |
Correct |
0 ms |
348 KB |
Output is correct |
9 |
Correct |
0 ms |
344 KB |
Output is correct |
10 |
Correct |
0 ms |
348 KB |
Output is correct |
11 |
Correct |
42 ms |
11460 KB |
Output is correct |
12 |
Correct |
48 ms |
15148 KB |
Output is correct |
13 |
Correct |
39 ms |
10952 KB |
Output is correct |
14 |
Correct |
44 ms |
13976 KB |
Output is correct |
15 |
Correct |
42 ms |
10956 KB |
Output is correct |
16 |
Correct |
38 ms |
10960 KB |
Output is correct |
17 |
Correct |
46 ms |
15020 KB |
Output is correct |
18 |
Correct |
52 ms |
15204 KB |
Output is correct |
19 |
Correct |
37 ms |
10668 KB |
Output is correct |
20 |
Correct |
44 ms |
14788 KB |
Output is correct |
21 |
Correct |
37 ms |
10696 KB |
Output is correct |
22 |
Correct |
41 ms |
14280 KB |
Output is correct |
23 |
Correct |
40 ms |
11720 KB |
Output is correct |
24 |
Correct |
41 ms |
14028 KB |
Output is correct |
25 |
Correct |
40 ms |
11468 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
348 KB |
Output is correct |
2 |
Correct |
0 ms |
348 KB |
Output is correct |
3 |
Correct |
0 ms |
348 KB |
Output is correct |
4 |
Correct |
1 ms |
348 KB |
Output is correct |
5 |
Correct |
0 ms |
348 KB |
Output is correct |
6 |
Correct |
0 ms |
348 KB |
Output is correct |
7 |
Correct |
0 ms |
348 KB |
Output is correct |
8 |
Correct |
0 ms |
348 KB |
Output is correct |
9 |
Correct |
0 ms |
344 KB |
Output is correct |
10 |
Correct |
0 ms |
348 KB |
Output is correct |
11 |
Correct |
0 ms |
348 KB |
Output is correct |
12 |
Correct |
1 ms |
348 KB |
Output is correct |
13 |
Correct |
0 ms |
344 KB |
Output is correct |
14 |
Correct |
0 ms |
348 KB |
Output is correct |
15 |
Correct |
0 ms |
344 KB |
Output is correct |
16 |
Correct |
0 ms |
348 KB |
Output is correct |
17 |
Correct |
0 ms |
456 KB |
Output is correct |
18 |
Correct |
0 ms |
460 KB |
Output is correct |
19 |
Correct |
0 ms |
348 KB |
Output is correct |
20 |
Correct |
0 ms |
348 KB |
Output is correct |
21 |
Correct |
1 ms |
348 KB |
Output is correct |
22 |
Correct |
1 ms |
604 KB |
Output is correct |
23 |
Correct |
1 ms |
348 KB |
Output is correct |
24 |
Correct |
1 ms |
604 KB |
Output is correct |
25 |
Correct |
1 ms |
348 KB |
Output is correct |
26 |
Correct |
42 ms |
11460 KB |
Output is correct |
27 |
Correct |
48 ms |
15148 KB |
Output is correct |
28 |
Correct |
39 ms |
10952 KB |
Output is correct |
29 |
Correct |
44 ms |
13976 KB |
Output is correct |
30 |
Correct |
42 ms |
10956 KB |
Output is correct |
31 |
Correct |
77 ms |
17092 KB |
Output is correct |
32 |
Correct |
75 ms |
17092 KB |
Output is correct |
33 |
Correct |
78 ms |
16840 KB |
Output is correct |
34 |
Correct |
76 ms |
17608 KB |
Output is correct |
35 |
Correct |
75 ms |
16688 KB |
Output is correct |
36 |
Correct |
96 ms |
17452 KB |
Output is correct |
37 |
Correct |
80 ms |
17092 KB |
Output is correct |
38 |
Correct |
86 ms |
16840 KB |
Output is correct |
39 |
Correct |
101 ms |
18140 KB |
Output is correct |
40 |
Correct |
91 ms |
17288 KB |
Output is correct |
41 |
Correct |
38 ms |
10960 KB |
Output is correct |
42 |
Correct |
46 ms |
15020 KB |
Output is correct |
43 |
Correct |
52 ms |
15204 KB |
Output is correct |
44 |
Correct |
37 ms |
10668 KB |
Output is correct |
45 |
Correct |
44 ms |
14788 KB |
Output is correct |
46 |
Correct |
37 ms |
10696 KB |
Output is correct |
47 |
Correct |
41 ms |
14280 KB |
Output is correct |
48 |
Correct |
40 ms |
11720 KB |
Output is correct |
49 |
Correct |
41 ms |
14028 KB |
Output is correct |
50 |
Correct |
40 ms |
11468 KB |
Output is correct |
51 |
Correct |
77 ms |
13508 KB |
Output is correct |
52 |
Correct |
92 ms |
16588 KB |
Output is correct |
53 |
Correct |
87 ms |
13508 KB |
Output is correct |
54 |
Correct |
97 ms |
16708 KB |
Output is correct |
55 |
Correct |
78 ms |
13740 KB |
Output is correct |
# |
결과 |
실행 시간 |
메모리 |
Grader output |
1 |
Correct |
0 ms |
348 KB |
Output is correct |
2 |
Correct |
0 ms |
348 KB |
Output is correct |
3 |
Correct |
0 ms |
348 KB |
Output is correct |
4 |
Correct |
1 ms |
348 KB |
Output is correct |
5 |
Correct |
0 ms |
348 KB |
Output is correct |
6 |
Correct |
0 ms |
348 KB |
Output is correct |
7 |
Correct |
0 ms |
348 KB |
Output is correct |
8 |
Correct |
0 ms |
348 KB |
Output is correct |
9 |
Correct |
0 ms |
344 KB |
Output is correct |
10 |
Correct |
0 ms |
348 KB |
Output is correct |
11 |
Correct |
0 ms |
348 KB |
Output is correct |
12 |
Correct |
1 ms |
348 KB |
Output is correct |
13 |
Correct |
0 ms |
344 KB |
Output is correct |
14 |
Correct |
0 ms |
348 KB |
Output is correct |
15 |
Correct |
0 ms |
344 KB |
Output is correct |
16 |
Correct |
0 ms |
348 KB |
Output is correct |
17 |
Correct |
0 ms |
456 KB |
Output is correct |
18 |
Correct |
0 ms |
460 KB |
Output is correct |
19 |
Correct |
0 ms |
348 KB |
Output is correct |
20 |
Correct |
0 ms |
348 KB |
Output is correct |
21 |
Correct |
1 ms |
348 KB |
Output is correct |
22 |
Correct |
1 ms |
604 KB |
Output is correct |
23 |
Correct |
1 ms |
348 KB |
Output is correct |
24 |
Correct |
1 ms |
604 KB |
Output is correct |
25 |
Correct |
1 ms |
348 KB |
Output is correct |
26 |
Correct |
42 ms |
11460 KB |
Output is correct |
27 |
Correct |
48 ms |
15148 KB |
Output is correct |
28 |
Correct |
39 ms |
10952 KB |
Output is correct |
29 |
Correct |
44 ms |
13976 KB |
Output is correct |
30 |
Correct |
42 ms |
10956 KB |
Output is correct |
31 |
Correct |
77 ms |
17092 KB |
Output is correct |
32 |
Correct |
75 ms |
17092 KB |
Output is correct |
33 |
Correct |
78 ms |
16840 KB |
Output is correct |
34 |
Correct |
76 ms |
17608 KB |
Output is correct |
35 |
Correct |
75 ms |
16688 KB |
Output is correct |
36 |
Correct |
96 ms |
17452 KB |
Output is correct |
37 |
Correct |
80 ms |
17092 KB |
Output is correct |
38 |
Correct |
86 ms |
16840 KB |
Output is correct |
39 |
Correct |
101 ms |
18140 KB |
Output is correct |
40 |
Correct |
91 ms |
17288 KB |
Output is correct |
41 |
Correct |
38 ms |
10960 KB |
Output is correct |
42 |
Correct |
46 ms |
15020 KB |
Output is correct |
43 |
Correct |
52 ms |
15204 KB |
Output is correct |
44 |
Correct |
37 ms |
10668 KB |
Output is correct |
45 |
Correct |
44 ms |
14788 KB |
Output is correct |
46 |
Correct |
37 ms |
10696 KB |
Output is correct |
47 |
Correct |
41 ms |
14280 KB |
Output is correct |
48 |
Correct |
40 ms |
11720 KB |
Output is correct |
49 |
Correct |
41 ms |
14028 KB |
Output is correct |
50 |
Correct |
40 ms |
11468 KB |
Output is correct |
51 |
Correct |
77 ms |
13508 KB |
Output is correct |
52 |
Correct |
92 ms |
16588 KB |
Output is correct |
53 |
Correct |
87 ms |
13508 KB |
Output is correct |
54 |
Correct |
97 ms |
16708 KB |
Output is correct |
55 |
Correct |
78 ms |
13740 KB |
Output is correct |
56 |
Execution timed out |
1076 ms |
127336 KB |
Time limit exceeded |
57 |
Halted |
0 ms |
0 KB |
- |