인프런 커뮤니티 질문&답변

안녕하세요 큰돌님,

다름이 아니고 3-N 완전이진 트리문제 이거 정석적인 방법이 잘 안떠올라서 무식하게 풀었는데 이렇게 풀어도 괜찮은가요?

항상 문제 먼저 풀어보고 안풀리면 큰돌님 영상보는 식으로 학습중이거든요.... 이번에는 영상 보기전에 풀어서 통과가 되었지만 이런 방식으로 문제에 접근해도 괜찮은 건지 여쭙니다.

#include <bits/stdc++.h>

using namespace std;
int K, ksize, input;
vector<vector<int>> result;
vector<int> v;

int main()
{
  cin >> K;
  ksize = pow(2, K) - 1;

  for (int i = 0; i < ksize; ++i)
  {
    cin >> input;
    v.push_back(input);
  }

  while (v.size())
  {
    vector<int> temp;
    for (int i = 0; i < v.size(); i += 2)
    {
      temp.push_back(v[i]);
    }
    result.push_back(temp);
    for (int i = v.size() - 1; i >= 0; i--)
    {
      if (i % 2 == 0)
      {
        v.erase(v.begin() + i);
      }
    }
  }

  for (auto i = result.rbegin(); i != result.rend(); ++i)
  {
    for (const auto &n : *i)
    {
      cout << n << ' ';
    }
    cout << '\n';
  }

  return 0;
}