Redis - 2 資料型別
阿新 • • 發佈:2020-11-27
有生以來第一次A掉洛谷月賽的第二題
思路:貪心
首先看資料,\(M\le10^{18}\),\(O(m)\)暴力肯定會T,然後...
然後就鴿了。
在考場上死想了三十分鐘,終於想出來正解思路:
小\(S\)只會在兩個村莊內徘徊!(易證)
-
首先我們要從所有點中找出一個值最大且序號最靠前的,以及他所有子節點中值最大且序號最靠前的的點只要對這兩個點進行操作即可。
-
先使這兩個點的權值相等,然後在按編號排序,最後用奇偶性判斷該輸出哪個點就好了。
-
注意\(n==1\)特判要放在讀完資料後,還有每次要重置所有資料。
AC Code:
#include <stdio.h> #include <algorithm> #define N 2000005 using namespace std; typedef long long LL; LL ans, n, m, tot, v[N], maxn, maxto; int main(){ int T; scanf("%d", &T); while(T--){ maxn=maxto=0; scanf("%lld%lld", &n, &m); for(LL i=1; i<=n; i++){ scanf("%lld", &v[i]); if(v[maxn]<v[i]) maxn=i; } for(LL i=1; i<n; i++){ LL a, b; scanf("%lld%lld", &a, &b); if(a==maxn){ if(v[maxto]<v[b]||(v[maxto]==v[b]&&maxto>b)) maxto=b; } if(b==maxn){ if(v[maxto]<v[a]||(v[maxto]==v[a]&&maxto>a)) maxto=a; } } if(n==1){ printf("1\n"); continue; } m-=v[maxn]-v[maxto]; if(m<0) printf("%lld\n", maxn); else{ if(maxn>maxto) swap(maxn, maxto); if(m%2) printf("%lld\n", maxto); else printf("%lld\n", maxn); } } return 0; }