【无标题】

今天，学的最小生成树，拿洛谷p3366练习了一下，自己写的代码一直tle,给我整无语了。

结果发现错误

int findx(int x)
{
    while (a[x] != x)
    {
        a[x] = findx(a[x]);
    }
    return a[x];
}

不能用while，应该用if,正确写法是

int findx(int x) {
    if (a[x] != x) {
        a[x] = findx(a[x]); 
    }
    return a[x];
}

给我记住了！！！！！！！！！！！！！！！！！！

还有一个问题

这是ac代码

#include <iostream>
#include <algorithm>
using namespace std;

int a[5005];
int findx(int x) {
    if (a[x] != x) {
        a[x] = findx(a[x]); // 修正：使用递归调用时需要返回结果
    }
    return a[x];
}

void merge(int x, int y) {
    int fx = findx(x);
    int fy = findx(y);
    if (fx != fy) {
        a[fy] = fx; // 修正：将y的根节点设置为x的根节点
    }
}

struct node {
    int x, y, z;
};

bool cmp(const node &a, const node &b) {
    return a.z < b.z; // 修正：使用const修饰cmp函数的参数
}

int main() {
    int n, m, ans = 0, cnt = 0;
    cin >> n >> m;
    
    for (int i = 1; i <= n; i++) {
        a[i] = i; // 每个顶点的初始根节点是它自己
    }

    node l[200005];
    for (int i = 1; i <= m; i++) {
        cin >> l[i].x >> l[i].y >> l[i].z;
    }
    sort(l + 1, l + m + 1, cmp); // 修正：从1开始排序

    for (int i = 1; i <= m; i++) {
        int fx = findx(l[i].x);
        int fy = findx(l[i].y);
        if (fx != fy) {
            ans += l[i].z;
            merge(l[i].x, l[i].y); // 使用merge函数
        }
    }

    for (int i = 1; i <= n; i++) { // 从1开始检查，因为0号元素未使用
        if (a[i] == i) {
            cnt++;
        }
    }

    if (cnt == 1) {
        cout << ans << endl;
    } else {
        cout << "orz" << endl;
    }
    return 0;
}

这是wa代码

#include <bits/stdc++.h>
using namespace std;
int a[5005];
int findx(int x)
{
    if (a[x] != x)
    {
        a[x] = findx(a[x]);
    }
    return a[x];
}

struct node
{
    int x, y, z;
} l[200005];
bool cmp(node a, node b)
{
    return a.z < b.z;
}
int main()
{
    int n, m, ans = 0, cnt = 0;
    cin >> n >> m;
    for (int i = 1; i <= n; i++)
    {
        a[i] = i;
    }

    for (int i = 1; i <= m; i++)
    {
        cin >> l[i].x >> l[i].y >> l[i].z;
    }
    sort(l + 1, l + m + 1, cmp);
    for (int i = 1; i <= m; i++)
    {
        int fx = findx(l[i].x);
        int fy = findx(l[i].y);
        if (fx != fy)
        {
            ans += l[i].z;
            a[l[i].x] = l[i].y;
        }
    }
    for (int i = 1; i <= n; i++)
    {
        if (a[i] = i)
            cnt++;
    }
    if (cnt == 1)
        cout << ans << endl;
    else
        cout << "orz" << endl;
    return 0;
}

我这个并查集的合并不知道哪里出了问题