巡逻【树的直径】【树形dp】【bfs】

题目传送门

在最初的树上,求出树的直径L1 然后将这条路径上的所有边权统统取反.
1变成−1
我们再求一次树的直径L2L2.

答案就是
2×(n−1)−(L1−1)−(L2−1)=2×(n−1)−L1+1−L2+1
假如说L2L2和L1L1有重叠部分.

那么当我们
−L1+1
的时候,我们就会发现,重叠的部分变成了只需要经过一次.

然后
−L2+1
−L2+1

相当于把重叠部分相加回来了.

此时变成了经过了两次.

假如说没有重叠部分,那么1,-1都不会有影响,反正我们不会经过这些边.

#include<cstdio>
#include<iostream>
#include<cstring>
#include<queue>
using namespace std;
const int MAXX=100005;
queue<int>q;
int h[MAXX],e[MAXX*2],w[MAXX*2],nxt[MAXX*2];
int d[MAXX],pre[MAXX],f[MAXX];
bool v[MAXX];
int n,k,p,tot=1,ans;
void add(int x,int y,int z){
    e[++tot]=y;
    nxt[tot]=h[x];
    h[x]=tot;
    w[tot]=z;
}
int bfs(int s){
   memset(d,0x3f,sizeof(d));
   d[s]=0;
   pre[s]=0;
   q.push(s);
   while(!q.empty()){
    int x=q.front();
    q.pop();
    for(int i=h[x];i!=-1;i=nxt[i]){
        int y=e[i];
        if(d[y]==0x3f3f3f3f){
           pre[y]=i;
           d[y]=d[x]+w[i];
           q.push(y);
        }
      }
   }
   int x,y;
   for( x=y=1;x<=n;x++){
    if(d[x]>d[y])y=x;
   }//找到端点
   return y;
}
void dp(int x){
    v[x]=1;
    for(int i=h[x];i!=-1;i=nxt[i]){
        int y=e[i];
        if(v[y])continue;
        dp(y);
        ans=max(ans,f[x]+f[y]+w[i]);
        f[x]=max(f[x],f[y]+w[i]);
    }
}
void change(){
    for(;pre[p];p=e[pre[p]^1])w[pre[p]]=w[pre[p]^1]=-1;
//取反操作，ver[pre[p]^1],实际上是反边的终点，也就是p的前继
//因为tot=1；++tot，边从2开始记录，2^1=3，即2,3互为反边
}
int main(){
    memset(h,-1,sizeof h);
    cin>>n>>k;
    for(int i=1;i<=n-1;i++){
        int x,y;
        cin>>x>>y;
        add(x,y,1);
        add(y,x,1);
    }
    p=bfs(1);
    p=bfs(p);
    int l1=d[p];
    if(k==1){
        cout<<2*(n-1)-(l1-1)<<endl;
    }else {
        change();
        dp(1);
        cout<<2*(n-1)-(l1-1)-(ans-1)<<endl;
    }
    return 0;
}