Submission #734017

#	Time	Username	Problem	Language	Result	Execution time	Memory
734017		lorenzoferrari	Fish (IOI08_fish)	C++17	0 / 100	3063 ms	44372 KiB

fish

#include "bits/stdc++.h"
using namespace std;

#define int long long

static constexpr int N = 5e5+5;

struct fish {
    int l;
    int g;
    bool operator<(const fish& oth) {
        return l < oth.l;
    }
};

static int f[N];    // frq of each colour
static int ff[N];   // frq of each frq
set<int> s;         // {j | ff[j] != 0}
vector<int> pw[N];

static void inc(int c, int m) {
    if (ff[f[c]] == 1) {
        s.erase(f[c]);
    }
    --ff[f[c]];
    ++f[c];
    ++ff[f[c]];
    if (ff[f[c]] == 1) {
        s.insert(f[c]);
    }
    pw[f[c]+1].emplace_back(pw[f[c]+1].back() * ((f[c]+1) % m) % m);
}

static int calc(int m) {
    int ans = 1;
    for (int j : s) {
        ans = ans * pw[j+1][ff[j]] % m;
    }
    return ans;
}

signed main() {
    for (int i = 0; i < N; ++i) {
        pw[i].emplace_back(1);
    }

    int n; cin >> n;
    int k; cin >> k;
    int m; cin >> m;
    vector<fish> a(n);
    for (int i = 0; i < n; ++i) {
        cin >> a[i].l >> a[i].g; --a[i].g;
    }
    sort(begin(a), end(a));
    vector<int> last(k);
    for (int i = 0; i < n; ++i) {
        last[a[i].g] = i;
    }

    int ans = 0;
    vector<int> d(n);
    vector<int> rs;
    for (int r = 0, l = 0; r < n; ++r) {
        if (last[a[r].g] != r) continue;
        while (2*a[l].l <= a[r].l) {
            inc(a[l].g, m);
            ++l;
            continue;

            if (last[a[l].g] < r) {
                inc(a[l].g, m);
            } else {
                ++d[last[a[l].g]];
            }
            ++l;
        }
        while (d[r]--) {
            inc(a[r].g, m);
        }

        rs.push_back(r);

        int cur = 1;
        for (int i : rs) {
            cur = (cur * (f[a[i].g]+1)) % m;
        }
        ans += cur;
        ans %= m;
        
        // ans = (ans + calc(m)) % m;
    }
    
    cout << ans << "\n";
}

Compilation message (stderr)

fish.cpp:34:12: warning: 'long long int calc(long long int)' defined but not used [-Wunused-function]
   34 | static int calc(int m) {
      |            ^~~~

• 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