DFS 소개
이진 트리의 사전 주문 순회와 유사하게 깊이가 먼저 순회 된 다음 다시 순회됩니다.
LeetCode 589. N 항 트리의 사전 주문 순회
N 항 트리가 주어지면 노드 값의 이전 순회를 반환합니다.
예를 들어 3 진 트리가 주어지면 :
이전 순회로 돌아 가기 : [1,3,5,6,2,4]
솔루션 : 재귀 + 스택으로 간단한 DFS 구현
암호
/*
// Definition for a Node.
class Node {
public:
int val;
vector<Node*> children;
Node() {}
Node(int _val) {
val = _val;
}
Node(int _val, vector<Node*> _children) {
val = _val;
children = _children;
}
};
*/
class Solution {
public:
vector<int> preorder(Node* root) {
vector<int> ans;
if(!root) return ans;
stack<Node*> s;
s.push(root);
while(!s.empty()){
Node *temp=s.top();
s.pop();
ans.push_back(temp->val);
for(int i=temp->children.size()-1;i>=0;--i){
s.push(temp->children[i]);
}
}
return ans;
}
};