Submission #419089

#	Time	Username	Problem	Language	Result	Execution time	Memory
419089		acm	Vudu (COCI15_vudu)	C++14	140 / 140	436 ms	41380 KiB

vudu

#include <bits/stdc++.h>
#define speed                   \
  ios_base::sync_with_stdio(0); \
  cin.tie(0);                   \
  cout.tie(0);
#define precision     \
  cout.precision(30); \
  cerr.precision(10);
#define ll long long
#define ld long double
#define pll pair<ll, ll>
#define pii pair<int, int>
#define forn(n) for (int i = 1; i <= n; i++)
#define forlr(l, r) for (int i = l; i != r; (l > r ? i-- : i++))
#define pb(x) push_back(x)
#define sz(x) (int)x.size()
#define mp(x, y) make_pair(x, y)
#define all(x) x.begin(), x.end()
#define pc(x) __builtin_popcount(x)
#define pcll(x) __builtin_popcountll(x)
#define F first
#define S second
using namespace std;
mt19937_64 rng(chrono::steady_clock::now().time_since_epoch().count());
void ioi(string name) {
  freopen((name + ".in").c_str(), "r", stdin);
  freopen((name + ".out").c_str(), "w", stdout);
}
ll n, k, t, a[1000005], p[1000005], f[1000005];
vector<ll> b;
void upd(ll x) {
  for (; x <= n + 1; x += (x & -x)) f[x]++;
}
ll get(ll x) {
  ll y = 0;
  for (; x > 0; x -= (x & -x)) y += f[x];
  return y;
}
int main() {
  speed;
  precision;
  // code
  cin >> n;
  for (int i = 1; i <= n; i++) cin >> a[i], p[i] = p[i - 1] + a[i];
  cin >> k;
  b.pb(0);
  for (int i = 1; i <= n; i++) p[i] -= k * i, b.pb(p[i]);
  sort(all(b));
  b.erase(unique(all(b)), b.end());
  upd(lower_bound(all(b), 0) - b.begin() + 1);
  for (int i = 1; i <= n; i++) {
    p[i] = lower_bound(all(b), p[i]) - b.begin() + 1;
    t += get(p[i]);
    upd(p[i]);
  }
  cout << t;
  // endl
#ifndef ONLINE_JUDGE
  cerr << "\nTime elapsed: " << 1.0 * clock() / CLOCKS_PER_SEC << " s.\n";
#endif
  return 0;
}

Compilation message (stderr)

vudu.cpp: In function 'void ioi(std::string)':
vudu.cpp:26:10: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   26 |   freopen((name + ".in").c_str(), "r", stdin);
      |   ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
vudu.cpp:27:10: warning: ignoring return value of 'FILE* freopen(const char*, const char*, FILE*)' declared with attribute 'warn_unused_result' [-Wunused-result]
   27 |   freopen((name + ".out").c_str(), "w", stdout);
      |   ~~~~~~~^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

• Subtask 1: 140 / 140