코딩테스트 문제풀이/프로그래머스

[프로그래머스] 종이접기

itaeiou 2020. 2. 26. 23:14
반응형

프로그래머스

종이접기

 

https://programmers.co.kr/learn/courses/30/lessons/62049?language=cpp#

 

코딩테스트 연습 - 종이접기 | 프로그래머스

직사각형 종이를 n번 접으려고 합니다. 이때, 항상 오른쪽 절반을 왼쪽으로 접어 나갑니다. 다음은 n = 2인 경우의 예시입니다. 먼저 오른쪽 절반을 왼쪽으로 접습니다. 다시 오른쪽 절반을 왼쪽으로 접습니다. 종이를 모두 접은 후에는 종이를 전부 펼칩니다. 종이를 펼칠 때는 종이를 접은 방법의 역순으로 펼쳐서 처음 놓여있던 때와 같은 상태가 되도록 합니다. 위와 같이 두 번 접은 후 종이를 펼치면 아래 그림과 같이 종이에 접은 흔적이 생기게 됩니다. 위

programmers.co.kr

 

단순 코드 (시간 초과남)

#include <string>
#include <vector>

using namespace std;

vector<int> solution(int n) {
    vector<int> answer;
    int size;
    
    answer.push_back(0);
    for(int i=1; i<n; i++) {
        size = answer.size();
        for(int j=0; j<=size; j++) {
            answer.insert(answer.begin()+j*2, j%2);
        }
    }
    
    return answer;
}

 

 

방식 변경

#include <string>
#include <vector>

using namespace std;

vector<int> solution(int n) {
    vector<int> answer;
    int size;
    
    answer.push_back(0);
    for(int i=1; i<n; i++) {
        size = answer.size();
        vector<int> temp(answer);
        answer.clear();
        for(int j=0; j<size; j++) {
            answer.push_back(j%2);
            answer.push_back(temp[j]);
        }
        answer.push_back(1);
    }
    
    return answer;
}
반응형