Submission #1033001

#	Time	Username	Problem	Language	Result	Execution time	Memory
1033001		socpite	Teams (IOI15_teams)	C++17	77 / 100	4025 ms	115400 KiB

teams

#include "teams.h"
#include<bits/stdc++.h>
using namespace std;

const int maxn = 5e5+5;
const long long INF = 1e18;

vector<int> FW[maxn];
vector<int> d[maxn];

int pfx_all[maxn], sfx_all[maxn];
int L[maxn], R[maxn];
long long dp[maxn];

int n;

void add(int x, int y){
	for(x; x <= n; x += x&(-x)){
		for(int idx = lower_bound(d[x].begin(), d[x].end(), y) - d[x].begin(); idx < d[x].size(); idx += idx&(-idx))FW[x][idx]++;
	}
}

int gt(int x, int y){
	int re = 0;
	for(x; x > 0; x -= x&(-x)){
		for(int idx = upper_bound(d[x].begin(), d[x].end(), y) - d[x].begin() - 1; idx > 0; idx -= idx&(-idx))re += FW[x][idx];
	}
	return re;
}

int gt_gap(int l, int r){
	if(l > r)return 0;
	return pfx_all[r] - gt(r, l-1);
}

void init(int N, int A[], int B[]) {
	n = N;
	for(int i = 0; i < n; i++){
		L[i] = A[i];
		R[i] = B[i];
		for(int x = B[i]; x <= n; x += x&(-x))d[x].push_back(A[i]);
		pfx_all[B[i]]++;
		sfx_all[A[i]]++;
	}
	for(int i = 1; i <= n; i++){
		pfx_all[i] += pfx_all[i-1];

		d[i].push_back(-1);
		sort(d[i].begin(), d[i].end());
		d[i].erase(unique(d[i].begin(), d[i].end()), d[i].end());
		FW[i].assign(d[i].size(), 0);
	}
	for(int i = n; i >= 1; i--)sfx_all[i] += sfx_all[i+1];
	for(int i = 0; i < n; i++)add(B[i], A[i]);
}

int can(int M, int K[]) {
	long long pfx_M = 0, pfx_gap = 0, mn = INF;
	sort(K, K+M);
	for(int i = 0; i < M; i++){
		dp[i] = K[i] + pfx_all[K[i]-1];
		for(int j = 0; j < i; j++){
			dp[i] = max(dp[i], dp[j] + K[i] + gt_gap(K[j]+1,K[i]-1));
		}
		if(dp[i] + sfx_all[K[i]+1] > n)return 0;
	}
	return 1;
	// sum_segment - sum_M < 0 => bad
	// sum_segment = pfx_all - sumgap
	
}

Compilation message (stderr)

teams.cpp: In function 'void add(int, int)':
teams.cpp:18:6: warning: statement has no effect [-Wunused-value]
   18 |  for(x; x <= n; x += x&(-x)){
      |      ^
teams.cpp:19:58: warning: conversion from '__gnu_cxx::__normal_iterator<int*, std::vector<int> >::difference_type' {aka 'long int'} to 'int' may change value [-Wconversion]
   19 |   for(int idx = lower_bound(d[x].begin(), d[x].end(), y) - d[x].begin(); idx < d[x].size(); idx += idx&(-idx))FW[x][idx]++;
      |                 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~~~~~~~~~~~~~
teams.cpp:19:78: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   19 |   for(int idx = lower_bound(d[x].begin(), d[x].end(), y) - d[x].begin(); idx < d[x].size(); idx += idx&(-idx))FW[x][idx]++;
      |                                                                          ~~~~^~~~~~~~~~~~~
teams.cpp: In function 'int gt(int, int)':
teams.cpp:25:6: warning: statement has no effect [-Wunused-value]
   25 |  for(x; x > 0; x -= x&(-x)){
      |      ^
teams.cpp:26:73: warning: conversion from '__gnu_cxx::__normal_iterator<int*, std::vector<int> >::difference_type' {aka 'long int'} to 'int' may change value [-Wconversion]
   26 |   for(int idx = upper_bound(d[x].begin(), d[x].end(), y) - d[x].begin() - 1; idx > 0; idx -= idx&(-idx))re += FW[x][idx];
      |                 ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~^~~
teams.cpp: In function 'int can(int, int*)':
teams.cpp:58:12: warning: unused variable 'pfx_M' [-Wunused-variable]
   58 |  long long pfx_M = 0, pfx_gap = 0, mn = INF;
      |            ^~~~~
teams.cpp:58:23: warning: unused variable 'pfx_gap' [-Wunused-variable]
   58 |  long long pfx_M = 0, pfx_gap = 0, mn = INF;
      |                       ^~~~~~~
teams.cpp:58:36: warning: unused variable 'mn' [-Wunused-variable]
   58 |  long long pfx_M = 0, pfx_gap = 0, mn = INF;
      |                                    ^~

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