백준 3653번: 영화 수집
15117 단어 Segment TreepscppSegment Tree
영화 수집
아이디어
쿼리가 최대 10만개 주어진다. 배열을 0~99,999, 100,000~199,999 두 구간으로 나누고, DVD를 가장 위로 올릴 때 10만부터 시작해서 하나씩 순서대로 올리면서 합을 구하면 풀 수 있다.
DVD 번호가 주어졌을 때 배열의 어디에 위치하는지 바로 알 수 있게 인덱스를 저장한다.
코드
#include <bits/stdc++.h>
using namespace std;
const int MAX = 100000;
int T, N, M, TOP;
vector<int> v;
vector<int> dvdpos; // dvd가 위치한곳 저장.
void update(int start, int end, int idx, int value, int node) {
if (idx < start || end < idx) return;
if (start == end) {
v[node] = value;
return;
}
update(start, (start+end)/2, idx, value, node*2);
update((start+end)/2+1, end, idx, value, node*2+1);
v[node] = v[node*2] + v[node*2+1];
}
int query(int start, int end, int left, int right, int node) {
if (right < start || end < left) return 0;
if (left <= start && end <= right) return v[node];
return query(start, (start+end)/2, left, right, node*2) + query((start+end)/2+1, end, left, right, node*2+1);
}
int main() {
ios_base::sync_with_stdio(0);
cin.tie(0);
cin >> T;
while (T--) {
cin >> N >> M;
TOP = MAX;
dvdpos = vector<int>(MAX+1, 0);
v = vector<int>(4*2*MAX, 0);
for (int i = 0; i < N; i++) {
update(0, 2*MAX-1, i, 1, 1);
dvdpos[i+1] = N-i-1;
}
for (int i = 0; i < M; i++) {
int x;
cin >> x;
cout << query(0, 2*MAX-1, dvdpos[x]+1, 2*MAX-1, 1) << ' ';
update(0, 2*MAX-1, dvdpos[x], 0, 1);
update(0, 2*MAX-1, TOP, 1, 1);
dvdpos[x] = TOP++;
}
cout << '\n';
}
return 0;
}
여담
쿼리 개수가 제한적일 때 유용한 테크닉인 것 같다. 지식이 늘었다.
Author And Source
이 문제에 관하여(백준 3653번: 영화 수집), 우리는 이곳에서 더 많은 자료를 발견하고 링크를 클릭하여 보았다 https://velog.io/@ks1ksi/백준-3653번-영화-수집저자 귀속: 원작자 정보가 원작자 URL에 포함되어 있으며 저작권은 원작자 소유입니다.
우수한 개발자 콘텐츠 발견에 전념 (Collection and Share based on the CC Protocol.)