alpar@994
|
1 |
#include <lemon/time_measure.h>
|
alpar@994
|
2 |
#include <vector>
|
alpar@994
|
3 |
|
alpar@994
|
4 |
template<class T>
|
alpar@994
|
5 |
class Matrix
|
alpar@994
|
6 |
{
|
alpar@994
|
7 |
public:
|
alpar@994
|
8 |
int m,n;
|
alpar@994
|
9 |
std::vector<T> values;
|
alpar@994
|
10 |
|
alpar@994
|
11 |
Matrix(int _m, int _n) : m(_m), n(_n), values(n*m) {}
|
alpar@994
|
12 |
typename std::vector<T>::reference operator()(int i, int j)
|
alpar@994
|
13 |
{ return values[i*n+j];}
|
alpar@994
|
14 |
|
alpar@994
|
15 |
class Row
|
alpar@994
|
16 |
{
|
alpar@994
|
17 |
public:
|
alpar@994
|
18 |
Matrix<T> &mat;
|
alpar@994
|
19 |
int row;
|
alpar@994
|
20 |
|
alpar@994
|
21 |
Row(Matrix<T> &_mat,int _row) : mat(_mat), row(_row) {}
|
alpar@994
|
22 |
typename std::vector<T>::reference operator()(int j)
|
alpar@994
|
23 |
{ return mat.values[row*mat.n+j];}
|
alpar@994
|
24 |
};
|
alpar@994
|
25 |
class Column
|
alpar@994
|
26 |
{
|
alpar@994
|
27 |
public:
|
alpar@994
|
28 |
Matrix<T> &mat;
|
alpar@994
|
29 |
int col;
|
alpar@994
|
30 |
|
alpar@994
|
31 |
Column(Matrix<T> &_mat,int _col) : mat(_mat), col(_col) {}
|
alpar@994
|
32 |
typename std::vector<T>::reference operator[](int i)
|
alpar@994
|
33 |
{ return mat.values[i*mat.n+col];}
|
alpar@994
|
34 |
};
|
alpar@994
|
35 |
|
alpar@994
|
36 |
Row operator[](int i) { return Row(*this,i); }
|
alpar@994
|
37 |
Column operator()(int j) { return Column(*this,j); }
|
alpar@994
|
38 |
};
|
alpar@994
|
39 |
|
alpar@994
|
40 |
main()
|
alpar@994
|
41 |
{
|
alpar@994
|
42 |
const int COUNT=1000000;
|
alpar@994
|
43 |
const int SIZE=100;
|
alpar@994
|
44 |
|
alpar@994
|
45 |
Matrix<int> A(SIZE,SIZE);
|
alpar@994
|
46 |
|
alpar@994
|
47 |
for(int i=0;i<SIZE;i++) for(int j=0;j<SIZE;j++) A(i,j)=i+j;
|
alpar@994
|
48 |
|
alpar@994
|
49 |
int s=0;
|
alpar@994
|
50 |
lemon::Timer T;
|
alpar@994
|
51 |
for(int c=0;c<COUNT;c++) {
|
alpar@994
|
52 |
for(int i=0;i<SIZE;i++) for(int j=0;j<SIZE;j++) s+=A[i](j);
|
alpar@994
|
53 |
}
|
alpar@994
|
54 |
std::cout << "A[i](j): " << T << ' ' << s <<'\n';
|
alpar@994
|
55 |
s=0;
|
alpar@994
|
56 |
T.reset();
|
alpar@994
|
57 |
for(int c=0;c<COUNT;c++) {
|
alpar@994
|
58 |
for(int i=0;i<SIZE;i++) for(int j=0;j<SIZE;j++) s+=A(i,j);
|
alpar@994
|
59 |
}
|
alpar@994
|
60 |
std::cout << "A(i,j): " << T << ' ' << s << '\n';
|
alpar@994
|
61 |
s=0;
|
alpar@994
|
62 |
T.reset();
|
alpar@994
|
63 |
for(int c=0;c<COUNT;c++) {
|
alpar@994
|
64 |
for(int i=0;i<SIZE;i++) for(int j=0;j<SIZE;j++) s+=A(j)[i];
|
alpar@994
|
65 |
}
|
alpar@994
|
66 |
std::cout << "A(j)[i]: " << T << ' ' << s << '\n';
|
alpar@994
|
67 |
s=0;
|
alpar@994
|
68 |
T.reset();
|
alpar@994
|
69 |
for(int c=0;c<COUNT;c++) {
|
alpar@994
|
70 |
for(int i=0;i<SIZE;i++) for(int j=0;j<SIZE;j++) s+=A(i,j);
|
alpar@994
|
71 |
}
|
alpar@994
|
72 |
std::cout << "A(j,i): " << T << ' ' << s << "\n\n";
|
alpar@994
|
73 |
|
alpar@994
|
74 |
return 0;
|
alpar@994
|
75 |
}
|