Submission #380250

# Submission time Handle Problem Language Result Execution time Memory
380250 2021-03-20T17:10:40 Z vishesh312 Vudu (COCI15_vudu) C++17
42 / 140
1000 ms 50068 KB
#include "bits/stdc++.h"
using namespace std;
 
using ll = long long;
 
#include <ext/pb_ds/assoc_container.hpp>
using namespace __gnu_pbds;
using ordered_set = tree<int, null_type, less_equal<int>, rb_tree_tag, tree_order_statistics_node_update>;
 
#define all(x) begin(x), end(x)
#define sz(x) (int)x.size()
 
const int mod = 1e9+7;
 
void solve(int tc) {
    int n;
    cin >> n;
    vector<pair<int, int>> v(n+1);
    for (int i = 1; i <= n; ++i) {
        cin >> v[i].first;
        v[i].second = i;
    }
    vector<int> a(n+1);
    ll p;
    cin >> p;
    for (int i = 1; i <= n; ++i) {
        v[i].first += v[i-1].first - p;
    }
    sort(all(v));
    ll cur = 0;
    for (int i = 0; i <= n; ++i) {
        if (i) if (v[i].first == v[i-1].first) --cur;
        a[v[i].second] = cur;
        ++cur;
    }
    ll ans = 0;
    ordered_set s;
    s.insert(a[0]-1);
    for (int i = 1; i <= n; ++i) {
        ans += s.order_of_key(a[i]);
        s.insert(a[i]-1);
    }
    cout << ans << '\n';
}
 
signed main() {
    ios_base::sync_with_stdio(false); cin.tie(NULL);
    int tc = 1;
    //cin >> tc;
    for (int i = 1; i <= tc; ++i) solve(i);
    return 0;
}
 
# Verdict Execution time Memory Grader output
1 Correct 4 ms 748 KB Output is correct
2 Correct 3 ms 748 KB Output is correct
3 Correct 3 ms 620 KB Output is correct
4 Execution timed out 1089 ms 50068 KB Time limit exceeded
5 Incorrect 685 ms 32620 KB Output isn't correct
6 Execution timed out 1094 ms 49148 KB Time limit exceeded
7 Execution timed out 1095 ms 49388 KB Time limit exceeded
8 Execution timed out 1059 ms 45788 KB Time limit exceeded
9 Execution timed out 1089 ms 49516 KB Time limit exceeded
10 Execution timed out 1091 ms 49132 KB Time limit exceeded