題目鏈接

容易發現，當加一條邊時，樹上會形成一個環，這個環上的每個點都是只要走一次的，也就是說我們的答案減少了這個環上點的個數，要使答案最小，即要使環上的點最多，求出直徑\(L\)，則答案為\(2(n-1)-L+1\)。
當加兩條邊時，同樣會形成一個新環，但這個新環可能和第一個環有交點，而這些交點仍是要走兩次的，所以我們要讓交點的個數盡可能小，所以，把原直徑上的所有邊權取反，代表若取了這條邊，答案會增大那麽多，然後再求一次樹的直徑\(L_1\)，則答案為\(2(n-1)-L+1-L_1+1=2n-L-L_1\)。
註意，第二次求直徑不能用兩邊\(DFS/BFS\)來求，因為樹中有負權邊，直接跑答案顯然是錯的，所以我們要用樹形\(DP\)

#include <cstdio>
const int MAXN = 5000010;
namespace IO{
  inline int read(){
    int s = 0, w = 1;
    char ch = getchar();
    while(ch < '0' || ch > '9') { if(ch == '-') w = -1; ch = getchar(); }
    while(ch >= '0' && ch <= '9') { s = s * 10 + ch - '0'; ch = getchar(); }
    return s * w;
  }
}using namespace IO;
namespace G{
  struct Edge{
      int next, to, dis;
  }e[MAXN << 1];
  int head[MAXN], num;
  inline void Add(int from, int to, int dis){
      e[++num].to = to;
      e[num].dis = dis;
      e[num].next = head[from];
      head[from] = num;
  }
}using namespace G;
int n, k, s, t;
int a, b;
int pre[MAXN];
int Max = 0;
inline int max(int a, int b){
    return a > b ? a : b;
}
void dfs(int u, int fa, int dep){
    if(dep > Max && fa) s = u, Max = dep;
    for(int i = head[u]; i; i = e[i].next)
       if(e[i].to != fa)
         dfs(e[i].to, u, dep + e[i].dis);
}
void DFS(int u, int fa, int dep){
    if(dep > Max) t = u, Max = dep;
    for(int i = head[u]; i; i = e[i].next)
       if(e[i].to != fa)
         pre[e[i].to] = u, DFS(e[i].to, u, dep + e[i].dis);
}
int d[MAXN], ans = -2147483647;
void dp(int u, int fa){
    for(int i = head[u]; i; i = e[i].next)
       if(e[i].to != fa){
         dp(e[i].to, u);
         ans = max(ans, d[u] + d[e[i].to] + e[i].dis);
         d[u] = max(d[u], d[e[i].to] + e[i].dis); 
       }
}
int main(){
    n = read(); k = read();
    for(int i = 1; i < n; ++i){
       a = read(); b = read();
       Add(a, b, 1); Add(b, a, 1);
    }
    Max = -2147483647; dfs(1, 0, 0);
    Max = -2147483647; DFS(s, 0, 0); 
    if(k == 1){
      printf("%d\n", (n << 1) - 1 - Max);
      return 0;
    }
    int now = t;
    while(now != s){
      for(int i = head[now]; i; i = e[i].next)
         if(e[i].to == pre[now]){
           e[i].dis = -1;
           break;
         }
      now = pre[now];
    }
    dp(t, 0);
    printf("%d\n", (n << 1) - Max - ans);
    return 0;
}
 

【洛谷 P3629】 [APIO2010]巡邏