Advertisement
minimario

<Matrix Expo and Mult> TEMPLATE

Feb 1st, 2016
85
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.72 KB | None | 0 0
  1. #include <bits/stdc++.h>
  2. using namespace std;
  3.  
  4. #define ll long long
  5. #define mp make_pair
  6. #define pb push_back
  7. #define pii pair <int, int>
  8. #define piii pair<pii, int>
  9. #define vi vector<int>
  10. #define vpii vector<pii>
  11.  
  12. #define read1(a) int a; scanf("%d", &a)
  13. #define read2(a, b) int a, b; scanf("%d %d", &a, &b)
  14. #define read3(a, b, c) int a, b, c; scanf("%d %d %d", &a, &b, &c)
  15.  
  16. #define FOR(i, a, b) for (int i=a; i<b; i++)
  17. #define F0R(i, a) for (int i=0; i<a; i++)
  18.  
  19. #define readgi(n) F0R(i, n) { scanf("%d", &arr[i]); }
  20. #define readgs(n) F0R(i, n) { scanf(" %c", &arr[i]); }
  21.  
  22. #define f first
  23. #define s second
  24.  
  25. #define usaco(in, out) freopen(in, "r", stdin); freopen(out, "w", stdout);
  26.  
  27. #define println1(a) printf("%d\n", a);
  28. #define println2(a, b) printf("%d %d\n", a, b);
  29. #define println3(a, b, c) printf("%d %d %d\n", a, b, c);
  30. #define pv(v) for (int i : v) { printf("%d ", i); } printf("\n");
  31.  
  32. const int MOD = 1000000007;
  33. const int MAX = 105;
  34.  
  35.  
  36. vector<vector<int> > mult(vector<vi> a, vector<vi> b) {
  37.     vector<vi> v(a.size(), vi(b[0].size(), 0));
  38.     F0R(i, a.size()) {
  39.         F0R(j, b[0].size()) {
  40.             F0R(k, a[0].size()) {
  41.                 v[i][j] = (v[i][j] + a[i][k] * b[k][j]) % MOD;
  42.             }
  43.         }
  44.     }
  45.     return v;
  46. }
  47.  
  48. vector<vector<int> > expo(vector<vi> a, int p) {
  49.     vector<vi> v(a.size(), vi(a.size(), 0));
  50.     F0R(i, a.size()) { v[i][i] = 1; }
  51.     while (p) {
  52.         if (p%2) { v = mult(v, a); }
  53.         a = mult(a, a);
  54.         p /= 2;
  55.     }
  56.     return v;
  57. }
  58.  
  59. int main() {
  60.     vector<vi> a = {{1,2},{2,3},{6,3}};
  61.     vector<vi> b = {{1,2,4,6},{2,1,2,3}};
  62.     vector<vi> c = mult(a, b);
  63.     for (auto i : c) { for (auto j : i) { cout << j << " "; } cout << endl; }
  64. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement