51nod 1307 绳子与重物 【二分/并查集】

题目：http://www.51nod.com/onlineJudge/questionCode.html#!problemId=1307

题意：

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

分析：

二分+dfs判断，O（nlogn）
看讨论还有O(n)的解法，用并查集做。从叶子向根统计子树和，如果当前子树和大于根的负重，就从最后一个添加的点开始删除。
参考博客：
http://blog.csdn.net/fsss_7/article/details/50545374

代码：

二分：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=5e4+9;
int c[N],w[N],p[N];
ll sonw[N];
vector<int> g[N];
bool dfs(int u)
{
    sonw[u]=w[u];
    for(int i=0;i<g[u].size();i++){
        int v=g[u][i];
        if(!dfs(v))return 0;
        sonw[u]+=sonw[v];
    }
    return sonw[u]<=c[u];
}
bool ok(int m)
{
    for(int i=0;i<=m;i++)g[i].clear();
    for(int i=1;i<=m;i++){
        g[p[i]].push_back(i);
    }
    return dfs(0);
}
int main() {

    //freopen("in.txt", "r", stdin);
    int n;
    scanf("%d",&n);
    c[0]=1e18,w[0]=0;
    for(int i=1;i<=n;i++)scanf("%d%d%d",&c[i],&w[i],&p[i]),p[i]++;
    int l=1,r=n;
   // cout<<ok(3);
    while(l<r){
        int m=l+(r-l+1)/2;
        if(ok(m))l=m;
        else r=m-1;
    }
    printf("%d\n",l);
    return 0;
}

并查集：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=5e4+9;
ll c[N],sumw[N],w[N];
int k,n,p[N],fa[N];
int tot,to[N],h[N],pre[N];
void add(int x,int y) {
    to[tot]=y;pre[tot]=h[x];h[x]=tot++;
}
int find_fa(int x) {
    if (fa[x]==x) return x;
    return fa[x]=find_fa(fa[x]);
}
void updata(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_fa(k)]-=w[k];k--;
    }
}
int main()
{
   // freopen("in.txt","r",stdin);
    scanf("%d", &n);
    tot=0;
    memset(h,-1,sizeof(h));
    for (int i=1;i<=n;i++) {
        scanf("%lld%lld%d", &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--) updata(i);
    printf("%d\n", k);
    return 0;
}