Znzryb's Blog
0%

ABC-388-D - Coming of Age Celebration

Posted on Edited on

思路讲解

赛时就做出来了，存档一下

好像有更简单的做法（倒过来想），但还是算了

AC代码

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
#include <iostream>
#include <cstring>
#include <algorithm>
#include <deque>
#include <queue>
#include <vector>
#include <set>
#include <map>
#include <unordered_map>
#include <cmath>
#include <bitset>
#include <iterator>
#include <random>
#include <iomanip>
#include <cctype>
#include <array>

typedef long long ll;
typedef std::pair<ll,ll> pll;
typedef std::array<ll,3> arr;
const ll MAXN=static_cast<ll>(1e6)+117;

ll N,T,A[MAXN],Ans[MAXN],tr[MAXN];
//std::set<ll> ali;
//// 数->第几个
//std::unordered_map<ll,ll> li;

inline ll lowbit(ll x){
	return x&(-x);
}

inline void add(ll x){
	ll idx=x+1;
	while (idx<=MAXN-10) {
		tr[idx]+=1;
		idx+=lowbit(idx);
	}
}

inline ll query(ll x){
	ll idx=x+1;
	ll res=0;
	// x是在后面的数，查找前面的全部比它大的数
	// 当然，直接查找全部比它大的数比较难
	// 所以返回的是所有比它<=的数的数量
	while(idx>0){
		res+=tr[idx];
		idx-=lowbit(idx);
	}
	return res;
}

int main()
{
	std::ios::sync_with_stdio(false);
	std::cin.tie(0);std::cout.tie(0);
	std::cin>>N;
	for(int i=1;i<=N;++i){
		std::cin>>A[i];
//		ali.insert(A[i]);
	}
//	for(std::set<ll>::iterator it=ali.begin();it!=ali.end();it++){
//		li[*it]=++idx;
//	}
	for(int i=1;i<=N;++i){
		// 在这之前的成年人，我们看他们能不能给到第i年
		ll lowerThanI=query(i-1);
		add(A[i]+i-1-lowerThanI+i);
		// 最初有多少宝石
		Ans[i]=A[i]+i-1-lowerThanI;
	}
	for(int i=1;i<=N;++i){
		Ans[i]=std::max(Ans[i]-(N-i),0LL);
		std::cout<<Ans[i]<<" ";
	}
	std::cout<<"\n";
	return 0;
}
// AC https://atcoder.jp/contests/abc388/submissions/61585347

心路历程（WA，TLE，MLE……）