제출 #1329422

#제출 시각아이디문제언어결과실행 시간메모리
1329422santi3223Travelling Trader (CCO23_day2problem2)C++20
11 / 25
214 ms31508 KiB
#include <bits/stdc++.h>
using namespace std;
#define ll long long
#define vl vector<ll>
#define ff(aa, bb, cc) for(ll aa = bb; aa < cc; aa++)
#define pb push_back
#define all(aaa) aaa.begin(), aaa.end()
#define rall(aaa) aaa.rbegin(), aaa.rend()
#define pll pair<ll, ll>
#define fi first
#define se second
#define ed "\n"

vector<vl> conexiones;
vl val, route;

ll dfs1(ll cur, ll par){
	ll maxx = 0;
	for(auto &x : conexiones[cur]){
		if(x != par){
			ll res = dfs1(x, cur);
			if(res > maxx){
				maxx = res;
				route[cur] = x;
			}
		}
	}
	if(maxx == 0) route[cur] = cur;
	return maxx+val[cur];
}

ll steps;

void dfs3(ll cur, ll par){
	bool visited = false;
	if(steps == 3){
		route.pb(cur);
		steps = 0;
		visited = true;
	}
	for(auto &x : conexiones[cur]){
		if(x != par){
			steps++;
			dfs3(x, cur);
		}
	}
	if(!visited){
		steps = 0;
		route.pb(cur);
	}
	steps++;
	
}

int main(){
	ll n, k;
	cin >> n >> k;
	conexiones = vector<vl>(n);
	val = vl(n);
	ff(i, 0, n-1){
		ll a, b;
		cin >> a >> b;
		a--;
		b--;
		conexiones[a].pb(b);
		conexiones[b].pb(a);
	}
	ll suma = 0;
	ff(i, 0, n){
		cin >> val[i];
		suma += val[i];
	}
	if(k == 1){
		route = vl(n);
		ll maxx = dfs1(0, -1);
		vl arr;
		ll cur = 0;
		while(route[cur] != cur){
			arr.pb(cur);
			cur = route[cur];
		}
		arr.pb(cur);
		cout << maxx << ed << arr.size() << ed;
		for(auto &a : arr) cout << a+1 << " ";
		return 0;
	}
	if(k == 2){
		
	}
	if(k == 3){
		steps = 3;
		dfs3(0, -1);
		cout << suma << ed << route.size() << ed;
		ff(i, 0, route.size()){
			cout << route[i]+1 << " ";
		}
		return 0;
	}
}
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...
#Verdict Execution timeMemoryGrader output
Fetching results...