Advertisement
Dang_Quan_10_Tin

MEDIAN

Mar 17th, 2022
1,011
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
C++ 2.02 KB | None | 0 0
  1. #define task "MEDIAN"
  2. #include <iostream>
  3. #include <cstdio>
  4. #include <algorithm>
  5.  
  6. using namespace std;
  7.  
  8. using ll = long long;
  9. using ld = long double;
  10.  
  11. constexpr int N = 1e6 + 5;
  12.  
  13. int n, a[N], b[N], m, c[N], id[N];
  14. ll g[N];
  15.  
  16. void Read()
  17. {
  18.     cin >> n;
  19.  
  20.     for (int i = 1; i <= n; ++i)
  21.         cin >> a[i];
  22. }
  23.  
  24. int Median(int n, int a[N])
  25. {
  26.     for (int i = 1; i <= n; ++i)
  27.         c[i] = a[i];
  28.  
  29.     sort(c + 1, c + n + 1);
  30.  
  31.     return c[(n + 1) / 2];
  32. }
  33.  
  34. void Sub_1()
  35. {
  36.     int ans = 0, v = Median(n, a);
  37.  
  38.     for (int i = 1; i <= n; ++i)
  39.         for (int j = i; j <= n; ++j)
  40.         {
  41.             m = 0;
  42.  
  43.             for (int t = i; t <= j; ++t)
  44.                 b[++m] = a[t];
  45.  
  46.             if (Median(m, b) == v)
  47.                 ++ans;
  48.         }
  49.  
  50.     cout << ans;
  51. }
  52.  
  53. struct FenwickTree
  54. {
  55.     int n, a[N];
  56.  
  57.     FenwickTree()
  58.     {
  59.         fill_n(a, N, 0);
  60.     }
  61.  
  62.     void Update(int p, int v)
  63.     {
  64.         for (; p <= n; p += p & -p)
  65.             a[p] += v;
  66.     }
  67.  
  68.     int Get(int p)
  69.     {
  70.         int ans(0);
  71.         for (; p; p -= p & -p)
  72.             ans += a[p];
  73.         return ans;
  74.     }
  75. } h;
  76.  
  77. ll f(int v)
  78. {
  79.     g[id[0] = 0] = 0;
  80.     for (int i = 1; i <= n; ++i)
  81.         g[id[i] = i] = g[i - 1] - 1 + 2 * (a[i] >= v);
  82.  
  83.     sort(id, id + n + 1, [&](const int &x, const int &y)
  84.          { return g[x] < g[y] || (g[x] == g[y] && x > y); });
  85.  
  86.     h.n = n + 1;
  87.  
  88.     ll ans(0);
  89.  
  90.     for (int i = 0; i <= n; ++i)
  91.     {
  92.         ans += h.Get(id[i] + 1);
  93.         h.Update(id[i] + 1, 1);
  94.     }
  95.     for (int i = 0; i <= n; ++i)
  96.         h.Update(id[i] + 1, -1);
  97.  
  98.     return ans;
  99. }
  100.  
  101. void Sub_3()
  102. {
  103.     int v = Median(n, a);
  104.  
  105.     cout << f(v) - f(v + 1) << "\n";
  106. }
  107.  
  108. int32_t main()
  109. {
  110.     ios_base::sync_with_stdio(0);
  111.     cin.tie(0);
  112.     cout.tie(0);
  113.  
  114.     if (fopen(task ".INP", "r"))
  115.     {
  116.         freopen(task ".INP", "r", stdin);
  117.         freopen(task ".OUT", "w", stdout);
  118.     }
  119.  
  120.     Read();
  121.  
  122.     if (n <= 100)
  123.         Sub_1();
  124.     else
  125.         Sub_3();
  126. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement