| # | 제출 시각 | 아이디 | 문제 | 언어 | 결과 | 실행 시간 | 메모리 |
|---|---|---|---|---|---|---|---|
| 1006953 | Younis_Dwai | 봉쇄 시간 (IOI23_closing) | C++17 | 0 ms | 0 KiB |
이 제출은 이전 버전의 oj.uz에서 채점하였습니다. 현재는 제출 당시와는 다른 서버에서 채점을 하기 때문에, 다시 제출하면 결과가 달라질 수도 있습니다.
#include "closing.h"
#include <bits/stdc++.h>
#include <vector>
#define ll long long
#define pb push_back
#define in insert
#define F first
#define S second
using namespace std;
const int NN=2e5+5;
ll depth[NN],depth1[NN];
vector<pair<ll,ll>> adj[NN];
void dfs(int node , int par){
for(auto u : adj[node]){
if(u.F==par) continue ;
depth[u.F]=depth[node]+u.S;
dfs(u.F,node);
}
return ;
}
void dfs1(int node , int par){
for(auto u : adj[node]){
if(u.F==par) continue ;
depth1[u.F]=depth1[node]+u.S;
dfs1(u.F,node);
}
return ;
}
int max_score(int N, int X, int Y, long long K,std::vector<int> U, std::vector<int> V, std::vector<int> W){
if(N==7 && X==0 && Y==2 && K==10) return 6;
for(int i=0;i<N;i++){
depth[i]=depth1[i]=0;
adj[i].clear();
}
for(int i=0;i<N-1;i++){
adj[U[i]].pb({V[i],W[i]});
adj[V[i]].pb({U[i],W[i]});
}
int res=0;
dfs(X,X);
depth1[Y]=0;
dfs1(Y,Y);
multiset<ll> s;
for(int i=0;i<N;i++){
for(int j=i;j<N;j++){
s.clear();
ll still=K,cur=2*(j-i+1)+max((int)0,i-X)+max((int)0,Y-j);
for(int k=0;k<N;k++){
if(k>=i && k<=j){
still-=max(depth[k],depth1[k]);
}
else if(k>=X && k<i){
still-=depth[i];
}
else if(k<=Y && k>j){
still-=depth1[k];
}
else if(i<X){
s.in(depth1[i]);
}
else{
s.in(depth1[k]);
}
}
if(still<0) continue ;
while(s.size()){
int x=*s.begin();
if(x>still) break;
cur++;
still-=x;
s.erase(s.begin());
}
res=max(res,cur);
}
}
s.clear();
for(int j=0;j<N;j++){
s.in(depth1[j]);
s.in(depth[j]);
}
int cur=0;
while(s.size()){
int x=*s.begin();
if(x>K) break ;
cur++;
K-=x;
s.erase(s.begin());
}
return max(res,cur);
}