Submission #204312

#	Time	Username	Problem	Language	Result	Execution time	Memory
204312		Segtree	Teams (IOI15_teams)	C++14	77 / 100	4081 ms	315720 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
#pragma gcc optimize("O3")
#pragma gcc optimize("unroll-loops")
#pragma gcc target("avx2,see4")

#define up(x,y) upper_bound(all(x),y)
#define NS (1<<19)
vector<P> v;
ll dat[NS*2],laste[2*NS],lastd[2*NS],correctA;
vector<ll> as[NS*2];
vector<ll> used;
void add(int K){
	correctA=K;
}
ll eval(ll k){
	if(k==0)return 0;
	used.emplace_back(k);
	if(k<NS){
		chmax(lastd[k*2],lastd[k]); used.emplace_back(k*2);
		chmax(lastd[k*2+1],lastd[k]); used.emplace_back(k*2+1);
	}
	if(lastd[k]==-1e17){
		if(laste[k]!=correctA){
			dat[k]+=up(as[k],correctA)-up(as[k],laste[k]);
		}
	}
	else{
		if(lastd[k]!=correctA){
			dat[k]=up(as[k],correctA)-up(as[k],lastd[k]);
		}
		else dat[k]=0;
	}
	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.emplace_back(make_pair(B[i],A[i]));
	sort(v.begin(),v.end());
	rep(i,NS){
		if(i<N)as[i+NS].emplace_back(v[i].second);
	}
	for(int i=NS-1;i>0;i--){
		for(auto t:as[i*2])as[i].emplace_back(t);
		for(auto t:as[i*2+1])as[i].emplace_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;
		}
		erase(0,l);
		if(eval(1)<K[i]){
			ans=0;
			break;
		}
		ll k=1,rui=0;
		while(k<NS){
			ll ev=eval(k*2);
			if(rui+ev>=K[i]){
				k=k*2;
			}
			else{
				rui+=ev;
				k=k*2+1;
			}
		}
		erase(0,k-NS);
	}
	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:16:0: warning: ignoring #pragma gcc optimize [-Wunknown-pragmas]
 #pragma gcc optimize("O3")
 
teams.cpp:17:0: warning: ignoring #pragma gcc optimize [-Wunknown-pragmas]
 #pragma gcc optimize("unroll-loops")
 
teams.cpp:18:0: warning: ignoring #pragma gcc target [-Wunknown-pragmas]
 #pragma gcc target("avx2,see4")
 
teams.cpp: In function 'll eval(ll)':
teams.cpp:36: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:77:9: warning: conversion to 'int' from 'll {aka long long int}' may alter its value [-Wconversion]
  return res;
         ^~~

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