Submission #28060

#	Time	Username	Problem	Language	Result	Execution time	Memory
28060		noobprogrammer	Teams (IOI15_teams)	C++14	100 / 100	1976 ms	145560 KiB

teams

#include "teams.h"
#include <bits/stdc++.h>
using namespace std ;
#define ii pair<int,int>
#define fi first 
#define se second 
#define vi vector<int>
#define vii vector<ii>
typedef long long ll ;

int st[10000010] , le[10000010] , r[10000010] , n , qs[500010] , rt[500010] , ncnt = 1 ;
vii srt , stk , srt2 ; 

void upd(int prev , int node , int f , int l ,int x){
	if(f == l){
		st[node] = st[prev] + 1 ;
		return ;
	}
	int mid = (f+l)>>1 ;
	if(mid < x){
		le[node] = le[prev] ; r[node] = ncnt++ ;
		upd( r[prev] , r[node] , mid+1 , l , x ) ;
	}
	else{
		le[node] = ncnt++ ; r[node] = r[prev] ;
		upd( le[prev] , le[node] , f , mid , x ) ;	
	}
	st[node] = st[le[node]] + st[r[node]] ;
}

int qr(int prev , int node , int f ,int l , int x ){
	if(x == 0) return 0 ;
	if(x >= l) return st[node] - st[prev] ;
	int mid = (f+l)>>1 ;
	if(mid < x) return st[le[node]] - st[le[prev]] + qr(r[prev] , r[node] , mid+1 , l , x ) ;
	else return qr(le[prev] , le[node] , f , mid , x ) ;
}

void init(int N, int A[], int B[]) {
	n = N ; 
	for(int i=0;i<n;i++) {
		qs[A[i]]++ ; qs[B[i]+1]-- ;
		srt.push_back( { B[i] + 1 , i } ) ; 
		srt2.push_back({A[i] , 1}) ; srt2.push_back( { B[i] + 1 , -1} ) ;
	}
	sort(srt.begin() , srt.end() ) ;
	for(int i=1;i<=n;i++) qs[i] += qs[i-1] ;
	 // printf("%d ", qs[i]) ;
	// cout << endl ; 
	int j = 0 , tmp , num ; 
	rt[0] = 0 ;
	for(int i=1;i<=n;i++){
		rt[i] = rt[i-1] ; 
		while( j<srt.size() && i >= srt[j].fi ) {
			// printf("%d %d\n" , srt[j].fi , srt[j].se) ;
			num = srt[j].se ;
			tmp = ncnt++ ;
			upd(rt[i] , tmp , 1 , n , A[num] ) ;
			rt[i] = tmp ; 
			j++ ;
		}
		// for(int j=1 ; j<=n ;j++) printf("%d " , qr(rt[i-1] , rt[i] , 1 , n , j) ) ;
		// cout << endl ;
	}
}

int can(int m, int arr[]) {
	int sum = 0 ;
	for(int i=0;i<m;i++) {
		sum += arr[i] ;
		if(sum > n) return 0 ; 
	}
	sort(arr , arr+m) ;
	int j ; 
	sum = 0 ;
	stk.clear() ; 
	// printf("--------\n") ;
	for(int i=0;i<m;){
		j = i ; sum = 0 ;
		while(j<m && arr[j] == arr[i]){
			sum += arr[j] ;
			j++ ;
		}
		stk.push_back({arr[i] , sum});
		i = j ;
	}
	int lst = 0 , mn = 0 , from = 0 , num , sz , prev , nw , cur ;
	// printf("--------\n") ;
	for(int i=0;i<stk.size();i++){
		from = 0 ; lst = 0 ; mn = 0 ;
		num = stk[i].fi ; sz = stk[i].se ; 
		if(i > 0 ) prev = stk[i-1].fi ;
		// printf("%d : %d %d\n" , i , num ,sz) ;
		for(j=0;j<i;j++){
			if(stk[j].se == 0) continue ;
			cur = stk[j].fi ;
			nw = qr(rt[prev] , rt[num] , 1 , n , cur) - qr(rt[prev] , rt[num] , 1 , n , lst) ;
			if(nw + from >= stk[j].se){
				nw -= stk[j].se ;
				stk[j].se = 0 ;
				from += nw ; 
			}
			else{
				stk[j].se -= (nw + from) ;
				from = 0 ;
			}
			// if(from < 0 || stk[j].se < 0) printf("%d",1/0);
			lst = stk[j].fi ;
			mn += stk[j].se ;
			// printf("---> %d : %d %d\n" , j , stk[j].fi , stk[j].se) ; 
		}
		// printf("%d : %d %d\n" , i , mn , qs[num]) ;
		if(qs[num] - mn < sz) return 0 ;
	}

	return 1;
}

Compilation message (stderr)

teams.cpp: In function 'void init(int, int*, int*)':
teams.cpp:54:11: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
   while( j<srt.size() && i >= srt[j].fi ) {
           ^
teams.cpp: In function 'int can(int, int*)':
teams.cpp:89:15: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
  for(int i=0;i<stk.size();i++){
               ^

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