more katas

This commit is contained in:
2019-12-17 22:09:58 +01:00
parent 0b4f33c123
commit 1c3c97a2d6
2 changed files with 96 additions and 0 deletions

View File

@@ -0,0 +1,43 @@
#include <vector>
#include <iostream>
using std::size_t;
std::vector<int> get_column(std::vector<std::vector<int>> input, int index){
std::vector<int> result;
for(int i = 0; i<input.size(); i++){
result.push_back(input[i][index]);
}
return result;
}
int multiply(std::vector<int> first, std::vector<int> second){
int result = 0;
for(int i = 0; i<first.size(); i++){
result += first[i] * second[i];
}
return result;
}
std::vector<std::vector<int>> matrix_multiplication(std::vector<std::vector<int>> &a, std::vector<std::vector<int>> &b, size_t n){
std::vector<std::vector<int>> result;
std::vector<int> row;
for(int i=0; i<n;i++){
for(int j=0; j<n; j++){
row.push_back(
multiply(a[i], get_column(b, j))
);
}
result.push_back(row);
row = {};
}
return result;
}
int main(){
std::vector<std::vector<int>> test1 = {{1,2},{3,4}};
std::vector<std::vector<int>> test2 = {{1,2},{3,4}};
std::vector<std::vector<int>> result = matrix_multiplication(test1, test2, 2);
std::cout << result[1][1];
}