有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
李陶冶
(题目提供者)
从后往前将重物挂上去
因为当处理到i时,所有i的孩子都已经挂在i上了
这时如果以i为根的子树重量sum[i]>c[i]
则从最后一个重物开始删除重物 直到sum[i]<=c[i]
如果使用并查集维护par[i]=i的根
每次删除x 就等价于sum[find(x)]-=w[x]
每个重物最多最多被挂一次,被删一次 复杂度O(n)
#include<stdio.h>
#include <bits/stdc++.h>
#define ll long long
#define pii pair<int,int>
#define MEM(a,x) memset(a,x,sizeof(a))
using namespace std;
const int inf=1e9+7;
const int N = 5e4+5;
int par[N],c[N],w[N],p[N];
ll sum[N];//以i为根的子树重量和
int find(int x){//找x的根
return x==par[x]?x:(par[x]=find(par[x]));
}
int slove(int n){
int ans=n;
for(int i=n;i>=1;--i){
while(sum[i]>c[i]){
int x=find(ans);
sum[x]-=w[ans];
--ans;
}
sum[p[i]]+=sum[i];//挂i到p[i]上
par[i]=p[i];
}
return ans;
}
int main()
{
//freopen("/home/lu/code/r.txt","r",stdin);
int n;
while(~scanf("%d",&n)){
MEM(sum,0);
for(int i=1;i<=n;++i){
par[i]=i;
scanf("%d%d%d",&c[i],&w[i],&p[i]);
sum[i]+=w[i];
++p[i];
}
printf("%d\n",slove(n));
}
return 0;
}