제출 #1194286

#	제출 시각	아이디	문제	언어	결과	실행 시간	메모리
1194286		Ak_16	Closing Time (IOI23_closing)	C++20	0 / 100	85 ms	28996 KiB

closing

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
#define ll long long

struct edg{
  int no;
  int len;
};

int tot1;

vector<edg> adj[200005];
ll disa[210000];
ll disb[210000];
int vis[200005];

ll cos1[210000];
ll cos2[210000];
ll ne[210000];

void dfs1(int nod, ll di){
  vis[nod] = 1;
  disa[nod] = di;
  for(auto edge : adj[nod]){
    int n1 = edge.no;
    int n2 = edge.len;
    if(vis[n1]==0){
      dfs1(n1, di+n2);
    }
  }
  
}

void dfs2(int nod, int di){
  vis[nod] = 1;
  disb[nod] = di;
  for(auto edge : adj[nod]){
    int n1 = edge.no;
    int n2 = edge.len;
    if(vis[n1]==0){
      dfs2(n1, di+n2);
    }
  }
  
}



int max_score(int n, int x, int y, ll k, vector<int> u, vector<int> v, vector<int> w){
  
  for(int i=0; i<n-1; i++){
    adj[u[i]].emplace_back(edg{v[i],w[i]});
    adj[v[i]].emplace_back(edg{u[i],w[i]});
  }
  
  for(int i=0; i<n; i++){
    vis[i]=0;
    disa[i]=0;
    disb[i]=0;
  }
  
  dfs1(x, 0);
  
  for(int i=0; i<n; i++){
    vis[i]=0;
  }
  
  dfs2(y,0);
  
  for(int i=0; i<n; i++){
    cos1[i] = min(disa[i], disb[i]);
    cos2[i] = max(disa[i], disb[i]);
  }
  
  for(int i=0; i<n; i++){
    ne[i] = cos1[i];
  }
  
  sort(ne, ne+n);
  

  tot1=0;
  
  ll sm=0;
  for(int i=0; i<n; i++){
    if(sm+ne[i]<=k){
      sm += ne[i]; tot1++;
    }
    else {break;}
  }
  
  
  
  return tot1;
  
}

• Subtask 0: 0 / 0
• Subtask 1: 0 / 8
• Subtask 2: 0 / 9
• Subtask 3: 0 / 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