【HihoCoder - 1851】D级上司 (树形图,dfs)
題干:
H公司一共有N名員工,編號(hào)為1~N,其中CEO的編號(hào)是1。除了CEO之外,每名員工都恰好有唯一的直接上司;N名員工形成了一個(gè)樹形結(jié)構(gòu)。 ?
我們定義X的1級(jí)上司是他的直接上司,2級(jí)上司是他上司的上司,以此類推…… ?
請(qǐng)你找出每名員工的D級(jí)上司是誰(shuí)。
Input
第一行包含2個(gè)整數(shù)N和D。 ?
以下N-1行每行包含一個(gè)整數(shù),依次代表編號(hào)是2-N的員工的直接上司的編號(hào)。 ?
對(duì)于50%的數(shù)據(jù),1 ≤ N, D ≤ 10000 ?
對(duì)于100%的數(shù)據(jù),1 ≤ N, D ≤ 100000
Output
依次輸出1~N的D級(jí)上司的編號(hào),每個(gè)一行。如果某員工沒(méi)有D級(jí)上司,輸出-1。
Sample Input
5 2 1 1 3 3Sample Output
-1 -1 -1 1 1解題報(bào)告:
? 開一個(gè)數(shù)組記錄dfs的過(guò)程中走過(guò)的節(jié)點(diǎn),跟走迷宮差不多啊只不過(guò)網(wǎng)格圖換成了樹形圖。。
AC代碼:
#include<cstdio> #include<iostream> #include<algorithm> #include<queue> #include<map> #include<vector> #include<set> #include<string> #include<cmath> #include<cstring> #define ll long long #define pb push_back #define pm make_pair #define fi first #define se second using namespace std; const int MAX = 2e5 + 5; vector<int> vv[MAX]; int n,d; int ans[MAX],dep[MAX]; void dfs(int cur,int root,int deep) {dep[deep] = cur;if(deep - d >= 0) ans[cur] = dep[deep-d];int up = vv[cur].size();for(int i = 0; i<up; i++) {int v = vv[cur][i];if(v == root) continue;dfs(v,cur,deep+1);} } int main() {memset(ans,-1,sizeof ans);cin>>n>>d;for(int i = 2,x; i<=n; i++) {scanf("%d",&x);vv[x].pb(i);vv[i].pb(x);}dfs(1,-1,0);for(int i = 1; i<=n; i++) {printf("%d\n",ans[i]);} return 0 ;}總結(jié):
? ?這題其實(shí)應(yīng)該加單向邊但是雙向邊也可以AC,,我感覺(jué)原因就在于是個(gè)樹形圖,又因?yàn)槭菑母?jié)點(diǎn)開始搜的所以每次搜到的邊都是上級(jí)搜到下級(jí)的邊,這可以算是個(gè)樹形圖的一個(gè)小結(jié)論了,其實(shí)也很常用,因?yàn)槊菜扑械臉湫螆D都可以這樣去使用雙向邊貌似還沒(méi)出過(guò)錯(cuò)。
創(chuàng)作挑戰(zhàn)賽新人創(chuàng)作獎(jiǎng)勵(lì)來(lái)咯,堅(jiān)持創(chuàng)作打卡瓜分現(xiàn)金大獎(jiǎng)總結(jié)
以上是生活随笔為你收集整理的【HihoCoder - 1851】D级上司 (树形图,dfs)的全部?jī)?nèi)容,希望文章能夠幫你解決所遇到的問(wèn)題。
- 上一篇: ScanMailOutLook.exe
- 下一篇: 【CodeForces - 255A】G