oj.uz

Submission #653243

# Submission time Handle Problem Language Result Execution time Memory
653243 2022-10-26T09:46:36 Z Vladth11 Fish (IOI08_fish) C++14
80 / 100
 407 ms 65536 KB 
#include <iostream>
#include <vector>
#include <set>
#include <unordered_set>
#include <algorithm>

#define debug(x) cerr << #x << " " << x << "\n"
#define debugs(x) cerr << #x << " " << x << " "

#pragma GCC optimize("Ofast")

using namespace std;
typedef long long ll;
typedef pair <ll, ll> pii;

const ll bSize = 11;
const ll bits = 23;
const ll NMAX = 500001;

ll MOD;
int dm;

class All {
public:
    ll all[NMAX * 4];
    void init() {
        for(ll i = 0; i < NMAX * 4; i++)
            all[i] = 1;
    }
    void add(ll node, ll st, ll dr, ll l, ll x) {
        if(st == dr) {
            all[node] += x;
            all[node] = max(all[node], 1LL);
            return;
        }
        ll mid = (st + dr) / 2;
        if(l <= mid) {
            add(node * 2, st, mid, l, x);
        } else {
            add(node * 2 + 1, mid + 1, dr, l, x);
        }
        all[node] = all[node * 2] * all[node * 2 + 1];
        all[node] %= MOD;
    }
    ll query(ll node, ll st, ll dr, ll a, ll b){
        if(a <= st && dr <= b)
            return all[node];
        int mid = (st + dr) / 2;
        ll prod = 1;
        if(a <= mid){
            prod *= query(node * 2, st, mid, a, b);
            prod %= MOD;
        }
        if(b > mid){
            prod *= query(node * 2 + 1, mid + 1, dr, a, b);
            prod %= MOD;
        }
        return prod;
    }
    void update(ll node, ll st, ll dr, ll l, ll x) {
        if(st == dr) {
            dm = all[node];
            all[node] = x;
            //all[node] = max(all[node], 1);
            return;
        }
        ll mid = (st + dr) / 2;
        if(l <= mid) {
            update(node * 2, st, mid, l, x);
        } else {
            update(node * 2 + 1, mid + 1, dr, l, x);
        }
        all[node] = all[node * 2] * all[node * 2 + 1];
        all[node] %= MOD;
    }
} all, banned;

ll imp[NMAX];
ll col[NMAX];
ll maxi[NMAX];
pii v[NMAX];
int f[NMAX];
pii cine[NMAX];
vector <int> g[NMAX];

int afla(int x, int col){
    int r = -1;
    int pas = (1 << bits);
    while(pas){
        if(r + pas < g[col].size() && g[col][r + pas] * 2 <= x)
            r += pas;
        pas /= 2;
    }
    return r;
}

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    ll n, k, i;
    cin >> n >> k >> MOD;
    for(i = 1; i <= n; i++) {
        cin >> v[i].first >> v[i].second;
        g[v[i].second].push_back(v[i].first);
    }
    sort(v + 1, v + n + 1);
    for(i = n; i >= 1; i--) {
        if(!col[v[i].second]) {
            imp[i] = 1;
        }
        col[v[i].second]++;
    }
    int cnt = 0;
    for(i = 1; i <= n; i++){
        if(imp[i]){
            cine[++cnt] = v[i];
            f[cine[cnt].second] = cnt;
        }
    }
    all.init();
    banned.init();
    for(i = 1; i <= n; i++){
        sort(g[i].begin(), g[i].end());
        maxi[i] = col[i];
        if(!f[i]) continue;
        all.update(1, 1, n, f[i], col[i] + 1);
        banned.update(1, 1, n, f[i], col[i] + 1);
    }
    ll dr = n, st = n;
    ll rez = 0;
    for(i = n; i >= 1; i--){
        while(st >= 1 && v[st].first * 2 > v[i].first){
            all.add(1, 1, n, f[v[st].second], -1);
            banned.add(1, 1, n, f[v[st].second], -1);
            col[v[st].second]--;
            st--;
        }
        if(i == n){
            for(ll j = 1; j <= n; j++){
                maxi[j] = col[j];
            }
        }
        if(imp[i]){
            if(i != n){
                all.update(1, 1, n, f[v[i].second], 1);
                int r = f[v[i].second];
                int pas = (1 << bits);
                while(pas){
                    if(r + pas <= k && v[i].first * 2 > cine[r + pas].first && col[v[i].second] - 1 == afla(cine[r + pas].first, v[i].second))
                        r += pas;
                    pas /= 2;
                }
                rez += all.query(1, 1, n, 1, r);
                all.update(1, 1, n, f[v[i].second], col[v[i].second] + 1);
                banned.add(1, 1, n, f[v[i].second], -1);
                rez %= MOD;
            }
            if(col[v[i].second] > 0 || i == n) /// avem frecventa mai mare ca 1..0
                rez += banned.all[1];
            rez %= MOD;
            banned.update(1, 1, n, f[v[i].second], 1);
        }
    }
    cout << rez;
    return 0;
}

Compilation message

fish.cpp: In function 'int afla(int, int)':
fish.cpp:90:20: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   90 |         if(r + pas < g[col].size() && g[col][r + pas] * 2 <= x)
      |            ~~~~~~~~^~~~~~~~~~~~~~~
fish.cpp: In function 'int main()':
fish.cpp:130:8: warning: unused variable 'dr' [-Wunused-variable]
  130 |     ll dr = n, st = n;
      |        ^~

Subtask #1
4.0 / 4.0

# Verdict Execution time Memory Grader output
1 Correct 20 ms 43364 KB Output is correct

Subtask #2
4.0 / 4.0

# Verdict Execution time Memory Grader output
1 Correct 22 ms 43348 KB Output is correct

Subtask #3
4.0 / 4.0

# Verdict Execution time Memory Grader output
1 Correct 19 ms 43344 KB Output is correct

Subtask #4
4.0 / 4.0

# Verdict Execution time Memory Grader output
1 Correct 19 ms 43348 KB Output is correct

Subtask #5
4.0 / 4.0

# Verdict Execution time Memory Grader output
1 Correct 18 ms 43348 KB Output is correct
2 Correct 21 ms 43440 KB Output is correct

Subtask #6
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 20 ms 43432 KB Output is correct
2 Correct 339 ms 58516 KB Output is correct

Subtask #7
6.0 / 6.0

# Verdict Execution time Memory Grader output
1 Correct 21 ms 43460 KB Output is correct

Subtask #8
6.0 / 6.0

# Verdict Execution time Memory Grader output
1 Correct 24 ms 43656 KB Output is correct

Subtask #9
6.0 / 6.0

# Verdict Execution time Memory Grader output
1 Correct 145 ms 50308 KB Output is correct
2 Correct 173 ms 53972 KB Output is correct

Subtask #10
6.0 / 6.0

# Verdict Execution time Memory Grader output
1 Correct 24 ms 43604 KB Output is correct
2 Correct 24 ms 43668 KB Output is correct
3 Correct 26 ms 43700 KB Output is correct

Subtask #11
7.0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 215 ms 53452 KB Output is correct
2 Correct 351 ms 64364 KB Output is correct
3 Correct 351 ms 64576 KB Output is correct

Subtask #12
7.0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 285 ms 59908 KB Output is correct
2 Correct 366 ms 65512 KB Output is correct

Subtask #13
7.0 / 7.0

# Verdict Execution time Memory Grader output
1 Correct 211 ms 55952 KB Output is correct
2 Correct 361 ms 59332 KB Output is correct

Subtask #14
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 403 ms 58876 KB Output is correct

Subtask #15
5.0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 395 ms 60796 KB Output is correct

Subtask #16
0 / 5.0

# Verdict Execution time Memory Grader output
1 Correct 407 ms 58148 KB Output is correct
2 Runtime error 336 ms 65536 KB Execution killed with signal 9
3 Halted 0 ms 0 KB -

Subtask #17
0 / 5.0

# Verdict Execution time Memory Grader output
1 Runtime error 299 ms 65536 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -

Subtask #18
0 / 5.0

# Verdict Execution time Memory Grader output
1 Runtime error 235 ms 65536 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -

Subtask #19
0 / 5.0

# Verdict Execution time Memory Grader output
1 Runtime error 235 ms 65536 KB Execution killed with signal 9
2 Halted 0 ms 0 KB -