제출 #256357

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
256357		dolphingarlic	팀들 (IOI15_teams)	C++14	34 / 100	4067 ms	181452 KiB

teams

#include "teams.h"

#include <bits/stdc++.h>
using namespace std;

const int MAX_N = 5e5, MAX_SEG = 1e7, B_SIZE = 800;

int N, cnt = 1, seg_val[MAX_SEG], left_c[MAX_SEG], right_c[MAX_SEG];

struct Segtree {
    map<int, int> points[MAX_N + 1];
    int roots[MAX_N + 2];

    void build() {
        for (int i = 1; i <= N; i++) {
            roots[i + 1] = roots[i];
            for (pair<int, int> j : points[i]) update(j.first, j.second, roots[i + 1]);
        }
    }

    void update(int pos, int val, int &node, int l = 1, int r = N) {
        seg_val[cnt] = seg_val[node] + val;
        left_c[cnt] = left_c[node];
        right_c[cnt] = right_c[node];
        node = cnt++;

        if (l == r) return;
        int mid = (l + r) / 2;
        if (pos > mid) update(pos, val, right_c[node], mid + 1, r);
        else update(pos, val, left_c[node], l, mid);
    }

    int query(int a, int b, int node, int l, int r) {
        if (a > r || b < l) return 0;
        if (a <= l && b >= r) return seg_val[node];
        int mid = (l + r) / 2;
        return query(a, b, left_c[node], l, mid) + query(a, b, right_c[node], mid + 1, r);
    }
    int query(int a1, int a2, int b1) {
        return query(a1, a2, roots[N + 1], 1, N) - query(a1, a2, roots[b1], 1, N);
    }
} segtree;

int dp[MAX_N];
pair<int, int> students[MAX_N];

void init(int N, int A[], int B[]) {
    ::N = N;
    for (int i = 0; i < N; i++) {
        students[i] = {B[i], A[i]};
        segtree.points[B[i]][A[i]]++;
    }
    segtree.build();
    sort(students, students + N);
}

int can(int M, int K[]) {
    sort(K, K + M);
    if (M > B_SIZE) {
        priority_queue<int> pq;
        for (int i = 0, j = 0; i < M; i++) {
            while (j < N && students[j].second <= K[i]) pq.push(-students[j++].first);
            while (pq.size() && -pq.top() < K[i]) pq.pop();
            for (int k = 0; k < K[i]; k++) {
                if (!pq.size()) return 0;
                pq.pop();
            } 
        }
        return 1;
    } else {
        for (int i = 0; i < M; i++) {
            dp[i] = segtree.query(1, K[i], K[i]) - K[i];
            for (int j = 0; j < i; j++) {
                dp[i] = min(dp[i], dp[j] + segtree.query(K[j] + 1, K[i], K[i]) - K[i]);
            }
            if (dp[i] < 0) return 0;
        }
        return 1;
    }
}

컴파일 시 표준 에러 (stderr) 메시지

teams.cpp: In function 'void init(int, int*, int*)':
teams.cpp:47:34: warning: declaration of 'N' shadows a global declaration [-Wshadow]
 void init(int N, int A[], int B[]) {
                                  ^
teams.cpp:8:5: note: shadowed declaration is here
 int N, cnt = 1, seg_val[MAX_SEG], left_c[MAX_SEG], right_c[MAX_SEG];
     ^

• Subtask 1: 21 / 21
• Subtask 2: 13 / 13
• Subtask 3: 0 / 43
• Subtask 4: 0 / 23