BZOJ1912: [Apio2010]patrol 巡逻 树的直径

最致命的bug是没有bug。
题面里最后一张图：加两条边以后答案为15 我跑出来一直是10 于是与空气斗智斗勇半个小时+
k = 1的时候不难发现求个直径就完事了
k = 2的时候就大力讨论 不会有新加的边在第一次加的边形成的环中使答案最优
重复统计的地方 边权搞成-1完事了
以及第二次有了负权边就不能dfs/bfs求直径了 得dp 因为这个wa了好久。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define Max(_A, _B) (_A > _B ? _A : _B)
#define Min(_A, _B) (_A < _B ? _A : _B)
const int maxn = 1e5 + 7;
int ans, n, k, head[maxn], eid, pre[maxn], dis[maxn], S, T, f[maxn], g;
bool vis[maxn];
struct edge{
    int v, nxt, w;
}e[maxn << 1];
void init(){memset(head, -1, sizeof(head)); eid = 0; }
void insert(int u, int v, int w) {e[eid].v = v; e[eid].nxt = head[u]; e[eid].w = w; head[u] = eid++;}
int read(){
    int s = 0; char c = getchar();
    while (c > '9' || c < '0') c = getchar();
    while (c >= '0' && c <= '9') {s = s * 10 + c - '0'; c = getchar();}
    return s;
}
void dfs(int u, int fa){
    //puts("2");
    for (int i = head[u]; ~i; i = e[i].nxt){
        int v = e[i].v;
        if (v == fa) continue;
        pre[v] = i;
        dis[v] = dis[u] + e[i].w;
        if (dis[v] > dis[S]) S = v;
        dfs(v, u);
    }
}
void dfs2(int u){
    vis[u] = 1;
    //puts("1");
    for (int i = head[u]; ~i; i = e[i].nxt) {
        int v = e[i].v;
        if (vis[v]) continue;
        dfs2(v);
        g = Max(g, f[u] + f[v] + e[i].w);
        f[u] = Max(f[u], f[v] + e[i].w);
    }
}
int main(){
    n = read(), k = read(); ans = 2 * (n - 1); 
    init();
    for (int i = 1; i < n; i++) {
        int a = read(), b = read();
        insert(a, b, 1); insert(b, a, 1);
    }
    dfs(1, 1);
   // for (int i = 1; i <= n; i++) if(dis[S] < dis[i]) S = i;
    T = S;
    memset(dis, 0, sizeof(dis));
    dfs(S, T = S);
    ans -= dis[S] - 1;
    if (k == 2) {
        for (int i = S; i != T; i = e[pre[i] ^ 1].v) {
            e[pre[i]].w = e[pre[i]^1].w = -1;
        }
        dfs2(1);
        ans -= g - 1;
    }
    //for (int i = 1; i <= n; i++) if(dis[S] < dis[i]) S = i;
    printf("%d\n", ans);
    return 0;
}