Advertisement
AlezM

Sonya HW

Mar 13th, 2019
86
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 3.68 KB | None | 0 0
  1. #include <iostream>
  2. #include <vector>
  3.  
  4. double RandomDouble(double min = -1, double max = 1)
  5. {
  6.     double f = (double)std::rand() / RAND_MAX;
  7.     return min + f * (max - min);
  8. }
  9.  
  10. std::vector<double> RandomDoubleVector(size_t size, double min = -1, double max = 1)
  11. {
  12.     auto vector = std::vector<double>(size);
  13.     for (int i = 0; i < size; ++i) {
  14.         vector[i] = RandomDouble(min, max);
  15.     }
  16.  
  17.     return vector;
  18. }
  19.  
  20. class Matrix {
  21. public:
  22.     Matrix(size_t size) : _size(size), _data(size * size) {}
  23.     Matrix(size_t size, std::vector<double> data) : _size(size), _data(size * size) {
  24.         _data = data;
  25.     }
  26.     double& operator()(size_t i, size_t j) {
  27.         return _data[i * _size + j];
  28.     }
  29.     double operator()(size_t i, size_t j) const {
  30.         return _data[i * _size + j];
  31.     }
  32.     static Matrix I(size_t size) {
  33.         auto matrix = Matrix(size);
  34.         for (size_t i = 0; i < size; ++i)
  35.             matrix(i, i) = 1;
  36.  
  37.         return matrix;
  38.     }
  39.  
  40.     static Matrix RandomMatrix(size_t size, double min = -1, double max = 1) {
  41.         auto matrix = Matrix(size);
  42.         for (size_t i = 0; i < size; ++i)
  43.             for (size_t j = 0; j < size; ++j)
  44.                 matrix(i, j) = RandomDouble(min, max);
  45.  
  46.         return matrix;
  47.     }
  48.  
  49.     Matrix operator+(const Matrix& b) {
  50.         auto sum = Matrix(_size, _data);
  51.         for (size_t i = 0; i < _size; ++i)
  52.             for (size_t j = 0; j < _size; ++j)
  53.                 sum(i, j) += b(i, j);
  54.  
  55.         return sum;
  56.     }
  57.  
  58.     Matrix operator+(const double & d) {
  59.         auto sum = Matrix(_size, _data);
  60.         for (size_t i = 0; i < _size; ++i)
  61.             sum(i, i) += d;
  62.  
  63.         return sum;
  64.     }
  65.  
  66.     Matrix operator+=(const Matrix& b) {
  67.         for (size_t i = 0; i < _size; ++i)
  68.             for (size_t j = 0; j < _size; ++j)
  69.                 operator()(i, j) += b(i, j);
  70.  
  71.         return *this;
  72.     }
  73.  
  74.     Matrix operator*(const Matrix& b) {
  75.         auto product = Matrix(_size);
  76.  
  77.         for (size_t i = 0; i < _size; ++i)
  78.             for (size_t j = 0; j < _size; ++j)
  79.                 for (size_t k = 0; k < _size; ++k) {
  80.                     product(i, j) += operator()(i, k) * b(k, j);
  81.                 }
  82.  
  83.         return product;
  84.     }
  85.  
  86.     Matrix operator*(const double& d) {
  87.         auto scalarMult = Matrix(_size, _data);
  88.         for (size_t i = 0; i < _size; ++i)
  89.             for (size_t j = 0; j < _size; ++j)
  90.                 scalarMult(i, j) *= d;
  91.  
  92.         return scalarMult;
  93.     }
  94.  
  95.     Matrix Pow(size_t n) {
  96.         auto result = Matrix(_size);
  97.  
  98.         for (size_t i = 0; i < _size; i++)
  99.         {
  100.             result(i, i) = 1;
  101.         }
  102.  
  103.         for (size_t i = 0; i < n; i++) {
  104.             result = result * *this;
  105.         }
  106.  
  107.         return result;
  108.     }
  109.  
  110.     void Show() {
  111.         for (size_t i = 0; i < _size; ++i) {
  112.             for (size_t j = 0; j < _size; ++j)
  113.                 std::cout << operator()(i, j) << " ";
  114.             std::cout << std::endl;
  115.         }
  116.         std::cout << std::endl;
  117.     }
  118.  
  119. private:
  120.     size_t _size;
  121.     std::vector<double> _data;
  122. };
  123.  
  124. int main() {
  125.     size_t matrixSize = 5;
  126.     auto A = Matrix::RandomMatrix(matrixSize);
  127.     std::vector<double> polinomialCoefs = RandomDoubleVector(10, -5, 5);
  128.  
  129.     auto P1 = Matrix(matrixSize);
  130.     for (int n = 0; n < polinomialCoefs.size(); n++) {
  131.         P1 += A.Pow(n) * polinomialCoefs[n];
  132.     }
  133.     P1.Show();
  134.  
  135.  
  136.     auto P2 = Matrix::I(matrixSize) * polinomialCoefs[polinomialCoefs.size() - 1];
  137.     for (int n = polinomialCoefs.size() - 2; n >= 0; n--) {
  138.         P2 = P2 * A + Matrix::I(matrixSize) * polinomialCoefs[n];
  139.     }
  140.     P2.Show();
  141.  
  142.     return 0;
  143. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement