Submission #963220

#	Time	Username	Problem	Language	Result	Execution time	Memory
963220		fzyzzz_z	Fish (IOI08_fish)	C++17	0 / 100	194 ms	7476 KiB

fish

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

using ll = long long; 

const int N = 500005; 

int m; 

struct seg {
	vector<int> t; 
	seg(vector<int> f) { 
		t = vector<int>(2 * f.size()); 
		for (int i = 0; i < f.size(); ++i) t[i + f.size()] = f[i] + 1; 
		for (int i = f.size() - 1; i >= 1; --i) t[i] = ((ll)t[2 * i] * t[2 * i + 1]) % m; 
	} 
	void upd(int p, int v) {
		p += t.size() / 2; 
		t[p] += v;  
		t[p] = max(t[p], 1); 
		for (; p > 1; p /= 2) {
			t[p / 2] = ((ll)t[p] * t[p ^ 1]) % m;
		}
	}
};

int main() {
	ios::sync_with_stdio(false);
	cin.tie(nullptr);

	int n, k; 
	cin >> n >> k >> m; 

	vector<int> f(k, 0); 
	vector<pair<int, int>> a(n); 
	for (auto & [x, y]: a) {
		cin >> x >> y; 
		f[--y]++; 
	}
	seg s(f); 

	sort(a.begin(), a.end(), greater<pair<int, int>>()); 

	int ans = 0; 

	f = vector<int>(k, 0);
	for (int i = 0, j = 0; i < n; ++i) {
		if (f[a[i].second]) continue; 
		f[a[i].second] = 1; 

		while (a[j].first * 2 > a[i].first && j < n) {
			s.upd(a[j].second, -1); 
			j++; 
		}
		if (i != 0) s.upd(a[i].second, -n); 
		ans += s.t[1]; 
		// cout << s.t[1] << '\n';
		if (i == 0) s.upd(a[i].second, -n);
		ans %= m; 
	}

	cout << ans << '\n'; 

	return 0; 
}

Compilation message (stderr)

fish.cpp: In constructor 'seg::seg(std::vector<int>)':
fish.cpp:14:21: warning: comparison of integer expressions of different signedness: 'int' and 'std::vector<int>::size_type' {aka 'long unsigned int'} [-Wsign-compare]
   14 |   for (int i = 0; i < f.size(); ++i) t[i + f.size()] = f[i] + 1;
      |                   ~~^~~~~~~~~~

• 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