CF 682C Alyona and the Tree

codeforces 682C

题意：n个节点，每个节点有一个权值， 节点之间有（n-1）条边连着，构成一个树，输入n-1行，每行有两个数y和w，第i行代表节点（i+1）与节点y之间的边权值为w。要求：节点u到它的祖先节点的路径长度<=节点u的权值，否则删掉u节点及以u为根节点的树的所有节点，问：最少需要删多少节点。

样例：

（4～1的距离）67 > 14（4的权值），删去节点；

（5～1的距离）64 <= 95（5的权值），满足，节点5不删；

（7～5的距离）12 <= 98（7的权值），满足，并且（7～1的距离）76 <= 98（7的权值），满足，节点7不删；

2～3不满足，2～7满足，2～5不满足，2～1不满足，只要有一个不满足就要删去，节点2删去；

同理，节点9删去，同时节点9之下的6、8都要删去。

结果为5。

思路：深搜，从根节点1开始，遍历节点的子节点，遍历每个节点时，记录下来该节点到其祖先节点的最大值，再与该节点权值比较，判断是否需要删。

代码：

#include <algorithm>
#include <iostream>
#include <cstring>
#include <cstdlib>
#include <sstream>
#include <cstdio>
#include <vector>
#include <string>
#include <cmath>
#include <stack>
#include <queue>
#include <map>
#include <set>
#define INF 0x3f3f3f3f
#define fori(a,b) for(LL i=a;i<=b;i++)
#define forj(a,b) for(LL j=a;j<=b;j++)
#define mem(a,b) memset(a,b,sizeof(a))
using namespace std;
typedef long long LL;
const double PI = acos(-1);
const LL M = 1e5+10;
//const LL MM = 1e5+10;
LL n,m,ans=0;
LL a[M],del[M],vis[M];
//vis[i]=1 代表节点i已经搜过
//del[i]=1 代表节点i删去
map<int,map<int ,int> >mp,len;
vector<int>v[M];

void _fs(int x,int y,LL mmax)
{
    vis[x]=1;
    if(vis[y])
        return ;
    if(mmax>a[y]){  
        del[y]=1;     //不满足，删去节点i
        return ;
    }
    if(!v[y].size())
        return ;
    for(int i=0;i<v[y].size();i++){
        if(!vis[v[y][i]]){
            //更新最大值，即如果mmax为负数，mmax更新为mp[y][v[y][i]]，否则mmax=mmax+mp[y][v[y][i]]
            LL tmmax=max(mmax+mp[y][v[y][i]],(LL)mp[y][v[y][i]]);  
            _fs(y,v[y][i],tmmax);
        }
    }
    return ;
}

void ccount(int n)
{
    vis[n]=1;                //标记
    if(!v[n].size())
        return ;
    for(int i=0;i<v[n].size();i++){
        if(!vis[v[n][i]]){
            ans++;
            ccount(v[n][i]);
        }
    }
    return ;
}
int main()
{
    //freopen("//home//acm//桌面//in","r",stdin);
    mem(del,0);
    mem(vnum,0);
    mem(vis,0);
    cin >> n;
    for(int i=1;i<=n;i++)
        cin >> a[i] ;
    for(int i=2;i<=n;i++){
        int y,w;
        cin >> y >> w;
        v[i].push_back(y);
        v[y].push_back(i);
        mp[y][i]=mp[i][y]=w;
    }
    
    for(int i=0;i<v[1].size();i++){
        LL mmax=mp[1][v[1][i]];
        _fs(1,v[1][i],mmax);
    }
    for(int i=2;i<=n;i++){
        if(del[i]){
            ans++;
            ccount(i);
        }
    }
    cout << ans << endl;
    return 0;
}