标签：tmp 15 int 2019.7 T3 tot && 直径 define
你看到这种东西就想到树的直径。
然后随便口胡反证一下就可以知道就是两种情况：一种是一条是直径，另一条是除去直径后的子树的最大直径，另一种是直径扣掉一段，然后扣掉的两个端点分别向下挂两条链。
第一种情况跑出直径随便处理，第二种情况预处理前缀\(max\)随便搞搞。时间复杂度\(O(n)\)
code:
#include<bits/stdc++.h>
#define I inline
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define abs(x) ((x)>0?(x):-(x))
#define re register
#define ll long long
#define db double
#define N 200000
#define M 50000
#define mod 1000000000
#define mod2 39989
#define eps (1e-7)
#define U unsigned int
#define it iterator
#define Gc() getchar()
#define Me(x,y) memset(x,y,sizeof(x))
#define R(x) (rand()*rand()%x+1)
using namespace std;
int n,m,k,x,y,fl[N+5],Id[N+5],Ih,Ids[N+5],Is,W[N+5],now,tot,fa[N+5],lcas,nowx,nowy;ll pus,ans,d[N+5],de[N+5],Maxn[N+5],Q[N+5];
struct yyy{int to,z;}tmp;
struct ljb{
int head,h[N+5];yyy f[N+5<<1];
I void add(int x,int y){f[++head]=(yyy){y,h[x]};h[x]=head;}
}s;
I void dfs(int x,int last){
yyy tmp;fa[x]=last;d[x]=d[last]+W[x];de[x]=de[last]+1;d[x]>=d[tot]&&(tot=x);for(int i=s.h[x];i;i=tmp.z) tmp=s.f[i],tmp.to^last&&!fl[tmp.to]&&(dfs(tmp.to,x),0);
}
I void swap(int &x,int &y){x^=y^=x^=y;}
int main(){
freopen("attack.in","r",stdin);freopen("attack.out","w",stdout);
re int i,j;scanf("%d",&n);for(i=1;i<=n;i++) scanf("%d",&W[i]);for(i=1;i<n;i++) scanf("%d%d",&x,&y),s.add(x,y),s.add(y,x);
dfs(1,0);now=tot;d[tot]=0;dfs(tot,0);ans=d[tot];nowx=now;nowy=tot;de[nowx]<de[nowy]&&(swap(nowx,nowy),0);while(de[nowx]^de[nowy]) nowx=fa[nowx];
while(nowx^nowy) nowx=fa[nowx],nowy=fa[nowy];lcas=nowx;nowx=now;while(nowx^lcas) Id[++Ih]=nowx,nowx=fa[nowx];Id[++Ih]=lcas;
nowx=tot;while(nowx^lcas) Ids[++Is]=nowx,nowx=fa[nowx];for(i=Is;i;i--) Id[++Ih]=Ids[i];
for(i=1;i<=Ih;i++) fl[Id[i]]=1;
for(i=1;i<=Ih;i++){
for(j=s.h[Id[i]];j;j=tmp.z){
tmp=s.f[j];if(fl[tmp.to]) continue;tot=tmp.to;d[tot]=0;dfs(tot,0);Maxn[i]=max(Maxn[i],d[tot]);
now=tot;d[tot]=0;dfs(tot,0);pus=max(pus,d[tot]+ans);
}Q[i]=Maxn[i]+d[Id[i]];
}
for(i=1;i<=Ih;i++) Q[i]=max(Q[i-1],Q[i]),pus=max(pus,Q[i-1]+Maxn[i]+ans-d[Id[i-1]]);printf("%lld\n",pus);
}
标签：tmp,15,int,2019.7,T3,tot,&&,直径,define 来源： https://www.cnblogs.com/275307894a/p/15017475.html
本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享； 2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关； 3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关； 4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除； 5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。