正解思路，感觉应该是先判特殊情况，比如新建的路线可以建在于1-n无关的地方，具体判别方法见代码
然后再判断旧路线两端的最大贡献值，即代码中的ans，枚举1-n线路中的每个节点i，求1-[1,i]节点的最大值和[i+1,n]-n节点的最大值，维护ans即可
#include<iostream> #include<cstdio> #include<cmath> #include<queue> #include<vector> #include<string.h> #include<cstring> #include<algorithm> #include<set> #include<map> #include<fstream> #include<cstdlib> #include<ctime> #include<list> #include<climits> using namespace std; #define fio ios::sync_with_stdio(false);cin.tie(0);cout.tie(0); #define fopen freopen("input.txt", "r", stdin);freopen("output.txt", "w", stdout); #define scd(a) scanf("%d",&a) #define scf(a) scanf("%lf",&a) #define scl(a) scanf("%lld",&a) #define sci(a) scanf("%I64d",&a) #define scs(a) scanf("%s",a) #define left asfdasdasdfasdfasfasdfasfsdfasfdas typedef long long ll; const int desll[4][2]={{0,1},{0,-1},{1,0},{-1,0}}; const ll mod=1e9+7; const int maxn=3e5+7; const int maxm=1e8+7; const double eps=1e-4; int m,n; int ar[maxn]; set<pair<ll,int> > se1,se2; pair<int,pair<ll,int> > pal1[maxn],pal2[maxn]; int le1,le2; vector<pair<int,int> > e[maxn]; int markl[maxn],mark; set<pair<int,int> > seMa; ll ans=0,ansMid; int ansl[maxn]; void dfs(int u,int pre) { if(u==n){ mark=1; markl[u]=1; return ; } for(int i=0;i<e[u].size();i++){ int v=e[u][i].first; if(v==pre)continue; dfs(v,u); if(mark){ markl[u]=1; return ; } } } void dfs1(int u,int pre,ll sum,int ins) { pal1[le1].first=ins; pal1[le1].second.first=sum; pal1[le1++].second.second=u; if(u==n){ ansMid=sum; } for(int i=0;i<e[u].size();i++){ int v=e[u][i].first; if(v==pre)continue; dfs1(v,u,sum+e[u][i].second,ins+markl[v]); } } void dfs2(int u,int pre,ll sum,int ins) { pal2[le2].first=ins; pal2[le2].second.first=sum; pal2[le2++].second.second=u; for(int i=0;i<e[u].size();i++){ int v=e[u][i].first; if(v==pre)continue; dfs2(v,u,sum+e[u][i].second,ins+markl[v]); } } int main() { scd(n); scd(m); for(int i=1;i<n;i++){ int a,b,c; scd(a);scd(b);scd(c); e[a].push_back(make_pair(b,c)); e[b].push_back(make_pair(a,c)); seMa.insert(make_pair(a,b)); seMa.insert(make_pair(b,a)); } memset(markl,0,sizeof(markl)); mark=0; le1 = le2 = 0; ans=0,ansMid=0; dfs(1,0);//求1-n所经过的各个节点 dfs1(1,0,0,0); dfs2(n,0,0,0); sort(pal1,pal1+le1); sort(pal2,pal2+le2); for(int i=1;i<=n;i++){//看新建线路能否建在与1-n无关的地方，若能，其实后面的while维护ans可省略 int ins=0; if(markl[i]){ for(int j=0;j<e[i].size();j++){ int x=e[i][j].first; if(markl[x]==0)ins++; } } else{ ins=e[i].size(); } if(ins>=2){ ans=ansMid; break; } } int maxNum=pal1[le1-1].first; int be=0; for(int i=0;i<le2;i++){ if(pal2[i].first==maxNum)break; se2.insert(pal2[i].second); be=i; } int ins=0,res=0; while(res<maxNum){//维护ans while(pal1[ins].first<=res){ se1.insert(pal1[ins].second); ins++; } auto au1=se1.end(); auto au2=se2.end(); au1--; au2--; if(seMa.count(make_pair(au1->second,au2->second))){//若两个节点本身相连，即都在1-n的主线路上，则需退一步判次大值 if(au1!=se1.begin()){ au1--; ans = max(ans, au1->first+au2->first); au1++; } if(au2!=se2.begin()){ au2--; ans = max(ans, au1->first+au2->first); au2++; } } else ans = max(ans, au1->first+au2->first); res++; while(pal2[be].first+res==maxNum){ se2.erase(pal2[be].second); be--; } } //cout<<maxNum<<" "<<ansMid<<" "<<ans<<endl; for(int i=0;i<m;i++){ int ins; scd(ins); printf("%I64d ",min(ansMid,ans+ins)); } return 0; }