Advertisement
El_GEMMY

Untitled

Apr 14th, 2022
53
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 3.15 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.  
  74. void solve(){
  75. int n, k; cin >> n >> k;
  76. ll sum = 0, ans = 0;
  77.  
  78. vector<ll> v(n), prefix(n + 1);
  79.  
  80. map<ll, int> suffix;
  81.  
  82. for(int i = 0; i < n; i++){
  83. cin >> v[i];
  84. sum += v[i];
  85. prefix[i + 1] = prefix[i] + v[i];
  86. prefix[i + 1] %= k;
  87. }
  88.  
  89. ll last = 0;
  90. for(int i = n - 1; i >= 0; i--){
  91. last += v[i];
  92. suffix[last % k]++;
  93. }
  94.  
  95. for(int i = 1; i <= n; i++){
  96. suffix[sum % k]--;
  97. sum -= v[i - 1];
  98. ll need = k - prefix[i];
  99.  
  100. ans += suffix[need * (need != k)];
  101. }
  102. cout << ans;
  103. }
  104.  
  105. int main(){
  106. Start_Crushing();
  107.  
  108. int t = 1;
  109. /*Multiple test cases?*/ cin >> t;
  110. while (t--) {
  111. solve();
  112. if(!t) break;
  113. cout << nl;
  114. }
  115.  
  116. // for(int tc = 1; tc <= t; tc++){
  117. // cout << "Case #" << tc << ": ";
  118. // solve();
  119. // if(tc != t)
  120. // cout << nl;
  121. // }
  122.  
  123. return 0;
  124. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement