北邮OJ-105. 寻找宝藏-11计院上机C

典型的树DFS模型

题目：

时间限制 1000 ms 内存限制 65536 KB Special Judge

题目描述

有一棵多叉树T，你在树根处，宝藏在某一叶子节点L。现在你沿着树枝向叶子方向走去，从不回头，如果遇到树叉节点，你等概率地挑选一个分支继续走。请问，在给定T，L的情况下，你有多大概率拿到宝藏？

输入格式

第一行，整数N，M，L(1< N< 1000，0< M< 1000，0< L< N)，分别代表树T上的节点数、树枝的个数，宝藏所在节点。树根为0号结点。

然后有M行，每行两个整数A，B(0≤ A,B< N)代表从节点A到节点B有一条树枝。可以假设节点A总是更靠近树根。

输出格式

所求的概率，输出四舍五入到6位小数，然后换行。

输入样例

6 5 5

0 1

1 3

0 2

2 4

2 5

输出样例

0.250000

#include <iostream>
#include <cstdio>
#include <vector>
#define MAXSIZE 1010
using namespace std;
typedef struct Node{
int data;//必要时可设置父节点指针来方便查询父节点
vector<int> sonList;
bool turnOn;
Node(){//initiate automatically except data
initNode();
}
void initNode(){
turnOn=false;
sonList.clear();
}
}*Tree;
Node treeList[MAXSIZE];
int cursor;
int n,m,l;
bool haveFind;

//按序分配结点
int createNode(int data){//return the index of new node
if (cursor+1==MAXSIZE)
return -1;//failed
int nowP=cursor;
cursor++;
//  treeList[nowP].initNode();
treeList[nowP].turnOn=true;
treeList[nowP].data=data;
return nowP;
}

//按照指定下标分配结点
int createNode(int data,int index){//return the index of new node
//  treeList[nowP].initNode();
treeList[index].turnOn=true;
treeList[index].data=data;
return index;
}
void freeNode(int index){
treeList[index].initNode();
}
void freeTree(int index){//postOrder
if (index!=-1){
//traverse
Node &nowN=treeList[index];
for (int i=0;i<nowN.sonList.size();i++){
freeTree(nowN.sonList[i]);
}
//visit
freeNode(index);
}
}
void preOrder(int root,double nowProb){
if (root!=-1){
//visit
Node &nowNode=treeList[root];
if (root==l){
printf("%.6lf\n",nowProb);
haveFind=true;
}
//traverse
if (haveFind==false){
for(int i=0;i<nowNode.sonList.size();i++)
preOrder(nowNode.sonList[i],nowProb/nowNode.sonList.size());
}
}
}
int main(){
int father,son;
while (scanf("%d",&n)!=EOF){
//initiate
for (int i=0;i<MAXSIZE;i++){
treeList[i].initNode();
}
haveFind=false;
//input
scanf("%d%d",&m,&l);
for (int i=0;i<m;i++){
scanf("%d%d",&father,&son);
createNode(father,son);
treeList[father].sonList.push_back(son);
}
//search
preOrder(0,1.0);
//output
}
return true;
}