Advertisement
Korotkodul

reg A stress

Oct 8th, 2022
922
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 1.85 KB | None | 0 0
  1. #include <iostream>
  2. #include <cmath>
  3. #include <vector>
  4. #include <queue>
  5. #include <algorithm>
  6. #include <string>
  7. #include <stack>
  8. #include <set>
  9. #include <map>
  10. #define pii pair <int, int>
  11. #define pb(x) push_back(x)
  12. using namespace std;
  13. using ll = long long;
  14. using ld = long double;
  15. using db = double;
  16. void cv(vector <int> &v) {
  17.     for (auto x : v) cout << x << ' ';
  18.     cout << "\n";
  19. }
  20.  
  21. void cvl(vector <ll> &v) {
  22.     for (auto x : v) cout << x << ' ';
  23.     cout << "\n";
  24. }
  25.  
  26.  
  27. void cvv(vector <vector <int> > &v) {
  28.     for (auto x : v) cv(x);
  29.     cout << "\n";
  30. }
  31.  
  32. void cvb(vector <bool> v) {
  33.     for (bool x : v) cout << x << ' ';
  34.     cout << "\n";
  35. }
  36.  
  37. void cvs(vector <string>  v) {
  38.     for (auto a : v) {
  39.         cout << a << "\n";
  40.     }
  41. }
  42.  
  43. void cvp(vector <pii> a) {
  44.     for (auto p : a) {
  45.         cout << p.first << ' ' << p.second << "\n";
  46.     }
  47.     cout << "\n";
  48. }
  49.  
  50. #include <ctime>
  51.  
  52. bool sh = 0;
  53.  
  54. ll l, r, a, n;
  55.  
  56. ll check() {
  57.     ll ans = 0;
  58.     for (int i = 1; i <= n; ++i) {
  59.         for (int j = i + 1; j <= n; ++j) {
  60.             if (j > i && (j - i) % a == 0) {
  61.                 ans++;
  62.             }
  63.         }
  64.     }
  65.     return ans;
  66. }
  67.  
  68. /*
  69. WA
  70. n a = 6 2
  71. my his = 5 6
  72. */
  73.  
  74. bool random = 1;
  75.  
  76. ll f() {
  77.     if (a >= n) {
  78.         return 0;
  79.     }
  80.     ll myn = n - a;
  81.     ll N = myn / a;
  82.     ll S =  a * (N + 1) * N / 2;
  83.     ll d = n % a * (N + 1);
  84.     //cout << "N = " << N << "\n";
  85.     ll ans = S + d;
  86.     return ans;
  87. }
  88.  
  89. int main() {
  90.     /*ios::sync_with_stdio(0);
  91.     cin.tie(0);
  92.     cout.tie(0);*/
  93.  
  94.     while (1) {
  95.         n = rand() % 200 + 40;
  96.         a = rand() % 30 + 13;
  97.         ll his = check();
  98.         ll my = f();
  99.         if (my != his) {
  100.             cout << "WA\n";
  101.             cout << "n a = " << n << ' ' << a << "\n";
  102.         } else {
  103.             cout << "OK\n";
  104.         }
  105.     }
  106. }
  107.  
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement