-
Notifications
You must be signed in to change notification settings - Fork 0
/
24_MEASURETIME.cpp
61 lines (56 loc) · 1.38 KB
/
24_MEASURETIME.cpp
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
#include <cstdio>
#include <vector>
using namespace std;
// 코드 24.7 펜윅 트리의 완전한 구현
// 펜윅 트리의 구현. 가상의 배열 A[]의 부분 합을
// 빠르게 구현할 수 있도록 한다. 초기화시에는 A[]의
// 원소가 전부 0이라고 생각한다.
struct FenwickTree {
vector<long long> tree;
FenwickTree(int n) : tree(n + 1) {}
// A[0..pos]의 부분 합을 구한다
long long sum(int pos) {
// 인덱스가 1 부터 시작한다고 생각하자
++pos;
long long ret = 0;
while (pos > 0) {
ret += tree[pos];
// 다음 구간을 찾기 위해 최종 비트를 지운다
pos &= (pos - 1);
}
return ret;
}
// A[pos]에 val을 더한다
void add(int pos, long long val) {
++pos;
while (pos < tree.size()) {
tree[pos] += val;
pos += (pos & -pos);
}
}
};
// 코드 24.8 삽입 정렬 시간 재기 문젤르 펜윅 트리로 해결하기
// 펜윅 트리를 이용해 문제를 해결한다
long long coundMove(const vector<int>& A) {
FenwickTree tree(1000000);
long long ret = 0;
for (int i = 0; i < A.size(); ++i) {
ret += tree.sum(999999) - tree.sum(A[i]);
tree.add(A[i], 1);
}
return ret;
}
int main(void) {
int C, N;
scanf("%d", &C);
while (C--) {
scanf("%d", &N);
vector<int> num;
num.resize(N);
for (int i = 0; i < N; i++) {
scanf("%d", &num[i]);
}
printf("%d\n", coundMove(num));
}
return 0;
}