[알고리즘] 대기열을 이용하여 두 갈래 나무의 폭을 구하다

5964 단어 두 갈래 나무

 
내 생각은 순환 대기열을 이용하여 각 층의 결점을 저장하는 것이다. 그러나 나는 각 층을 저장한 후에 1개의 flag 표시를 추가했다. 이 층이 완성되면 다음 층의 결점을 통계할 수 있다. maxLevel을 이용하여 현재 층 이상(현재 층 포함)의 너비,currentLevel을 저장하고 아래층의 결점을 저장한다. 이 층이 완성된 후에 maxLevel,currentLevel,maxLevel=max(maxLevel,currentLevel)를 비교한다.current Level을 0으로 복원하고 다음 단계로 계속 이동합니다.마지막으로 maxLevel 코드는 다음과 같습니다.
 1 void enQueue(BTNode* qu[],BTNode* p,int &rear)
 2 {
 3     rear = (rear+1)%maxSize ;
 4         qu[rear] = p;
 5 }
 6 int width(BTNode* p)
 7 {
 8     if(p==NULL) return 0;
 9     BTNode* qu[maxSize];
10     int front = rear = 0;
11     BTNode* flag = new BTNode;   // BTNode* flag = (BTNode*)malloc(sizeof(BTNode));
12     enQueue(qu,p,rear);
13     enQueue(qu,flag,rear);
14     int maxLevel = 1, currentLevel = 0;// 1, 0, 
15     while(front!=rear)
16     {
17         while(qu[(front+1)%maxSize]!=flag)
18         {
19             front = (front+1)%maxSize;
20             if(qu[front]->lchild!=NULL)
21             {
22                 enQueue(qu,qu[front]->lchild,rear);
23             }
24             if(qu[front]->rchild!=NULL)
25             {
26                 enQueue(qu,qu[front]->rchild,rear);
27             }
28         }
29         front = (front+1)%maxSize;// flag 
30         if(currentLevel != 0)// flag
31         {
32             enQueue(qu,flag,rear);
33         }
34         if(maxLevel < currentLevel)
35         {
36             maxLevel = currentLevel;
37         }
38         currentLevel = 0;
39     }
40     delete flag;// free(flag);
41     return maxLevel;
42 }      

좋은 웹페이지 즐겨찾기