Submission #204300

#	Time	Username	Problem	Language	Result	Execution time	Memory
204300		Segtree	Teams (IOI15_teams)	C++14	13 / 100	4065 ms	72404 KiB

teams

#include"teams.h"
#include<iostream>
#include<algorithm>
#include<vector>
#include<queue>
#include<unordered_set>
using namespace std;
typedef long long ll;
typedef pair<ll,ll> P;
#define chmin(a,b) a=min(a,b)
#define chmax(a,b) a=max(a,b)
#define all(x) x.begin(),x.end()
#define rep(i,n) for(int i=0;i<n;i++)
#define mod 1000000007
#define mad(a,b) a=(a+b)%mod

#define up(x,y) upper_bound(all(x),y)
#define NS 100010
vector<P> v;
ll dat[NS*2],laste[2*NS],lastd[2*NS],correctA;
vector<ll> as[NS*2];
unordered_set<ll> used;
void add(int K){
	correctA=K;
}
ll eval(ll k){
	if(k==0)return 0;
	used.insert(k);
	if(k<NS){
		chmax(lastd[k*2],lastd[k]);
		chmax(lastd[k*2+1],lastd[k]);
	}
	if(lastd[k]==-1e17){
		dat[k]+=up(as[k],correctA)-up(as[k],laste[k]);
	}
	else{
		dat[k]=up(as[k],correctA)-up(as[k],lastd[k]);
	}
	laste[k]=correctA;
	lastd[k]=-1e17;
	return dat[k];
}
void erase(ll l,ll r){
	l+=NS,r+=NS+1;
	for(ll d=20;~d;d--){
		eval(l>>d);
		eval(r>>d);
	}
	for(ll a=l,b=r;a<b;a>>=1,b>>=1){
		if(a&1){chmax(lastd[a],correctA);a++;}
		if(b&1){b--;chmax(lastd[b],correctA);}
	}
	for(ll a=l,b=r;a+b;a>>=1,b>>=1){
		dat[a/2]=eval(a)+eval(a^1);
		dat[b/2]=eval(b)+eval(b^1);
	}
}
int sum(ll l,ll r){
	l+=NS,r+=NS+1;
	ll res=0;
	for(ll d=20;~d;d--){
		eval(l>>d);
		eval(r>>d);
	}
	for(ll a=l,b=r;a<b;a>>=1,b>>=1){
		if(a&1){res+=eval(a);a++;}
		if(b&1){b--;res+=eval(b);}
	}
	return res;
}

void init(int N,int A[],int B[]){
	rep(i,N)v.push_back(make_pair(B[i],A[i]));
	sort(v.begin(),v.end());
	rep(i,NS){
		if(i<N)as[i+NS].push_back(v[i].second);
	}
	for(int i=NS-1;i>0;i--){
		for(auto t:as[i*2])as[i].push_back(t);
		for(auto t:as[i*2+1])as[i].push_back(t);
		sort(as[i].begin(),as[i].end());
	}
}

int can(int M,int K[]){
	used.clear();
	correctA=0;
	sort(K,K+M);
	bool ans=1;
	rep(i,M){
		add(K[i]);
		ll l,r,mid;
		l=-1,r=v.size();
		while(l<r-1){
			mid=(l+r)>>1;
			(v[mid].first<K[i]?l:r)=mid;
		}
		//cerr<<"##"<<sum(0,v.size()-1)<<endl;
		erase(0,l);//cerr<<"$"<<l<<endl;
		l=-1,r=v.size();
		//cerr<<"#"<<sum(0,v.size()-1)<<endl;
		while(l<r-1){
			mid=(l+r)>>1;
			(sum(0,mid)>=K[i]?r:l)=mid;
		}
		if(r==v.size()){
			ans=0;
			break;
		}
		erase(0,r);
	}
	for(auto i:used)dat[i]=0,laste[i]=lastd[i]=-1e17;
	return ans;
}
/*
int main(){
	int n,a[110],b[110];
	cin>>n;
	rep(i,n)cin>>a[i]>>b[i];
	init(n,a,b);
	while(1){
		int m,k[110];
		cin>>m;
		rep(i,m)cin>>k[i];
		cout<<(can(m,k)?"YES":"NO")<<endl;
	}
}*/

Compilation message (stderr)

teams.cpp: In function 'll eval(ll)':
teams.cpp:33:12: warning: conversion to 'double' from 'll {aka long long int}' may alter its value [-Wconversion]
  if(lastd[k]==-1e17){
     ~~~~~~~^
teams.cpp: In function 'int sum(ll, ll)':
teams.cpp:69:9: warning: conversion to 'int' from 'll {aka long long int}' may alter its value [-Wconversion]
  return res;
         ^~~
teams.cpp: In function 'int can(int, int*)':
teams.cpp:106:7: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   if(r==v.size()){
      ~^~~~~~~~~~

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