Submission #1065250

#	Time	Username	Problem	Language	Result	Execution time	Memory
1065250		pcc	Closing Time (IOI23_closing)	C++17	21 / 100	1092 ms	27684 KiB

closing

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

#define pll pair<ll,ll>
#define pii pair<int,int>
#define fs first
#define sc second
#define ll long long
const int mxn = 2e5+10;

vector<pii> tree[mxn];
ll dp[mxn],arr[mxn],brr[mxn],need[mxn];
int N;
int X,Y;

void init(){
	for(int i = 0;i<=N;i++)tree[i].clear();
}

void dfs(int now,int par){
	if(now == par)dp[now] = 0;
	for(auto [nxt,w]:tree[now]){
		if(nxt == par)continue;
		dp[nxt] = dp[now]+w;
		dfs(nxt,now);
	}
	return;
}

int check(ll lim){
	for(int i = N;i>=1;i--)need[i] = need[i-1];
	need[0] = 0;
	for(int i = 1;i<=N;i++)need[i] += need[i-1];
	int rp = N;
	int re = 0;
	for(int i = Y;i>=0;i--){
		while(need[rp]-need[i]>lim)rp--;
		if(rp<=Y)break;
		re = max(re,rp-i);
		//cerr<<lim<<":"<<i<<','<<rp<<endl;
	}
	return re;
}

int max_score(int NN, int XX, int YY, long long K,
              std::vector<int> U, std::vector<int> V, std::vector<int> W){
	X = XX,Y = YY;
	init();
	N = NN;
	for(int i = 0;i<U.size();i++){
		int a = U[i],b = V[i],w = W[i];
		tree[a].push_back(pii(b,w));
		tree[b].push_back(pii(a,w));
	}
	dfs(X,X);
	for(int i = 0;i<N;i++)arr[i] = dp[i];
	dfs(Y,Y);
	for(int i = 0;i<N;i++)brr[i] = dp[i];
	//for(int i = 0;i<N;i++)cerr<<arr[i]<<' ';cerr<<endl;
	//for(int i = 0;i<N;i++)cerr<<brr[i]<<' ';cerr<<endl;
	int ans = 0;
	for(int i = 0;i<=X;i++){
		for(int j = X;j<N;j++){
			ll sum = 0;
			for(int k = 0;k<N;k++)need[k] = brr[k];
			for(int k = i;k<=j;k++){
				sum += arr[k];
				need[k] = max(0ll,brr[k]-arr[k]);
			}
			if(sum>K)continue;
			ans = max(ans,j-i+1+check(K-sum));
		}
	}
	return ans;
}

Compilation message (stderr)

closing.cpp: In function 'int max_score(int, int, int, long long int, std::vector<int>, std::vector<int>, std::vector<int>)':
closing.cpp:52:17: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   52 |  for(int i = 0;i<U.size();i++){
      |                ~^~~~~~~~~

• Subtask 0: 0 / 0
• Subtask 1: 0 / 8
• Subtask 2: 9 / 9
• Subtask 3: 12 / 12
• Subtask 4: 0 / 14
• Subtask 5: 0 / 9
• Subtask 6: 0 / 11
• Subtask 7: 0 / 10
• Subtask 8: 0 / 10
• Subtask 9: 0 / 17