标签：rt int 题解 APIO2010 mx2 dfs mx1 P3629 ans
\(k=1\) 的时候就是边数 \(\times 2\) 再减去直径长度。考虑 \(k=2\) 的时候，减去的相当于是选出的两条路径的对称差。考虑一个贪心，第一次取某一条直径一定不会更劣，这个感性理解就好。然后把这个直径的边权都设成 \(-1\) 再跑一遍直径就行了。
点击查看代码
const int N=1e5+13;
struct Edge{int v,nxt;}e[N<<1];
int n,k,h[N],etot=1;
inline void add_edge(int u,int v){e[++etot]=(Edge){v,h[u]};h[u]=etot;}
namespace sub1{
int ans,dep[N];
void dfs(int u,int fa){
dep[u]=dep[fa]+1;
for(int i=h[u];i;i=e[i].nxt){
int v=e[i].v;if(v==fa) continue;
dfs(v,u);
}
}
inline void solve(){
dfs(1,0);
int rt=1;
for(int i=2;i<=n;++i)
if(dep[i]>dep[rt]) rt=i;
dfs(rt,0);
for(int i=1;i<=n;++i)
if(dep[i]-1>ans) ans=dep[i]-1;
println(2*(n-1)-ans+1);
}
}
namespace sub2{
int ans,dep[N],mx1[N],mx2[N];
bool vis[N<<1];
void dfs(int u,int fa){
dep[u]=dep[fa]+1;
for(int i=h[u];i;i=e[i].nxt){
int v=e[i].v;if(v==fa) continue;
dfs(v,u);
}
}
bool dfs1(int u,int fa,int ed){
if(u==ed) return 1;
for(int i=h[u];i;i=e[i].nxt){
int v=e[i].v;if(v==fa) continue;
if(dfs1(v,u,ed)) return vis[i]=vis[i^1]=1;
}
return 0;
}
void dfs2(int u,int fa){
mx1[u]=0,mx2[u]=-INF;
for(int i=h[u];i;i=e[i].nxt){
int v=e[i].v;if(v==fa) continue;
int w=(vis[i]?-1:1);
dfs2(v,u);
if(mx1[v]+w>mx1[u]) mx2[u]=mx1[u],mx1[u]=mx1[v]+w;
else if(mx1[v]+w>mx2[u]) mx2[u]=mx1[v]+w;
}
ans=max(ans,mx1[u]+mx2[u]);
}
inline void solve(){
dfs(1,0);
int x=1;
for(int i=2;i<=n;++i)
if(dep[i]>dep[x]) x=i;
dfs(x,0);
int y=1;
for(int i=2;i<=n;++i)
if(dep[i]>dep[y]) y=i;
dfs1(x,0,y);
dfs2(1,0);
println(2*(n-1)-(ans+dep[y]-1)+2);
}
}
int main(){
read(n),read(k);
for(int i=1;i<n;++i){int u,v;read(u),read(v);add_edge(u,v),add_edge(v,u);}
if(k==1) sub1::solve();
else sub2::solve();
return 0;
}
标签：rt,int,题解,APIO2010,mx2,dfs,mx1,P3629,ans 来源： https://www.cnblogs.com/winterfrost/p/p3629-solution.html
本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享； 2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关； 3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关； 4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除； 5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。