51nod - 1307 绳子与重物

有N条绳子编号 0 至 N - 1，每条绳子后面栓了一个重物重量为Wi，绳子的最大负重为Ci。每条绳子或挂在别的绳子下或直接挂在钩子上（编号-1）。如果绳子下所有重物的重量大于绳子的最大负重就会断掉（等于不会断）。依次给出每条绳子的负重Ci、重物的重量Wi以及绳子会挂在之前的哪条绳子的下面，问最多挂多少个绳子而不会出现绳子断掉的情况。

 

例如下图：

 

5, 2, -1

3, 3, 0

6, 1, -1

3, 1, 0

3, 2, 3

 

 

 

挂到第4个时会有绳子断掉，所以输出3。

 

 

Input

第1行：1个数N，表示绳子的数量(1 <= N <= 50000)。
第2 - N + 1行：每行3个数，Ci, Wi, Pi，Ci表示最大负重，Wi表示重物的重量，Pi表示挂在哪个绳子上，如果直接挂在钩子上则Pi = -1（1 <= Ci <= 10^9，1 <= Wi <= 10^9，-1 <= Pi <= N - 2)。

Output

输出1个数，最多挂到第几个绳子，不会出现绳子断掉的情况。

Input示例

5
5 2 -1
3 3 0
6 1 -1
3 1 0
3 2 3

Output示例

3

思路：

用链表记录每个节点悬挂的子节点。最后遍历时从后向前，不断更新每个节点的父节点。

#include <iostream>
#include <cstring>
using namespace std;
 
typedef long long ll;
const int MAXN = 5e4 + 10;
ll c[MAXN], sumW[MAXN], w[MAXN];
int k, n, p[MAXN], fa[MAXN];
int tot, to[MAXN], h[MAXN], pre[MAXN];
 
void add(int x, int y)
{
    to[tot] = y;
    pre[tot] = h[x];
    h[x] = tot;
    tot++;
}
 
int find(int x)
{
    if (fa[x] == x)
    {
        return x;
    }
    else
    {
        fa[x] = find(fa[x]);
    }
    
    return fa[x];
}

void update(int x)
{
    for (int i = h[x]; i != -1; i = pre[i])
    {
        sumW[x] += sumW[to[i]];
        fa[to[i]] = x;
    }
    
    while (sumW[x] > c[x])
    {
        sumW[find(k)] -= w[k];
        k--;
    }
}

int main()
{
	cin >> n;
    tot = 0;
    memset(h, -1, sizeof(h));
    for (int i = 1; i <= n; i++) 
	{
		cin >> c[i] >> w[i] >> p[i];
        p[i]++;
		fa[i] = i;
		add(p[i], i);
		sumW[i] = w[i];
    }
    k = n;
    for (int i = n; i > 0; i--) 
	{
		update(i);
	}
    cout << k << endl;
	return 0;
}