#include <bits/stdc++.h> #define int long long #de...

你这份代码里最可能导致答案错误的核心 bug 有 4 个，其中第 1、2 个是实质性逻辑错误。

---

1. dfs() 中 dis[vto] 赋值顺序错了

你现在写的是：

cpp
res+=dfs(vto,x)+wto.second;
siz[x]+=siz[vto];
dis[vto]=dis[x]+wto.first+wto.second;

问题是：dfs(vto, x) 执行时，vto 的 dis[vto] 还没有被赋值。
所以 vto 的子节点、孙子节点计算 dis 时会用到错误的 dis[vto]。

应改为：

cpp
int dfs(int x,int fa) {
    pa[x]=fa;
    int res=0;
    siz[x]=1;

    for (int i=0;i<v[x].size();i++) {
        int vto=v[x][i];
        pii wto=w[x][i];
        if (vto==fa) continue;

        dis[vto]=dis[x]+wto.first+wto.second; // 必须放在递归前

        res+=dfs(vto,x)+wto.second;
        siz[x]+=siz[vto];
    }
    return res;
}

这是一个非常关键的 bug。

---

2. dfs4() 中把父节点也标成了 istop

你现在写的是：

cpp
for (int i=0;i<nv[x].size();i++) {
    if (nv[x][i]!=lson[x]) istop[nv[x][i]]=1;
}

这里没有排除父节点 fa，所以遍历到非根节点时，会把父亲也标记为 istop，导致后面优先队列 q 里塞进很多错误链。

应改为：

cpp
for (int i=0;i<nv[x].size();i++) {
    int vto=nv[x][i];
    if (vto==fa) continue;
    if (vto!=lson[x]) istop[vto]=1;
}

完整修正版：

cpp
void dfs4(int x,int fa,int ww) {
    diss[x]=diss[fa]+ww;
    dismax[x]=diss[x];
    lson[x]=0;

    for (int i=0;i<nv[x].size();i++) {
        int vto=nv[x][i], wto=nw[x][i];
        if (vto==fa) continue;

        dfs4(vto,x,wto);

        dismax[x]=max(dismax[x],dismax[vto]);
        if (lson[x]==0 || dismax[lson[x]]<dismax[vto]) {
            lson[x]=vto;
        }
    }

    for (int i=0;i<nv[x].size();i++) {
        int vto=nv[x][i];
        if (vto==fa) continue;
        if (vto!=lson[x]) istop[vto]=1;
    }
}

---

3. 你标记 answu 到 answv 路径的方式不严谨，会漏掉 LCA

你现在的写法是：

cpp
while (answu!=0) tmp[answu]++,answu=pa[answu];
while (answv!=0) tmp[answv]++,answv=pa[answv];

for (int i=1;i<=n;i++) {
    if (i==lstu||i==lstv) tmp[i]=1;
    else {
        if (tmp[i]==1) tmp[i]=1;
        else tmp[i]=0;
    }
}

这个逻辑相当于保留两条根路径的对称差部分，但如果 answu 和 answv 的 LCA 不是端点，那么 LCA 会被标记两次，即 tmp[lca]==2，最后会变成 0。

如果你的目的是把 answu 到 answv 的整条路径缩成一个点，那么 LCA 应该被保留。

建议直接改成这样：

cpp
int u=lstu, vv=lstv;

static int mark[MAXN];

while (u) {
    mark[u]=1;
    u=pa[u];
}

while (!mark[vv]) {
    tmp[vv]=1;
    vv=pa[vv];
}

int lca=vv;
tmp[lca]=1;

u=lstu;
while (u!=lca) {
    tmp[u]=1;
    u=pa[u];
}

这样 tmp[i]==1 才是真正表示节点 i 在 lstu 到 lstv 的路径上。

---

4. 调试输出和 freopen 会导致提交 WA

你的 main() 里有这些调试代码：

cpp
freopen("in.txt","r",stdin);

以及：

cpp
for (int i=1;i<=n;i++) cout<<f[i]<<' ';
cout<<'\n';
for (int i=1;i<=n;i++) cout<<dis[i]<<' ';
cout<<'\n';

cout<<answu<<' '<<answv<<'\n';

正式提交时必须删掉，否则输出格式一定错。

改成：

cpp
signed main() {
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    cin>>n;
    ...
}

---

你最少需要改这几处

修改 dfs

cpp
int dfs(int x,int fa) {
    pa[x]=fa;
    int res=0;
    siz[x]=1;

    for (int i=0;i<v[x].size();i++) {
        int vto=v[x][i];
        pii wto=w[x][i];
        if (vto==fa) continue;

        dis[vto]=dis[x]+wto.first+wto.second;

        res+=dfs(vto,x)+wto.second;
        siz[x]+=siz[vto];
    }
    return res;
}

修改 dfs4

cpp
void dfs4(int x,int fa,int ww) {
    diss[x]=diss[fa]+ww;
    dismax[x]=diss[x];
    lson[x]=0;

    for (int i=0;i<nv[x].size();i++) {
        int vto=nv[x][i],wto=nw[x][i];
        if (vto==fa) continue;

        dfs4(vto,x,wto);

        dismax[x]=max(dismax[vto],dismax[x]);
        if (lson[x]==0 || dismax[lson[x]]<dismax[vto]) {
            lson[x]=vto;
        }
    }

    for (int i=0;i<nv[x].size();i++) {
        int vto=nv[x][i];
        if (vto==fa) continue;
        if (vto!=lson[x]) istop[vto]=1;
    }
}

修改路径标记

把原来的：

cpp
while (answu!=0) tmp[answu]++,answu=pa[answu];
while (answv!=0) tmp[answv]++,answv=pa[answv];
for (int i=1;i<=n;i++) {
    if (i==lstu||i==lstv) tmp[i]=1;
    else {
        if (tmp[i]==1) tmp[i]=1;
        else tmp[i]=0;
    }
}

换成：

cpp
static int mark[MAXN];

int u=lstu, vv=lstv;

while (u) {
    mark[u]=1;
    u=pa[u];
}

while (!mark[vv]) {
    tmp[vv]=1;
    vv=pa[vv];
}

int lca=vv;
tmp[lca]=1;

u=lstu;
while (u!=lca) {
    tmp[u]=1;
    u=pa[u];
}

---

另外还有一个潜在风险

你的 n 最大如果是 5e5，这些递归：

cpp
dfs
dfs2
dfs3
dfs4

在链状树上可能会爆栈。
如果本地运行直接 RE，但逻辑没错，那大概率就是递归栈爆了。正式稳妥做法是改成非递归 DFS，或者在支持的环境里手动扩大栈。

---

优先改这四点：
**dfs 的 dis 赋值顺序、dfs4 排除父节点、路径标记包含 LCA、删除调试输出。**这几个不改，答案很容易整体偏掉。