Submission #653087

#	Time	Username	Problem	Language	Result	Execution time	Memory
653087		Vladth11	Fish (IOI08_fish)	C++14	0 / 100	742 ms	48640 KiB

fish

#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 = 17;
const ll NMAX = 500001;

ll MOD;

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;
    }
    void update(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) {
            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 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;
    }
    sort(v + 1, v + n + 1);
    for(i = n; i >= 1; i--) {
        if(!col[v[i].second]) {
            imp[i] = 1;
        }
        col[v[i].second]++;
    }
    all.init();
    banned.init();
    for(i = 1; i <= n; i++){
        maxi[i] = col[i];
        all.update(1, 1, n, i, col[i] + 1);
        banned.update(1, 1, n, i, col[i] + 1);
    }
    ll dr = n + 1, 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, v[st].second, -1);
            banned.add(1, 1, n, 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]){
            rez += banned.all[1];
            rez %= MOD;
            if(v[i].first * 2 > v[n].first && i != n && maxi[v[i].second] == col[v[i].second]){
                all.update(1, 1, n, v[i].second, 1);
                rez += all.all[1];
                all.update(1, 1, n, v[i].second, col[v[i].second] + 1);
                rez %= MOD;
            }
        }
        //all.update(1, 1, n, v[i].second, 1);
        banned.update(1, 1, n, v[i].second, 1);
    }
    cout << rez;
    return 0;
}

Compilation message (stderr)

fish.cpp: In function 'int main()':
fish.cpp:89:8: warning: unused variable 'dr' [-Wunused-variable]
   89 |     ll dr = n + 1, st = n;
      |        ^~

• Subtask 1: 0 / 4
• Subtask 2: 0 / 4
• Subtask 3: 0 / 4
• Subtask 4: 0 / 4
• Subtask 5: 0 / 4
• Subtask 6: 0 / 5
• Subtask 7: 0 / 6
• Subtask 8: 0 / 6
• Subtask 9: 0 / 6
• Subtask 10: 0 / 6
• Subtask 11: 0 / 7
• Subtask 12: 0 / 7
• Subtask 13: 0 / 7
• Subtask 14: 0 / 5
• Subtask 15: 0 / 5
• Subtask 16: 0 / 5
• Subtask 17: 0 / 5
• Subtask 18: 0 / 5
• Subtask 19: 0 / 5