Advertisement
El_GEMMY

pisano

Apr 11th, 2022
59
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 3.46 KB | None | 0 0
  1. // Those who cannot remember the past are
  2. // condemned to repeat it (use DP -_-)
  3. // - George Santayana
  4.  
  5. #include <bits/stdc++.h>
  6. #include <ext/pb_ds/assoc_container.hpp>
  7. #include <ext/pb_ds/tree_policy.hpp>
  8.  
  9. using namespace std;
  10. using namespace __gnu_pbds;
  11.  
  12. #define all(v) v.begin(),v.end()
  13. #define rall(v) v.rbegin(),v.rend()
  14. #define ll long long
  15. #define ull unsigned long long
  16. #define MOD 1000000007
  17. #define PI 3.14159265
  18. #define ceil(a, b) ((a / b) + (a % b ? 1 : 0))
  19. #define imin INT_MIN
  20. #define imax INT_MAX
  21. #define inf 2000000000
  22. #define nl '\n'
  23. #define ppcnt __builtin_popcount
  24. #define ppcntll __builtin_popcountll
  25. #define clz __builtin_clz
  26. #define clzll __builtin_clzll
  27. #define ctz __builtin_ctz
  28. #define ctzll __builtin_ctzll
  29. #define modulo(a, b, mod) ((((a) % mod) * ((b) % mod)) % mod)
  30. #define cnte(v, x) count(all(v), (x))
  31. #define mine(v) min_element(all(v))
  32. #define maxe(v) max_element(all(v))
  33. #define debug(x) cout << "x: " << x << nl;
  34. #define debug2(x, y) cout << "x: " << x << " y: " << y << nl;
  35. #define ordered_set tree<pair<int, int>, null_type, less<>, rb_tree_tag, tree_order_statistics_node_update>
  36. #define ordered_map tree<int, int, less<>, rb_tree_tag, tree_order_statistics_node_update>
  37.  
  38. //vector<int> dx = {0, 0, 1, -1, 1, 1, -1, -1}, dy = {1, -1, 0, 0, 1, -1, 1, -1};
  39. //vector<int> dx = {0, 0, 1, -1}, dy = {1, -1, 0, 0};
  40.  
  41. template<typename T = int> istream& operator>>(istream& in, vector<pair<int, int>>& v){
  42.     for (auto& [x, y] : v) in >> x >> y;
  43.     return in;
  44. }
  45.  
  46. template<typename T = int> istream& operator>>(istream& in, vector<T>& v){
  47.     for (T& i : v) in >> i;
  48.     return in;
  49. }
  50.  
  51. template<typename T = int> ostream& operator<<(ostream& out, const vector<T>& v){
  52.     for (const T& x : v)
  53.         out << x << ' ';
  54.     return out;
  55. }
  56.  
  57. template<typename T = pair<int, int>> ostream& operator << (ostream& out, const vector<pair<int, int>>& v){
  58.     for(auto& [x, y] : v){
  59.         out << x << ' ' << y << nl;
  60.     }
  61.     return out;
  62. }
  63.  
  64. void Start_Crushing() {
  65.     ios_base::sync_with_stdio(false);
  66.     cin.tie(nullptr);
  67.     cout.tie(nullptr);
  68. #ifndef ONLINE_JUDGE
  69.     freopen("input.txt", "r", stdin);
  70.     freopen("output.txt", "w", stdout);
  71. #endif
  72. }
  73. int mod;
  74.  
  75. template<typename T>
  76. ll BinExp(T a, T b){
  77.     ll res = 1;
  78.     while(b){
  79.         if(b & 1)
  80.             res *= a;
  81.  
  82.         a *= a;
  83.         b >>= 1;
  84.     }
  85.     return res;
  86. }
  87.  
  88. ll pisano(ll x, ll a, ll b){
  89.     ll prev = a, curr = b;
  90.  
  91.     ll len = 0;
  92.     for(int i = 0; i < x * x; i++){
  93.         ll tmp = (prev + curr) % x;
  94.         prev = curr;
  95.         curr = tmp;
  96.  
  97.         if(curr == a and prev == b)
  98.             len = i + 1;
  99.     }
  100.     return len;
  101. }
  102.  
  103. void solve(){
  104.     ll a, b, n; cin >> a >> b >> n >> mod;
  105.  
  106.     ll p = pisano(mod, a, b);
  107.     ll base = n / p;
  108.     n %= p;
  109.  
  110.     map<ll, ll> freq;
  111.  
  112.     for(int i = 0; i < n; i++){
  113.         freq[a]++;
  114.         ll tmp = (a + b) % mod;
  115.         a = b;
  116.         b = tmp;
  117.     }
  118.  
  119.     debug(p);
  120.     ll ans = 0;
  121.     for(auto& [x, y] : freq){
  122.         debug2(x, y + base);
  123.         ans += BinExp(y + base, 2ll);
  124.     }
  125.     cout << ans;
  126. }
  127.  
  128. int main(){
  129.     Start_Crushing();
  130.  
  131.     int t = 1;
  132.     /*Multiple test cases?*/ cin >> t;
  133.     while (t--) {
  134.         solve();
  135.         if(!t) break;
  136.         cout << nl;
  137.     }
  138.  
  139. //    for(int tc = 1; tc <= t; tc++){
  140. //        cout << "Case #" << tc << ": ";
  141. //        solve();
  142. //        if(tc != t)
  143. //            cout << nl;
  144. //    }
  145.  
  146.     return 0;
  147. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement