백준 23760번 : 까다로운 아이들과 선물상자(C/C++)

<문제>

백준 23760번 

 

 

<풀이>

2243번 사탕상자와 비슷한 문제이다.

 

풀이의 핵심은 bi번째로 사탕수가 많은 상자를 선택해야한다. 세그먼트 트리를 이용하면 log n만에 선택할 수 있다.

배열의 index가 "사탕수"이고, value는 동일한 사탕수가 담긴 "상자의 개수"라고 해보자.

예를 들어 사탕수가 5개가 담긴 상자가 1개 있다고 하면 arr[5] = 1이다.

이 배열을 세그먼트 트리로 만들어주면 된다. 트리의 리프 노드에는 사탕수가 x인 상자의 개수가 저장되있을 것이다.

 

query라는 함수는 트리에서 cnt번째로 사탕수가 많은 index를 찾아주는 함수이다.

만약 cnt <= tree[2*node+1]일 경우 오른쪽 트리를 방문하고, 반대일 경우 왼쪽 트리를 방문한다.

 

update함수는 일반적인 세그먼트 트리 문제와 동일하여 코드를 참고하면 될것 같다.

 

<코드>

#include<bits/stdc++.h>
#define fio()                     \
    ios_base::sync_with_stdio(0); \
    cin.tie(0)
using namespace std;
 
typedef long long ll;
typedef pair<int, int> pii;
typedef pair<double, double> pll;
typedef tuple<int, int, int> tpi;
typedef tuple<ll, ll, ll> tpl;
typedef pair<double, ll> pdl;
 
const int INF = 0x3f3f3f3f;
const ll LINF = 0x3f3f3f3f3f3f3f3f;
const int dx[] = { 0, 1, 0, -1 };
const int dy[] = { 1, 0, -1, 0 };
const double pi = acos(-1);
const int MOD = 1000000007;
 
int tree[300000];
 
int query(int node, int s, int e, int cnt){
    if(s == e) return s;
    int mid = (s+e)/2;
    if(cnt <= tree[2*node+1]) return query(2*node+1, mid+1, e, cnt);
    else return query(2*node, s, mid, cnt-tree[2*node+1]);
}
 
void update(int node, int s, int e, int idx, int diff){
    if(idx < s || idx > e) return;
    tree[node] += diff;
    if(s == e) return;
    int mid = (s+e)/2;
    update(2*node, s, mid, idx, diff);
    update(2*node+1, mid+1, e, idx, diff);
 
    return;
}
 
int main(){
    int n, m; scanf("%d %d", &n, &m);
    for(int i = 0; i < n; i++){
        int c; scanf("%d", &c);
        update(1, 0, 100000, c, 1);
    }
    int w[101010], b[101010];
    for(int i = 0; i < m; i++) scanf("%d", &w[i]);
    for(int j = 0; j < m; j++) scanf("%d", &b[j]);
    for(int i = 0; i < m; i++){
        int idx = query(1, 0, 100000, b[i]);
        if(w[i] > idx){
            printf("0");
            return 0;
        }
        update(1, 0, 100000, idx, -1);
        update(1, 0, 100000, idx-w[i], 1);
    }
    printf("1");
}

 

잘못된 점은 지적 부탁드립니다~