Spicy Tuna Sushi
본문 바로가기
문제를 풀자

[프로그래머스] 행렬의 곱셈(C++)

by 말린malin 2022. 8. 13.

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

 

프로그래머스

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

programmers.co.kr

행렬의 곱을 풀어쓰면 3중 반복문을 어떻게 써야 할지 알 수 있다.

나는 미리 벡터 크기를 할당해 주었는데, push_back을 활용해도 된다.

#include <string>
#include <vector>
#include <iostream>

using namespace std;

vector<vector<int>> solution(vector<vector<int>> arr1, vector<vector<int>> arr2) {
    vector<vector<int>> answer(arr1.size(),vector<int>(arr2[0].size()));

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

댓글