样例全对,但是提交全0分。求助
查看原帖
样例全对,但是提交全0分。求助
332549
幽灵特工楼主2023/10/5 21:27

RT,这道题也没有开放下载数据,我实在是调不出来了。。。求助

思路是找到好感值最高的点A,在和A点相邻的所有点中找到好感值最高的B,于是答案就在A、B之中。讨论一下m的奇偶性即可。

#include<bits/stdc++.h>
using namespace std;
struct edge {
	int from, to;
	edge(int a, int b) { from = a; to = b; }
};
vector<edge>G[2000003];
int n;long long m;
int t;
int a[2000006];
int maxi, maxn;
int main() {
	ios::sync_with_stdio(0);
	cin >> t;
	while (t--) {
		cin >> n >> m;
		
		for (int i = 1; i <= n; i++) {
			cin >> a[i];
			if (a[i] > maxn) {
				maxn = a[i];
				maxi = i;
			}
		}
		int x, y;
		for (int i = 0; i < n - 1; i++) {
			cin >> x >> y;
			G[x].push_back(edge(x, y));
			G[y].push_back(edge(y, x));
		}
		int mn = -1, mi = -1;//找B点
		for (int i = 0; i < G[maxi].size(); i++) {
			if (a[G[maxi][i].to] > mn) {
				mn = a[G[maxi][i].to];
				mi = G[maxi][i].to;
			}
		}
		m -= a[maxi] - a[mi];
		if (m > 0)a[mi] = a[maxi];
		else {
			cout << maxi << endl;
			for (int i = 1; i <= n; i++)G[i].clear();
			continue;
		}
		if (mi < maxi) {
			if (m % 2 == 0)cout << mi << endl;
			else cout << maxi << endl;
		}
		else {
			if (m % 2 == 1)cout << mi << endl;
			else cout << maxi << endl;
		}
		for (int i = 1; i <= n; i++)G[i].clear();
	}
}
2023/10/5 21:27
加载中...