第十屆省賽 B題.情報傳遞

chanchancl發表於2017-05-07
原文網址 : https://blog.csdn.net/chanchancl/article/details/71374192

題目：



這道題是昨天熱身賽A的的擴充。
這道題裡面有一些東西需要根據樣例的輸出來獲取
比如： 0 也要向上級傳遞資訊
          這個樹的關係是確定的，但通道並不一定建立
          這可以通過樣例的前5條推斷出來

是一道需要動點腦筋的題，把樣例給的圖（樹）畫出來，

分析輸出的值，來確定準確的題意。 

程式碼：


#include<iostream>
#include<vector>
#include<queue>
#include<string.h>
#include<cmath>
#include<string>
#include<set>
#include<list>
#include<stack>
#include<cctype>
#include<functional>
#include<algorithm>
#include<iostream>
#include<stdio.h>
#include<sstream>
usingnamespace std;
 
int n, m;
int up[5100];
int build[5100];
vector<int> down[5100];
 
int Send(int s)
{
    int ret = 0;
    // 順著s一直找上級，一直找到0
    while (s)
    {
         if (!build[s])
         {
             build[s] =true;
             ret++;
         }
         s = up[s];
    }
    // 注意0本身也要向上建立通道。
    // 因為樣例的第一行 Send 0  輸出為 0
    if (!build[0])
    {
         build[0] = true;
         ret++;
    }
    return ret;
}
 
int Danger(int s)
{
    int ret = 0;
    queue<int> q;
    q.push(s);
    while (!q.empty())
    {
         //從佇列中拉取一個節點
         s = q.front();
         q.pop();
        
         //看 s 自己是否向上級建立了通道
         if (build[s])
         {
             ret++;
             build[s] =false;
         }
         //遍歷 s 所有的下級
         for (int i = 0; i < down[s].size(); ++i)
         {
             // 看這個下級是否建立了通道
             if (build[down[s][i]])
                  q.push(down[s][i]);
         }
    }
    return ret;
}
 
int main()
{
    int i;
    cin >> n;
    for (i = 1; i <n; ++i)
    {
         scanf("%d", &up[i]);     // 設定 i的上級
         down[up[i]].push_back(i); // 設定屬於up[i] 的下級
    }
    char buffer[20];
    cin >> m;
    for (i = 0; i <m; ++i)
    {
         int num;
         scanf("%s", buffer);
         scanf("%d", &num);
         //只用比較第一個字元就可以知道是哪條命令
         if (buffer[0] =='S')
             printf("%d\n", Send(num));
         else
             printf("%d\n", Danger(num));
    }
    return 0;
}







相關文章