蓝书例题----树的直径 ---- 巡逻

做法蓝书上已经很清楚了，主要的问题就是---

void update(int q,int p)
{
    while(q!=p)
    {
        edge[pre[q]]=-1;
        edge[pre[q]^1]=-1;    //无向边是有两条相邻的
        q=e[pre[q]^1];
    }
}

 为什么这样可以给直径上的点标为 - 1 ，原因在于 由于树是双向路径，因此，取异或时，相当于原本记录的idx相邻的那一项，也就是自身的反方向。

#include<bits/stdc++.h>
//树的深搜 
using namespace std;
typedef long long ll;
const int N = 100000 + 10, M = 2 * N;
ll ans = 0;
int n, k;
ll d[N];
bool st[N];
int h[N],e[M],ne[M],idx;
int  edge[N];
ll dis[N];
int pre[N];
queue<int> q;
void add (int a,int b,ll c)
{
    e[idx]=b;
    ne[idx]=h[a];
    edge[idx] = c;
    h[a]=idx++;
    
}
void dp(int x)
{
    st[x] = 1;
    for(int i = h[x] ; i != - 1 ; i = ne[i])
    {
    	 int y=e[i];
        if(st[y]) continue;
        dp(y);
    	ans = max(ans,d[x]+d[y]+edge[i]);
        d[x]=max(d[x],d[y]+edge[i]);
	}
}
 
int dfs(int u)
{
    memset(d,-1,sizeof d);
    d[u] = 0;
 
    q.push(u);
    while(q.size())
    {
        int t = q.front();
        q.pop();
        for(int i = h[t] ;i != -1 ;i = ne[i] )
        {
            int y = e[i];
            if(d[y] != -1) continue;
            else if(d[y] == -1)
            {
                d[y] = d[t] + 1;
                pre[y] = i;
                q.push(y);
            }
        }
        
    }
    int p = u;
    for(int i =  1 ; i <= n ; i++) if(d[i] > d[p]) p = i;
    return p;
}
void update(int q,int p)
{
    while(q!=p)
    {
        edge[pre[q]]=-1;
        edge[pre[q]^1]=-1;    //无向边是有两条相邻的
        q=e[pre[q]^1];
    }
}
 

int main()
{
    memset(h,-1,sizeof h);
    cin >> n >> k;
    int x,y;
    for(int i = 1 ; i < n ;i++)
    {
        cin >> x >> y;
        add(x,y,1);
        add(y,x,1);
    }
    int p = dfs(1);
    int q = dfs(p);
    ll l1 = d[q];
    // update
    if(k == 1) cout << 2 * (n - 1) - l1 + 1;
    else {
    update(q,p);
    memset(d,0,sizeof d);
    dp(1);
    cout << 2 * (n - 1) - l1 + 1 - ans + 1 << endl;
    }
    
}