#UVa:122-Trees on the level

照著題目所給的順序將樹的節點建出來,然後把值放進去,最後將檢查結果輸出即可。

C++(0.000)

/*******************************************************/
/* UVa 122 Trees on the level                          */
/* Author: Maplewing [at] knightzone.studio            */
/* Version: 2019/04/08                                 */
/*******************************************************/
#include <iostream>
#include <cstdio>
#include <string>
#include <queue>
using namespace std;

struct Node{
  int value;
  Node* left;
  Node* right;

  Node(int value) : value(value), left(NULL), right(NULL) {}
};

bool isAllNodesHasValue(Node* root){
  if(root == NULL) return true;
  if(root->value == -1) return false;

  bool hasValue = isAllNodesHasValue(root->left) && isAllNodesHasValue(root->right);
  return hasValue;
}

void destroyTree(Node* root){
  if(root == NULL) return;
  destroyTree(root->left);
  destroyTree(root->right);

  free(root);
}

void printTree(Node* root){
  queue<Node*> nodeQueue;
  nodeQueue.push(root);

  bool isPrint = false;
  while(!nodeQueue.empty()){
    if(isPrint) printf(" ");

    Node* currentNode = nodeQueue.front();
    nodeQueue.pop();

    printf("%d", currentNode->value);
    if(currentNode->left != NULL) nodeQueue.push(currentNode->left);
    if(currentNode->right != NULL) nodeQueue.push(currentNode->right);
    isPrint = true;
  }

  printf("\n");
}

int main(){
  string node;
  Node* tree = NULL;
  bool isValid = true;

  while(cin >> node){
    if(node == "()"){
      if(isValid && isAllNodesHasValue(tree)){
        printTree(tree);
      }
      else{
        printf("not complete\n");
      }
      destroyTree(tree);
      tree = NULL;
      isValid = true;
      continue;
    }

    if(!isValid) continue;

    int value = 0;
    int index;
    for(index = 1 ; node[index] != ',' ; ++index){
      value = value * 10 + node[index] - '0';
    }

    if(tree == NULL) tree = new Node(-1);
    Node* currentNode = tree;
    for(++index ; node[index] != ')' ; ++index){
      if(node[index] == 'L'){
        if(currentNode->left == NULL) currentNode->left = new Node(-1);
        currentNode = currentNode->left;
      }
      else if(node[index] == 'R'){
        if(currentNode->right == NULL) currentNode->right = new Node(-1);
        currentNode = currentNode->right;
      }
    }
    if(currentNode->value != -1){
      isValid = false;
      continue;
    }

    currentNode->value = value;
  }

  return 0;
}

回應

目前無留言,歡迎留言!

發表迴響

這個網站採用 Akismet 服務減少垃圾留言。進一步瞭解 Akismet 如何處理網站訪客的留言資料

%d 位部落客按了讚: