本蒟蒻又来发题解了

这题是不是有点像并查集,但是那个询问的个数是不是有点骚;

所以,普通的并查集是无法解决这个问题的,这个时候就需要用到带权并查集了;

每次跑的时候都记录下它的下面有几个点,然后询问的时候的复杂度是O(1)的;

好,接下来就可以愉快的码代码了:

#include<bits/stdc++.h>
using namespace std;

#define maxn 500010
char ch;
int n, x, y, fa[maxn], sum[maxn], num[maxn];

//比常规的并查集多了一行 
int find(int x)
{
    if(fa[x] == x)
        return x;
    sum[x] += sum[fa[x]];//还需要统计它下面的积木数 
    return fa[x] = find(fa[x]);
}

//fa[]就不说了,sum[]是来存它下面的积木数,num[]是这一列的积木数; 
int main()
{
    scanf("%d", &n);
    for(int i = 1; i <= maxn; ++ i)
    {
        fa[i] = i;//嗯,它的祖先就是它自己,常规操作; 
        sum[i] = 0;//它的下面QwQ肯定没有积木啊 
        num[i] = 1;//目前这一列只有一个积木 
    }
    for(int i = 1; i <= n; ++ i)
    {
        cin >> ch;
        if(ch == 'M')//如果这是'M',那就要合并
        {
            scanf("%d%d", &x, &y);
            x = find(x), y = find(y);
            if(x != y)//如果他们的祖先不同,就是他们不在同一个联通块上 
            {
                fa[x] = y;//合并它们 
                sum[x] = num[y]; //数量要相加
                num[y] += num[x]; //因为x要和y合并
            }
        }
        else
        {
            scanf("%d", &x);
            find(x);//还需要再来一次;
            printf("%d\n", sum[x]);//O(1)输出; 
        }
    }
    return 0;
}

嗯,你愉快的拿了27分;

好,开始查错了,

戴上眼镜,睁大眼睛

哪里有错?

就是过不去数据,哎;

好吧,我猜这么神仙的你们肯定找到哪里错了,不像蒟蒻我(划掉

#include<bits/stdc++.h>
using namespace std;

#define maxn 500010
char ch;
int n, x, y, fa[maxn], sum[maxn], num[maxn];
//fa[]就不说了,sum[]是来存它下面的积木数,num[]是这一列的积木数; 

//比常规的并查集多了一行 
int find(int x)
{
    if(fa[x] == x)
        return x;
    int f = find(fa[x]);
    sum[x] += sum[fa[x]];//还需要统计它下面的积木数 
    return fa[x] = f;
}

int main()
{
    scanf("%d", &n);
    for(int i = 1; i <= maxn; ++ i)
    {
        fa[i] = i;//嗯,它的祖先就是它自己,常规操作; 
        sum[i] = 0;//它的下面QwQ肯定没有积木啊 
        num[i] = 1;//目前这一列只有一个积木 
    }
    for(int i = 1; i <= n; ++ i)
    {
        cin >> ch;
        if(ch == 'M')//如果这是'M',那就要合并
        {
            scanf("%d%d", &x, &y);
            x = find(x), y = find(y);
            if(x != y)//如果他们的祖先不同,就是他们不在同一个联通块上 
            {
                fa[x] = y;//合并它们 
                sum[x] = num[y]; //数量要相加
                num[y] += num[x]; //因为x要和y合并
            }
        }
        else
        {
            scanf("%d", &x);
            find(x);//还需要再来一次;
            printf("%d\n", sum[x]);//O(1)输出; 
        }
    }
    return 0;
}

有没有发现哪里错了,

是的,就是在find函数里面;

第一个代码是:

int find(int x)
{
    if(fa[x] == x)
        return x;
    sum[x] += sum[fa[x]];//还需要统计它下面的积木数 
    return fa[x] = find(fa[x]);
}

第二个代码是:

int find(int x)
{
    if(fa[x] == x)
        return x;
    int f = find(fa[x]);
    sum[x] += sum[fa[x]];//还需要统计它下面的积木数 
    return fa[x] = f;
}

是吧,第一个是先更新sum[],再递归;

而第二个是先递归,再更新sum[];

所以,我们应该是要先递归完,再来更新;

如果要先更新的话,那这个find函数对与sum[]的更新就没有作用;

嗯,就这么结束了;

PS. AC的代码是第二篇,第一篇有问题;