알고리즘 스터디 1일차

박경록 저자님의 코딩 테스트 합격자 되기 c++ 편을 완독 하는 것을 목표로 남기는 공부 일지
(배열 파트)

1. 두 개 뽑아서 더하기

1) 문제

문제 설명
정수 배열 numbers가 주어집니다. numbers에서 서로 다른 인덱스에 있는 두 개의 수를 뽑아 더해서 만들 수 있는 모든 수를 배열에 오름차순으로 담아 return 하도록 solution 함수를 완성해주세요.

제한사항
numbers의 길이는 2 이상 100 이하입니다.
numbers의 모든 수는 0 이상 100 이하입니다.

입출력 예
numbers result
[2,1,3,4,1][2,3,4,5,6,7]
[5,0,2,7][2,5,7,9,12]

2) 풀이

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

vector<int> solution(vector<int> numbers) {
    vector<int> answer;
    for (int i=0; i<numbers.size()-1; i++)
    {
        for (int j=i+1; j<numbers.size();j++)
            answer.push_back(numbers[i]+numbers[j]);
    }

    sort(answer.begin(), answer.end());
    answer.erase(unique(answer.begin(), answer.end()), answer.end());

    return answer;
}

이중 for문을 통해서 합을 구하는 문제이다.
여기서 중복을 처리하는 방향이 3가지가 떠올랐다.

• 일단 값을 다 answer에 넣고 나중에 중복 삭제

• push_back() 하기 전에 중복여부 확인

• set을 이용해 중복 허용 x

  여기서 1번을 선택했다.

2. 모의고사

1) 문제

문제 설명
수포자는 수학을 포기한 사람의 준말입니다. 수포자 삼인방은 모의고사에 수학 문제를 전부 찍으려 합니다. 수포자는 1번 문제부터 마지막 문제까지 다음과 같이 찍습니다.

1번 수포자가 찍는 방식: 1, 2, 3, 4, 5, 1, 2, 3, 4, 5, ...
2번 수포자가 찍는 방식: 2, 1, 2, 3, 2, 4, 2, 5, 2, 1, 2, 3, 2, 4, 2, 5, ...
3번 수포자가 찍는 방식: 3, 3, 1, 1, 2, 2, 4, 4, 5, 5, 3, 3, 1, 1, 2, 2, 4, 4, 5, 5, ...

1번 문제부터 마지막 문제까지의 정답이 순서대로 들은 배열 answers가 주어졌을 때, 가장 많은 문제를 맞힌 사람이 누구인지 배열에 담아 return 하도록 solution 함수를 작성해주세요.

제한 조건
시험은 최대 10,000 문제로 구성되어있습니다.
문제의 정답은 1, 2, 3, 4, 5중 하나입니다.
가장 높은 점수를 받은 사람이 여럿일 경우, return하는 값을 오름차순 정렬해주세요.

입출력 예
answers return
[1,2,3,4,5][1]
[1,3,2,4,2][1,2,3]

2) 풀이

#include <string>
#include <vector>
#include <algorithm>

using namespace std;

vector<int> solution(vector<int> answers) {
    vector<int> answer;
    vector<vector<int>> students = {{1,2,3,4,5}, {2,1,2,3,2,4,2,5}, {3,3,1,1,2,2,4,4,5,5}};
    vector<int> Ocount(3);
    for (int i=0; i < answers.size(); i++)
    {
        for (int j=0 ; j < 3; j++)
            if (answers[i] == students[j][i % students[j].size()]) 
                Ocount[j] += 1;
    }

    int maxCount = *max_element(Ocount.begin(),Ocount.end());

    for (int i=0 ; i < 3; i++)
        if (Ocount[i] == maxCount) answer.push_back(i+1);    

    return answer;
}

각 수포자가 찍는 패턴을 찾아 배열에 저장하고 답안을 순회하며 비교를 해 가장 많이 맞춘 사람을 찾는 방식.
모듈러 연산을 통해서 간단하게 비교 가능

최댓값을 찾고 이 값과 같으면 answer에 추가

3. 행렬의 곱셈

1) 문제

문제 설명
2차원 행렬 arr1과 arr2를 입력받아, arr1에 arr2를 곱한 결과를 반환하는 함수, solution을 완성해주세요.

제한 조건
행렬 arr1, arr2의 행과 열의 길이는 2 이상 100 이하입니다.
행렬 arr1, arr2의 원소는 -10 이상 20 이하인 자연수입니다.
곱할 수 있는 배열만 주어집니다.

입출력 예
arr1 arr2 return
[[1, 4], [3, 2], [4, 1]] [[3, 3], [3, 3]] [[15, 15], [15, 15], [15, 15]]
[[2, 3, 2], [4, 2, 4], [3, 1, 4]] [[5, 4, 3], [2, 4, 1], [3, 1, 1]] [[22, 22, 11], [36, 28, 18], [29, 20, 14]]

2) 풀이

#include <string>
#include <vector>

using namespace std;

vector<vector<int>> solution(vector<vector<int>> arr1, vector<vector<int>> arr2) {
    vector<vector<int>> answer;
    /*
    a11 a12     b11 b12 b13 => (a11 * b11  + a12 * b21) ( a11 * b12 + a12 * b22 ) ( a11 * b13 + a12 * b23 )
    a21 a22     b21 b22 b23    (a21 * b11  + a22 * b21) ( a21 * b12 + a22 * b22 ) ( a21 * b13 + a22 * b23 )
    행렬곱 예시
    {1 2} {1 2 3 } => {(1*1 + 2*4), (1*2 + 2*5), (1*3 + 2*6)} = {9 ,12, 15}
    {3 4} {4 5 6 }    {(3*1 + 4*4), (3*2 + 4*5), (3*3 + 4*6)}   {19,26, 33}
    */

    //a*b x b*c 면 a*c의 행렬 생성
    answer.resize(arr1.size(), vector<int>(arr2[0].size()));

    int leftMatrixRowCnt = arr1.size();
    // left col cnt == right row cnt 이경우만 행렬의 곱셈을 계산할 수 있음
    int leftMatrixColCnt = arr1[0].size();
    int rightMatrixColCnt = arr2[0].size();

    for (int i=0; i < leftMatrixRowCnt; i++)
    {
        for (int j=0; j < rightMatrixColCnt; j++)
        {
            for (int k=0; k < leftMatrixColCnt; k++)
               answer[i][j] += arr1[i][k] * arr2[k][j];
        }
    }
    return answer;
}

행렬곱의 원리를 알면 쉽게 풀 수 있는 문제.
자세한 설명은 위 코드의 주석으로 대체