poj 1988 Cube Stacking(带权并查集)

Cube Stacking
题意：
农夫John和Betsy在玩一个游戏，有N(1≤N≤30,000)块相同的立方，
标记从1到N。开始时是N个栈，每个栈只有一个立方体。农夫John
请Betsy执行P(1≤P≤100,000)个操作，有两类操作：move和count。
在一个move操作中，农夫John请Bessie将包含立方体X的栈移到包
含立方体Y的栈的栈顶。
在一个count操作中，农夫John请Bessie计算包含立方体X的栈中在X
下的立方体个数，并返回值。
请您编写一个程序返回游戏结果。

思路：
多开两个数组，记录某点所在栈中方块数量以及某点下面有多少数量方块
然后在进行合并节点的时候去更新这两个数组，注意，在查询的时候，要先找一次该点的父节点，因为不确定该点下面有多少数量方块是否更新过

我写了一个非递归的查找和一个递归的查找，但是非递归的wa了，找到了错误，但是不会改。。目前只会 带权并查集 的递归查找压缩路径 TNT

先贴一下非递归的查找函数

int Find(int root)
{
    int son =  root;
    while(root != f[root]){
        root = f[root];
    }
    while(son != root){
        int temp = f[son];
        v[son] += v[temp];
        f[son] = root;
        son = temp;
    }
    return root;
}

下面是递归查找函数

#include <cstdio>
#include <cstring>

const int maxn = 30000 + 5;

int f[maxn],s[maxn],v[maxn];
//f记录父节点，s表示某点所在栈中的方块数量，v表示某节点下面有几个方块

void init()
{
    for(int i = 1; i < maxn; i++){
        f[i] = i;
        s[i] = 1;
    }
    memset(v,0,sizeof(v));
}

int Find(int root)
{
    if(root != f[root]){
        int t = f[root];
        f[root] = Find(t);
        v[root] += v[t];
    }
    return f[root];
}

void add(int p,int q)
{
    f[p] = q;//表示p根节点更新后其根节点为q
    //旧根节点p下面为q所在栈的方块数
    v[p] = s[q];
    //更新后的根节点q所在栈的数量为为更新前p+q的方块数
    s[q] += s[p];
}

int main()
{
    init();
    int n;
    scanf("%d",&n);
    for(int i = 0; i < n; i++){
        char s[2];
        int a,b;
        scanf("%s",s);
        if(s[0] == 'M'){
            scanf("%d%d",&a,&b);
            int p = Find(a),q = Find(b);
            if(p != q){
                add(p,q);
            }
        }
        else{
            scanf("%d",&a);
            Find(a);
            printf("%d\n",v[a]);
        }
    }
    return 0;
}