L2_020功夫傳人(DFS)

weixin_34107955發表於2018-03-30

一門武功能否傳承久遠並被髮揚光大,是要看緣分的。一般來說,師傅傳授給徒弟的武功總要打個折扣,於是越往後傳,弟子們的功夫就越弱…… 直到某一支的某一代突然出現一個天分特別高的弟子(或者是吃到了靈丹、挖到了特別的祕笈),會將功夫的威力一下子放大N倍 —— 我們稱這種弟子為“得道者”。
這裡我們來考察某一位祖師爺門下的徒子徒孫家譜:假設家譜中的每個人只有1位師傅(除了祖師爺沒有師傅);每位師傅可以帶很多徒弟;並且假設輩分嚴格有序,即祖師爺這門武功的每個第i代傳人只能在第i-1代傳人中拜1個師傅。我們假設已知祖師爺的功力值為Z,每向下傳承一代,就會減弱r%,除非某一代弟子得道。現給出師門譜系關係,要求你算出所有得道者的功力總值。


輸入格式:
輸入在第一行給出3個正整數,分別是:N(<=105)——整個師門的總人數(於是每個人從0到N-1編號,祖師爺的編號為0);Z——祖師爺的功力值(不一定是整數,但起碼是正數);r ——每傳一代功夫所打的折扣百分比值(不超過100的正數)。接下來有N行,第i行(i=0, ..., N-1)描述編號為i的人所傳的徒弟,格式為:
Ki ID[1] ID[2] ... ID[Ki]
其中Ki是徒弟的個數,後面跟的是各位徒弟的編號,數字間以空格間隔。Ki為零表示這是一位得道者,這時後面跟的一個數字表示其武功被放大的倍數。


輸出格式:
在一行中輸出所有得道者的功力總值,只保留其整數部分。題目保證輸入和正確的輸出都不超過1010。


輸入樣例:
10 18.0 1.00
3 2 3 5
1 9
1 4
1 7
0 7
2 6 1
1 8
0 9
0 4
0 3
輸出樣例:
404


  • 思路
    首先一個level陣列存每一層(離祖師爺的距離)的得道者放大倍數之和
    visited陣列存是否被訪問過
    maxlevel記錄最大的擁有得道者的層數(避免等下要訪問整個level陣列)
    徒弟結構包含,徒弟id,指向下一位徒弟的指標
    結構體peop表示每位成員,有id,第一位徒弟的指標(得道者就是null),放大倍數big(非得道者就是1),到祖師爺的路徑長,isdd標記是否是得道者
    先讀資料非得道者就讀徒弟,lt是上一個徒弟的指標,若lt是null就把當前徒弟賦給這個節點的第一個徒弟指標,否則就把當前節點賦給lt的next,記得更新lt
    若遇到得道者就讀入它的放大倍數big
    將祖師爺的path置為0
    開始DFS
    若為訪問過該節點,就標記訪問,若該節點十個得道者isdd為1,則該節點的path為索引的level陣列的值加上該節點的放大倍數,並可能更新maxlevel
    設定一個指標t指向該節點的第一個孩子指標
    while迴圈若t補為null就去把t的id對應的path賦為該節點的path+1
    然後DFS(t->id),t=t->next
    最後遍歷level陣列進行計算即可
    即在普通的連結串列儲存DFS的基礎上加上訪問時對得道者的操作,以及遍歷孩子時對path的更新
  • 小注意點,注意那個r是相對於100的!!然後想把一個數轉小數1.0不要0.1==
#include <iostream>
#include <math.h>
#define N 100000
using namespace std;
int level[N] = { 0 };//這一層的得道者數目
int visited[N] = { 0 };
int maxlevel = 0;
struct td
{
    //徒弟連結串列節點
    int id;
    td* nexttd;
};
struct peop
{
    int id;
    td* firtd;
    int big;//放大倍數
    int isdd;//0為非得到,1為得道
    int path;//到祖師爺的路徑長度
};
peop peops[N];
void DFS(int start)
{
    if (!visited[start]) {
        visited[start] = 1;
        if (peops[start].isdd) {
            level[peops[start].path] += peops[start].big;
            maxlevel = (maxlevel < peops[start].path) ? peops[start].path : maxlevel;
        }
        td* t = peops[start].firtd;
        while (t != NULL) {
            if (!visited[t->id]) {
                peops[t->id].path = peops[start].path + 1;
                DFS(t->id);
            }
            t = t->nexttd;
        }
    }
}
int main()
{
    int n;
    double z, r;
    cin >> n >> z >> r;
    for (int i = 0; i < n; ++i) {
        peops[i].id = i;
        peops[i].firtd = NULL;
        int k; cin >> k;
        if (k) {
            peops[i].big = 1;
            peops[i].isdd = 0;
            td *lt=NULL;
            for (int j = 0; j < k; ++j) {
                int x; cin >> x;
                td *t = new td();
                t->id = x;
                t->nexttd = NULL;
                if(lt){
                    //不是第一個
                    lt->nexttd = t;
                    lt = t;
                }else {
                    lt = t;
                    peops[i].firtd = t;
                }
            }
        }
        else {
            cin>>peops[i].big;
            peops[i].isdd = 1;
        }
    }
    peops[0].path = 0;
    DFS(0);
    double ans=0;
    for (int i = 0; i <= maxlevel;++i){
        ans += z*pow((1 - r/100*1.0), i)*level[i];
    }
    int a = ans;
    cout << a;
    system("pause");
    return 0;
}
  • 這位不用連結串列用二維向量的很簡潔
    https://www.liuchuo.net/archives/3835
    vector<vector<int> >,用來存每個人的徒弟,沒有徒弟就在index[i][0]存它的放大倍數
    用一個陣列標記某個編號是否是得到者
    power初值為z每遞迴一層就(1-r/100)
    DFS時傳入(當前訪問的節點的編號,power)
    若遇到得道者則將power
    放大倍數並加到ans上,return
    否則遍歷這一行(即它的所有徒弟)DFS