oj.uz

Submission #847235

# Submission time Handle Problem Language Result Execution time Memory
847235 2023-09-09T10:04:31 Z iskhakkutbilim Closing Time (IOI23_closing) C++17
8 / 100
 137 ms 37056 KB 
#include "closing.h"
#include <bits/stdc++.h>
using namespace std;
#define ff first
#define ss second
const int MAXN = 2e5;
vector<int> parent;
void dfs(int v, int par, long long deep, vector<long long> &dist, 
	vector< vector<pair<int, int> > >& g){
	dist[v] = deep;
	parent[v] = par;
	for(auto [to, w] : g[v]){
		if(to == par) continue;
		dfs(to, v, deep + w, dist, g);
	}
}


int max_score(int N, int X, int Y, long long K,
              vector<int> U, vector<int> V, vector<int> W){
    vector< vector<pair<int, int> > > g(N);
    parent.clear();
    parent.resize(N);
    vector<long long> dist(N), dist1(N);
    int ans = 0;         
	for(int i = 0;i < N-1; i++){
    	g[U[i]].push_back({V[i], W[i]});
		g[V[i]].push_back({U[i], W[i]});
	}
//	if(X > Y) swap(X,Y);
    dfs(X, X, 0LL, dist, g);
    dfs(Y, Y, 0LL, dist1, g);
    if(dist[Y] > 2*K){
    	vector<long long> times;
    	for(int i = 0;i < N; i++) times.push_back(dist[i]);
    	
    	for(int i = 0;i < N; i++) times.push_back(dist1[i]);
    	long long s = 0;
		sort(times.begin(), times.end());
		for(long long cost : times){
			if(s > K-cost) break;
			s+= cost;
			ans++;
		}
		return ans;
	}
	vector<long long> comp;
	for(int i = 1; i <= N; i++)comp.push_back(min(dist[i-1], dist1[i-1]));
    sort(comp.begin(), comp.end());
    long long curr = 0;
    for(int i = 0; i < N; i++){
        if(curr + comp[i] <= K){
            curr += comp[i];
			 ans++;
        }
    }
	int node = X;
	vector<int> in(N + 10, 0);
	do{
		in[node + 1] = 1;
		node = parent[node];
	}while(node != Y);
	in[Y + 1] = 1;
	vector< vector<long long> > dp(N+10, vector<long long>(N*2+10, INT_MAX));
	dp[0][0] = 0;
	for(int i = 0;i < N; i++){
		for(int j = 0;j <= 2*N; j++){
			if(in[i+1]){
				dp[i+1][j+1] = min(dp[i+1][j+1], dp[i][j] + min(dist[i+1], dist1[i+1]));
				dp[i+1][j+2] = min(dp[i+1][j+2], dp[i][j] + max(dist[i+1], dist1[i+1]));
			}else{
				dp[i+1][j] = min(dp[i+1][j], dp[i][j]);
				dp[i+1][j+1] = min(dp[i+1][j+1], dp[i][j] + min(dist[i+1], dist1[i+1]));
				dp[i+1][j+2] = min(dp[i+1][j+2], dp[i][j] + max(dist[i+1], dist1[i+1]));
			}
		}
	}
	for(int j = 0;j <= 2*N; j++){
		if(dp[N][j] <= K) ans = max(ans, j);
	}
	return ans;
}


//main(){
//	int N, X, Y; cin >> N >> X >> Y;
//	long long K; cin >> K;
//	vector<int> U, V, W;
//	for(int i = 0;i < N-1; i++){
//		int a, b, w; cin >> a >> b >> w;
//		U.push_back(a);
//		V.push_back(b);
//		W.push_back(w);
//	}
//	cout << max_score(N, X, Y, K, U, V, W) << '\n';
//	return 0;
//}

Subtask #0
0.0 / 0.0

# Verdict Execution time Memory Grader output
1 Incorrect 0 ms 344 KB 1st lines differ - on the 1st token, expected: '6', found: '5'
2 Halted 0 ms 0 KB -

Subtask #1
8.0 / 8.0

# Verdict Execution time Memory Grader output
1 Correct 125 ms 31972 KB Output is correct
2 Correct 137 ms 37056 KB Output is correct
3 Correct 75 ms 2896 KB Output is correct

Subtask #2
0.0 / 9.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Incorrect 1 ms 344 KB 1st lines differ - on the 1st token, expected: '30', found: '32'
3 Halted 0 ms 0 KB -

Subtask #3
0.0 / 12.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Incorrect 1 ms 344 KB 1st lines differ - on the 1st token, expected: '30', found: '32'
3 Halted 0 ms 0 KB -

Subtask #4
0.0 / 14.0

# Verdict Execution time Memory Grader output
1 Correct 0 ms 344 KB Output is correct
2 Incorrect 1 ms 344 KB 1st lines differ - on the 1st token, expected: '30', found: '32'
3 Halted 0 ms 0 KB -

Subtask #5
0.0 / 9.0

# Verdict Execution time Memory Grader output
1 Incorrect 0 ms 344 KB 1st lines differ - on the 1st token, expected: '6', found: '5'
2 Halted 0 ms 0 KB -

Subtask #6
0.0 / 11.0

# Verdict Execution time Memory Grader output
1 Incorrect 0 ms 344 KB 1st lines differ - on the 1st token, expected: '6', found: '5'
2 Halted 0 ms 0 KB -

Subtask #7
0.0 / 10.0

# Verdict Execution time Memory Grader output
1 Incorrect 0 ms 344 KB 1st lines differ - on the 1st token, expected: '6', found: '5'
2 Halted 0 ms 0 KB -

Subtask #8
0.0 / 10.0

# Verdict Execution time Memory Grader output
1 Incorrect 0 ms 344 KB 1st lines differ - on the 1st token, expected: '6', found: '5'
2 Halted 0 ms 0 KB -

Subtask #9
0.0 / 17.0

# Verdict Execution time Memory Grader output
1 Incorrect 0 ms 344 KB 1st lines differ - on the 1st token, expected: '6', found: '5'
2 Halted 0 ms 0 KB -