제출 #106412

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
106412		figter001	팀들 (IOI15_teams)	C++17	0 / 100	2444 ms	193184 KiB

teams

#include "teams.h"
#include <bits/stdc++.h>

using namespace std;

typedef long long ll;
typedef long double ld;
typedef unsigned long long ull;
const int nax = 5e5+50;

struct node{
	int val,l,r;
	node(){
		l = r = -1;
		val = 0;
	}
};

vector<node> seg;

int idx[nax],cnt,k,l,r;
vector<pair<int,int>> p;

void build(int n,int s , int e){
	if(s == e){
		seg[n].val = 0;
		return;
	}
	seg[n].l = cnt++;
	seg[n].r = cnt++;
	build(seg[n].l,s,(s+e)/2);
	build(seg[n].r,(s+e)/2+1,e);
}

int update(int n , int s , int e,int at){
	if(s > at || e < at)
		return n;
	int newId = cnt++;
	seg[newId].val = seg[n].val;
	if(s == e){
		seg[newId].val++;
		return newId;
	}
	seg[newId].l = update(seg[n].l,s,(s+e)/2,at);
	seg[newId].r = update(seg[n].r,(s+e)/2+1,e,at);
	seg[newId].val = seg[seg[newId].l].val + seg[seg[newId].r].val;
	return newId;
}

void get(int n,int s,int e){
	assert(n != -1);
	if(l > r || s > r || e < l)
		return;
	if(s == e){
		k -= min(k,seg[n].val);
		return;
	}
	if(s >= l && e <= r){
		int le = seg[n].l;
		int ri = seg[n].r; 
		if(seg[le].val <= k){
			k -= seg[le].val;
			get(ri,(s+e)/2+1,e);
		}else{
			get(le,s,(s+e)/2);
		}
	}else{
		get(seg[n].l,s,(s+e)/2);
		get(seg[n].r,(s+e)/2+1,e);
	}
}

int n;

void init(int N, int A[], int B[]) {
	n = N;
	seg.resize(30*n);
	for(int i=0;i<n;i++){
		p.push_back({A[i],B[i]});
		assert(B[i] <= N);
	}
	cnt = 1;
	int lst = cnt;
	build(cnt++,1,n);
	sort(p.begin(),p.end());
	for(int i=0;i<n;i++){
		int a = p[i].first;
		int b = p[i].second;
		idx[a] = cnt;
		update(lst,1,n,b);
		lst = idx[a];
	}
}

int can(int M, int K[]) {
	sort(K,K+M);
	pair<int,int> lst = {-1,0};
	for(int i=0;i<M;i++){
		int st = lower_bound(p.begin(),p.end(),make_pair(K[i]+1,0)) - p.begin();
		if(st == 0)
			return 0;
		// cout << i << endl;
		if(lst.first != -1){
			k = lst.second;
			l = K[i-1];
			r = K[i] - 1;
			get(lst.first,1,n);
		}
		st = idx[p[st-1].first];
		k += K[i];
		l = K[i];
		r = n;
		lst = {st,k};
		// cout << l << ' ' << r << ' ' << k << endl;
		// cout << k << ' ';
		// cout << l << ' ' << r << ' ' << k << endl;
		get(st,1,n);
		// cout << k << endl;
		assert(k >= 0);
		if(k != 0)
			return 0;
	}
	return 1;
}

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

teams.cpp: In function 'int can(int, int*)':
teams.cpp:99:63: warning: conversion to 'int' from '__gnu_cxx::__normal_iterator<std::pair<int, int>*, std::vector<std::pair<int, int> > >::difference_type {aka long int}' may alter its value [-Wconversion]
   int st = lower_bound(p.begin(),p.end(),make_pair(K[i]+1,0)) - p.begin();
            ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~~~~~~~~~~

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