Submission #540856

#	Time	Username	Problem	Language	Result	Execution time	Memory
540856		GioChkhaidze	Teams (IOI15_teams)	C++14	0 / 100	4097 ms	52376 KiB

teams

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

#define pb push_back

using namespace std;

const int Nn = 5e5 + 5;

int n, a[Nn], b[Nn], f[Nn];
vector < int > G[Nn];

void init(int N, int A[], int B[]) {
	n = N;
	for (int i = 0; i < n; ++i) {
		a[i] = A[i], b[i] = B[i];
	}
	for (int i = 0; i < N; ++i) {
		G[A[i]].pb(i);
	}
}

int can(int M, int K[]) {
	for (int i = 0; i < M; ++i) {
		f[K[i]]++;
	}
	multiset < int > st;
	bool ok = true;
	for (int i = 1; i <= n; ++i) {
		for (int j = 0; j < G[i].size(); ++j) {
			int id = G[i][j];
			st.insert(b[id]);
		}
		for (int j = 1; j <= 1ll * f[i] * i; ++j) {
			if (!st.size()) {
				ok = false;
				break;
			}
			if ((*st.begin()) < i) {
				ok = false;
				break;
			}
			st.erase(st.find(*st.begin()));
		}
		if(!ok) break;
	}
	for (int i = 0; i < M; ++i) {
		f[K[i]] = 0;
	}
	return ok;
}

Compilation message (stderr)

teams.cpp: In function 'int can(int, int*)':
teams.cpp:30:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   30 |   for (int j = 0; j < G[i].size(); ++j) {
      |                   ~~^~~~~~~~~~~~~

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