题意:一个树上建两个加油站,使得所有点到达其最近加油站的最大距离最小。
解法:二分答案。关键时二分时候,要最合理话布局两个点的位置,做法是处理出来树的直径,然后在直径两端分别向中间移动二分的x步的两个点布下加油站。贪心可以证明正确性;
代码:
/****************************************************** * @author:xiefubao *******************************************************/ #pragma comment(linker, "/STACK:102400000,102400000") #include <iostream> #include <cstring> #include <cstdlib> #include <cstdio> #include <queue> #include <vector> #include <algorithm> #include <cmath> #include <map> #include <set> #include <stack> #include <string.h> //freopen ("in.txt" , "r" , stdin); using namespace std; #define eps 1e-8 #define zero(_) (_<=eps) const double pi=acos(-1.0); typedef long long LL; const int Max=200010; const LL INF=0x3FFFFFFF; vector<int> vec[Max]; int n; vector<int> diameter; int rem[Max]; int rem1[Max]; int rem2[Max]; int dis[Max]; int st,en; int dui[Max*2]; int now=0; int count1=10; void dfs(int u) { memset(rem,0,sizeof rem); rem[u]=-1; int left=0; int right=1; dui[0]=u; dis[u]=0; while(left<right) { for(int i=0; i<vec[dui[left]].size(); i++) { if(rem[vec[dui[left]][i]]==0) { rem[vec[dui[left]][i]]=dui[left]; dis[vec[dui[left]][i]]=dis[dui[left]]+1; dui[right++]=vec[dui[left]][i]; } } left++; } } void ran1(int u,int* re) { int left=0; int right=1; dui[0]=u; dis[u]=0; re[u]=count1; while(left<right) { if(dis[dui[left]]<now) for(int i=0; i<vec[dui[left]].size(); i++) { if(re[vec[dui[left]][i]]!=count1) { re[vec[dui[left]][i]]=count1; dis[vec[dui[left]][i]]=dis[dui[left]]+1; if(dis[vec[dui[left]][i]]<now) dui[right++]=vec[dui[left]][i]; } } left++; } } bool OK(int middle) { count1++; now=middle; ran1(diameter[middle],rem1); ran1(diameter[diameter.size()-1-middle],rem2); for(int i=1; i<=n; i++) { if(rem1[i]!=count1&&rem2[i]!=count1) return false; } return true; } int main() { int t; cin>>t; while(t--) { for(int i=1;i<Max;i++) vec[i].clear(); diameter.clear(); scanf("%d",&n); for(int i=0; i<n-1; i++) { int a,b; scanf("%d%d",&a,&b); vec[a].push_back(b); vec[b].push_back(a); } dfs(1); int d=0; for(int i=1; i<=n; i++) { if(dis[i]>d) { d=dis[i]; st=i; } } dfs(st); d=0; for(int i=1; i<=n; i++) { if(dis[i]>d) { d=dis[i]; en=i; } } while(en!=-1) { diameter.push_back(en); en=rem[en]; } int left=0,right=diameter.size(); while(left<=right) { int middle=(left+right)/2; if(!OK(middle)) { left=middle+1; } else { right=middle-1; } } int a=diameter[left]; int b=diameter[diameter.size()-1-left]; if(a==b) { a=b==n?n-1:b+1; } //cout<<diameter.size()<<" "<<OK(0)<<OK(1)<<endl; cout<<left<<" "<<a<<" "<<b<<'\n'; } return 0; } /* 34 4 1 2 1 3 1 4 5 1 2 2 3 3 4 4 5 3 1 2 2 3 */
时间: 2024-12-11 05:01:42