[프로그래머스] 코딩 기초 트레이닝 PART 19(C++)

코딩 테스트/프로그래머스

[프로그래머스] 코딩 기초 트레이닝 PART 19(C++)

JongHoon 2024. 2. 13. 21:23

프로그래머스 코딩 테스트 연습 : 코딩 기초 트레이닝(C++)

https://school.programmers.co.kr/learn/challenges?order=recent&languages=cpp&partIds=44139&page=5

코딩테스트 연습 | 프로그래머스 스쿨

개발자 취업의 필수 관문 코딩테스트를 철저하게 연습하고 대비할 수 있는 문제를 총망라! 프로그래머스에서 선발한 문제로 유형을 파악하고 실력을 업그레이드해 보세요!

school.programmers.co.kr

프로그래머스 코딩 기초 트레이닝 : 세 개의 구분자

https://school.programmers.co.kr/learn/courses/30/lessons/181862

프로그래머스

코드 중심의 개발자 채용. 스택 기반의 포지션 매칭. 프로그래머스의 개발자 맞춤형 프로필을 등록하고, 나와 기술 궁합이 잘 맞는 기업들을 매칭 받으세요.

programmers.co.kr

문제 설명

임의의 문자열이 주어졌을 때 문자 "a", "b", "c"를 구분자로 사용해 문자열을 나누고자 합니다.

예를 들어 주어진 문자열이 "baconlettucetomato"라면 나눠진 문자열 목록은 ["onlettu", "etom", "to"] 가 됩니다.

문자열 myStr이 주어졌을 때 위 예시와 같이 "a", "b", "c"를 사용해 나눠진 문자열을 순서대로 저장한 배열을 return 하는 solution 함수를 완성해 주세요.

단, 두 구분자 사이에 다른 문자가 없을 경우에는 아무것도 저장하지 않으며, return할 배열이 빈 배열이라면 ["EMPTY"]를 return 합니다.

제한사항

1 ≤ myStr의 길이 ≤ 1,000,000
- myStr은 알파벳 소문자로 이루어진 문자열 입니다.

입출력 예

코드

// stringstream을 사용해서 해결하는 1번 방법

#include <string>
#include <vector>
#include <sstream>

using namespace std;

vector<string> solution(string myStr) {
    vector<string> answer;
    
    for (auto &str : myStr)
        if (str == 'a' || str == 'b' || str == 'c')
            str = ' ';
    
    string temp;
    stringstream ss;
    ss.str(myStr);
    while (ss >> temp)
        answer.push_back(temp);
    
    if (!answer.size())
        answer.push_back("EMPTY");
    
    return answer;
}

// 단순하게 if else 문으로 해결하는 2번 방법

#include <string>
#include <vector>

using namespace std;

vector<string> solution(string myStr) {
    vector<string> answer;
    string temp = "";
    
    for (auto str : myStr) {
        if (str == 'a' || str == 'b' || str == 'c') {
            if (temp != "") {
                answer.push_back(temp);
                temp = "";
            }
        }
        else
            temp += str;
    }
        
    if (temp != "")
        answer.push_back(temp);
    
    if (!answer.size())
        answer.push_back("EMPTY");
    
    return answer;
}

결과

프로그래머스 코딩 기초 트레이닝 : 배열의 원소만큼 추가하기

https://school.programmers.co.kr/learn/courses/30/lessons/181861

프로그래머스

programmers.co.kr

문제 설명

아무 원소도 들어있지 않은 빈 배열 X가 있습니다.

양의 정수 배열 arr가 매개변수로 주어질 때, arr의 앞에서부터 차례대로 원소를 보면서 원소가 a라면 X의 맨 뒤에 a를 a번 추가하는 일을 반복한 뒤의 배열 X를 return 하는 solution 함수를 작성해 주세요.

제한사항

1 ≤ arr의 길이 ≤ 100
1 ≤ arr의 원소 ≤ 100

입출력 예

코드

#include <string>
#include <vector>

using namespace std;

vector<int> solution(vector<int> arr) {
    vector<int> answer;
    
    for (auto num : arr)
        for (int i = 0; i < num; i++)
            answer.push_back(num);
    
    return answer;
}

결과

프로그래머스 코딩 기초 트레이닝 : 빈 배열에 추가, 삭제하기

https://school.programmers.co.kr/learn/courses/30/lessons/181860

프로그래머스

programmers.co.kr

문제 설명

아무 원소도 들어있지 않은 빈 배열 X가 있습니다.

길이가 같은 정수 배열 arr과 boolean 배열 flag가 매개변수로 주어질 때, flag를 차례대로 순회하며 flag[i]가 true라면 X의 뒤에 arr[i]를 arr[i] × 2 번 추가하고, flag[i]가 false라면 X에서 마지막 arr[i]개의 원소를 제거한 뒤 X를 return 하는 solution 함수를 작성해 주세요.

제한사항

1 ≤ arr의 길이 = flag의 길이 ≤ 100
arr의 모든 원소는 1 이상 9 이하의 정수입니다.
현재 X의 길이보다 더 많은 원소를 빼는 입력은 주어지지 않습니다.

입출력 예

코드

#include <string>
#include <vector>

using namespace std;

vector<int> solution(vector<int> arr, vector<bool> flag) {
    vector<int> answer;
    
    for (int i = 0; i < arr.size(); i++)
        for (int j = 0; j < arr[i]; j++) {
            if (flag[i]) {
                answer.push_back(arr[i]);
                answer.push_back(arr[i]);
            }
            else
                answer.erase(answer.end() - 1);
        }
        
    return answer;
}

// 다른 사람의 풀이 : 삼항 연산자와 insert를 사용해서 해결하는 방법

#include <vector>
using namespace std;

vector<int> solution(vector<int> arr, vector<bool> flag) {
    vector<int> v;
    for(int i = 0; i < flag.size(); i++)
        flag[i] ? v.insert(v.end(), arr[i]*2, arr[i]) : v.erase(v.end() - arr[i], v.end());
    return v;
}

결과

프로그래머스 코딩 기초 트레이닝 : 배열 만들기 6

https://school.programmers.co.kr/learn/courses/30/lessons/181859

프로그래머스

programmers.co.kr

문제 설명

0과 1로만 이루어진 정수 배열 arr가 주어집니다. arr를 이용해 새로운 배열 stk을 만드려고 합니다.

i의 초기값을 0으로 설정하고 i가 arr의 길이보다 작으면 다음을 반복합니다.

만약 stk이 빈 배열이라면 arr[i]를 stk에 추가하고 i에 1을 더합니다.
stk에 원소가 있고, stk의 마지막 원소가 arr[i]와 같으면 stk의 마지막 원소를 stk에서 제거하고 i에 1을 더합니다.
stk에 원소가 있는데 stk의 마지막 원소가 arr[i]와 다르면 stk의 맨 마지막에 arr[i]를 추가하고 i에 1을 더합니다.

위 작업을 마친 후 만들어진 stk을 return 하는 solution 함수를 완성해 주세요.

단, 만약 빈 배열을 return 해야한다면 [-1]을 return 합니다.

제한사항

1 ≤ arr의 길이 ≤ 1,000,000
- arr의 원소는 0 또는 1 입니다.

입출력 예

코드

#include <string>
#include <vector>

using namespace std;

vector<int> solution(vector<int> arr) {
    vector<int> answer;
    
    for (int i = 0; i < arr.size(); i++) {
        if (!answer.empty() && answer.back() == arr[i])
            answer.erase(answer.end() - 1);
        else
            answer.push_back(arr[i]);
    }
    
    if (answer.empty())
        answer.push_back(-1);
    
    return answer;
}

결과

프로그래머스 코딩 기초 트레이닝 : 무작위로 K개의 수 뽑기

https://school.programmers.co.kr/learn/courses/30/lessons/181858

프로그래머스

programmers.co.kr

문제 설명

랜덤으로 서로 다른 k개의 수를 저장한 배열을 만드려고 합니다.

적절한 방법이 떠오르지 않기 때문에 일정한 범위 내에서 무작위로 수를 뽑은 후, 지금까지 나온적이 없는 수이면 배열 맨 뒤에 추가하는 방식으로 만들기로 합니다.

이미 어떤 수가 무작위로 주어질지 알고 있다고 가정하고, 실제 만들어질 길이 k의 배열을 예상해봅시다.

정수 배열 arr가 주어집니다.

문제에서의 무작위의 수는 arr에 저장된 순서대로 주어질 예정이라고 했을 때, 완성될 배열을 return 하는 solution 함수를 완성해 주세요.

단, 완성될 배열의 길이가 k보다 작으면 나머지 값을 전부 -1로 채워서 return 합니다.

제한사항

1 ≤ arr의 길이 ≤ 100,000
- 0 ≤ arr의 원소 ≤ 100,000
1 ≤ k ≤ 1,000

입출력 예

코드

#include <string>
#include <vector>
#include <unordered_set>

using namespace std;

vector<int> solution(vector<int> arr, int k) {
    vector<int> answer;
    unordered_set<int> us;
    
    for (auto num : arr)
        if (us.insert(num).second)
            answer.push_back(num);
    
    while (answer.size() != k) {
        if (answer.size() > k)
            answer.pop_back();
        else
            answer.push_back(-1);
    }
    
    return answer;
}

// 다른 사람의 풀이 1 : set를 활용하는 방법

#include <string>
#include <vector>
#include <set>

using namespace std;

vector<int> solution(vector<int> arr, int k) {
    vector<int> answer;

    set<int> table;
    for(const auto v : arr)
    {
        if(table.size() < k && !table.count(v))
        {
            answer.emplace_back(v);
            table.insert(v);
        }        
    }
    for(int t = table.size(); t < k; ++t)
    {
        answer.emplace_back(-1);
    }

    return answer;
}

// 다른 사람의 풀이 2 : unordered_map을 활용하는 방법

#include<bits/stdc++.h>
using namespace std;

vector<int> solution(vector<int> arr, int k) {
    vector<int> answer;
    unordered_map<int, int> counter;

    for (const auto& i : arr)
    {
        if (counter[i]++ == 0)
            answer.push_back(i);
    }
    answer.resize(k, -1);
    return answer;
}

결과

여담

첫 번째 문제 '세 개의 구분자'는 2개의 정답을 제출했다.

사실 1번만 제출하고 끝내려고 했지만, 결과 풀이 시간이 너무 긴 것 같아서 stringstream 때문일까싶어서 if else문도 같이 제출해봤다. if else를 사용해서 해결한게 조금 더 빠르긴 했지만 큰 차이는 없었는지 추가 점수를 받거나 하진 않았다ㅋㅋ..

그리고 마지막 문제 '무작위로 K개의 수 뽑기'는 처음에는 문제 지문만 난해(?)하지만 그냥 unique를 쓰고 while문으로 나머지 처리만 하면 되는 쉬운 문제라 생각했는데, 막상 제출해보니 다 틀려서 다시 읽어보고 질문을 찾아보니 생각보다 쉽지 않은 문제였다.

무엇보다 정렬 없이 무작위로 나온 순서를 지켜야하는 것(테스트 케이스는 우연히 정렬된 것처럼 나온 거라고 봐야한다. 만약 [1, 2, 0, 1, 2], 3 이 주어지면 [1, 2, 0]이 되어야하는데 정렬과 함께 써야하는 unique는 못 쓰는 것이다.) 때문에 이것저것 다른 방법을 고민한 끝에 unordered_set이라는걸 알게 되어서 겨우 해결했다.

지금 생각해보면 map을 사용했어도 어떻게 해결되지 않았을까? 싶지만, 다른 사람의 풀이에서 굳이 unordered_map을 사용한 것 보면 아마 잘 안되지 않았을까싶기도 하다ㅋㅋ

C++에서도 배워야 하고 알아야할게 많다는걸 이번에도 또 느낀다.

저작자표시 비영리 (새창열림)