#include <iostream>
#include <conio.h>
using namespace std;

class Vect;

class Array {
private:
    int **x;
    int size;

    friend void multiplyDiagonalElems(Array &arr, Vect &vect);
public:

    Array(int size_of_array) {
        size = size_of_array;
        x = new int*[size];
        for (int i = 0; i < size; i++) {
            x[i] = new int[size];
        }
    }

    void input() {
        cout << "Starting to input values into array: \n";
        for (int i = 0; i < size; i++)
        {
            for (int j = 0; j < size; j++)
                cin >> x[i][j];
        }
    }

    void output() {
        cout << "Outputting Array : \n";
        for (int i = 0; i < size; i++)
        {
            for (int j = 0; j < size; j++)
                cout << x[i][j] << " ";
            cout << endl;
        }
    }

    int getElem(int i, int j) {
        return x[i][j];
    }

    void setElem(int i, int j, int value) {
        x[i][j] = value;
    }
    void multiplyDiagonalElems2(Vect &vector);
};

class Vect {
private:
    int *x;
    int size;

    friend void Array::multiplyDiagonalElems2(Vect &vector);
public:
    friend void multiplyDiagonalElems(Array &arr, Vect &vect);
    Vect(int size_of_array) {
        size = size_of_array;
        x = new int[size];
    }

    ~Vect() {
        delete[] x;
    }

    void input() {
        cout << "Starting to input values into vector: \n";
        for (int i = 0; i < size; i++)
            cin >> x[i];
    }

    void output() {
        cout << "Outputting Vector : \n";
        for (int i = 0; i < size; i++)
        {
            cout << x[i] << " ";
        }
        cout << endl;
    }

    int getElem(int pos) {
        return x[pos];
    }

    void setElem(int pos, int value) {
        x[pos] = value;
    }
};


void Array::multiplyDiagonalElems2(Vect &vector)
{
    for (int i = 0; i < size; i++)
    {
        x[i][i] *= vector.x[i];
    }
}

void multiplyDiagonalElems(Array &arr, Vect &vect) {
    cout << "Outputting Array : \n";
    for (int i = 0; i < arr.size; i++)
    {
        arr.x[i][i] *= vect.x[i];
    }
}

int main()
{
    int n = 3;
    Array matrix(n);
    matrix.input();
    matrix.output();

    Vect vector(n);
    vector.input();
    vector.output();

    //multiplyDiagonalElems(matrix, vector);
    matrix.multiplyDiagonalElems2(vector);

    matrix.output();

    getch();
    return 0;
}