Submission #1059036

#	Time	Username	Problem	Language	Result	Execution time	Memory
1059036		MuhammadSaram	Teams (IOI15_teams)	C++17	0 / 100	742 ms	22964 KiB

teams

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

using namespace std;

const int M = 1e5 + 1;

#define all(v) v.begin(), v.end()

int n,pre[M];
vector<int> V,seg[M*2];

void make(int v=1,int s=0,int e=n)
{
	for (int i=s;i<e;i++)
		seg[v].push_back(V[i]);
	sort(all(seg[v]));
	if (s+1==e)
		return;
	int mid=(s+e)/2,lc=v+1,rc=v+(mid-s)*2;
	make(lc,s,mid);
	make(rc,mid,e);
}

int get(int l,int r,int x,int v=1,int s=0,int e=n)
{
	if (l>=e or r<=s)
		return 0;
	if (l<=s && e<=r)
	{
		int y=upper_bound(all(seg[v]),x)-begin(seg[v])+1;
		return y;
	}
	int mid=(s+e)/2,lc=v+1,rc=v+(mid-s)*2;
	return get(l,r,x,lc,s,mid)+get(l,r,x,rc,mid,e);
}

void init(int N,int A[],int B[])
{
	n=N;
	vector<pair<int,int>> v1;
	for (int i=0;i<n;i++)
		v1.push_back({B[i],A[i]}),pre[B[i]]++;
	sort(v1.begin(),v1.end());
	for (auto i:v1)
		V.push_back(i.second);
	for (int i=1;i<M;i++)
		pre[i]+=pre[i-1];
	make();
}

int can(int m,int k[])
{
	map<int,int> cnt;
	for (int i=0;i<m;i++)
		cnt[k[i]]++;
	vector<pair<int,int>> v;
	for (auto i:cnt)
		v.push_back(i);
	m=v.size();
	int val=0;
	for (int i=0;i<m;i++)
	{
		int s=pre[v[i].first-1],e=n+1;
		while (s+1<e)
		{
			int mid=(s+e)/2;
			if (get(pre[v[i].first-1],mid,v[i].first)>=val+v[i].first*v[i].second)
				e=mid;
			else
				s=mid;
		}
		if (e==n+1)
			return 0;
		if (i!=m-1)
			val=get(pre[v[i+1].first-1],e,v[i].first);
	}
	return 1;
}

Compilation message (stderr)

teams.cpp: In function 'int get(int, int, int, int, int, int)':
teams.cpp:31:49: warning: conversion from '__gnu_cxx::__normal_iterator<int*, std::vector<int> >::difference_type' {aka 'long int'} to 'int' may change value [-Wconversion]
   31 |   int y=upper_bound(all(seg[v]),x)-begin(seg[v])+1;
      |         ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~
teams.cpp: In function 'int can(int, int*)':
teams.cpp:60:10: warning: conversion from 'std::vector<std::pair<int, int> >::size_type' {aka 'long unsigned int'} to 'int' may change value [-Wconversion]
   60 |  m=v.size();
      |    ~~~~~~^~

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