牛客算法周练2 D题 树建立 思维

**D. 小H和游戏**

链接:(https://ac.nowcoder.com/acm/contest/5203/D)

题目描述

小H正在玩一个战略类游戏,她可以操纵己方的飞机对敌国的N座城市(编号为1~N)进行轰炸
敌国的城市形成了一棵树,小H会依次进行Q次轰炸,每次会选择一个城市A进行轰炸,和这座城市距离不超过2的城市都会受损(这里距离的定义是两点最短路径上的边数),轰炸结束后,小H还想知道当前城市A受损的次数
作为游戏的开发者之一,你有义务回答小H的问题

输入描述:
第1行,两个整数N(1≤N≤750000)、Q(1≤Q≤750000)
第2~N行,每行两个整数表示树上的一条边
第N+1~N+Q行,每行一个整数,表示小H这次轰炸的城市
输出描述:
输出Q行,每行一个整数表示这一次轰炸的城市在此次轰炸后共计受损几次

#include<bits/stdc++.h>
#define ll long long

using namespace std;

int fa[750001];
vector<int>G;
int fath[75000][3]; // 0代表自己受伤的次数
// 1代表它的儿子被波及的次数    //2代表自己孙子被波及的次数
void dfs(int u,int f){   //建树

    fa[u]=f;
    for (int i=0;i<G[u].size();i++){
        int v=G[u][i];
        if(f==v) continue;   //防止出现 fa[2]=1 fa[1]=2  建树
        dfs(v,u);
    }

}
/*
1 2 
2 3 
3 4 建成的树
      0
      |
      1
     / \
    3   2
   / \  
  4   5
  
*/
int main()
{
    int n,q;
    cin>>n>>q;
    for (int i=0;i<n-1;i++){
        int a,b;
        cin>>a>>b;
        G[a].push_back(b);
        G[b].push_back(a);
    }
    dfs(1,0); //将最大的根节点的父亲设为0
    while(q--){
        ll ans=0;
        int m;
        cin>>m;
        fath[m][1]++;  //伤害波及到它的儿子
        fath[m][2]++;   //伤害波及到它的孙子
        fath[fath[m]][0]++; //伤害波及到它的父亲
        fath[fath[m]][1]++; //伤害波及它父亲的儿子(包括自己)   
        //就是父亲它的儿子被其他儿子波及以及父亲自己被伤害时波及它儿子总次数  
        //也就是它的每个儿子 所受伤的总次
        fath[fath[fath[m]]][0]++; //伤害波及到它的爷爷
        ans=fath[m][0]+fath[fath[m]][1]+fath[fath[fath[m]]][2];
        //fath[m][0]是由它的孙子波及的以及儿子波及的 
        //fath[fath[m]][1] 是它的父亲伤害的次数以及它父亲其它儿子波及到它总数
        //fath[fath[fath[m]]][2] 它的爷爷自己被攻击次数 (不包括爷爷被)
        cout<<ans;
    }
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值