[BOJ] 1321 군인

문제: https://www.acmicpc.net/problem/1321

 

 

소스 코드:

#include <iostream>
#include <string.h>
#include <queue>
#include <vector>
#include <algorithm>
using namespace std;

int N,M,seg[500001*4];
int update(int pos, int val, int node, int x, int y) {
	if (pos < x || y < pos) return seg[node];
	if (x == y) return seg[node] += val;
	int mid = (x + y) >> 1;
	return seg[node] = update(pos, val, node * 2, x, mid) + update(pos, val, node * 2 + 1, mid + 1, y);
}
int query(int sol,int node, int x, int y) {
	if (x==y) return x;
	int mid = (x + y) >> 1;
	if (sol <= seg[node * 2]) return query(sol, node * 2, x, mid);
	
	return query(sol-seg[node*2], node * 2+1, mid+1, y);
}
// seg[node]로 위치 파악하고 싶은데 못 하겠다 다음에 하장...
int main() 
{
	ios::sync_with_stdio(false);
	cin.tie(NULL), cout.tie(NULL);
	cin >> N;
	
	int a,b,c;
	for (int i = 0; i < N; i++) {
		cin >> a;
		update(i, a, 1, 0, N - 1);
	}
	cin >> M;
	while (M--) {
		cin >> a;
		if (a == 1) {
			cin >> b>>c;
			update(b - 1, c, 1, 0, N - 1);
		}
		else {
			cin >> b;
			cout << query(b,1,0,N-1)+1 << '\n';
		}
	}
	return 0;
}