Submission #776566

#	Time	Username	Problem	Language	Result	Execution time	Memory
776566		jamezzz	Fish (IOI08_fish)	C++17	70 / 100	758 ms	59480 KiB

fish

#include <bits/stdc++.h>
using namespace std;
 
#define sf scanf
#define pf printf
#define fi first
#define se second
#define pb push_back
#define sz(x) ((int)x.size())
#define all(x) x.begin(),x.end()
#define LINF 1023456789123456789
typedef long long ll;
typedef pair<int,int> ii;
typedef pair<ll,int> li;
 
#define maxn 500005
#define maxk 7005
 
int n,k,m,cnt[maxk],bad[maxk],pred[maxk];
bool num[maxk][maxk];
ll mn[maxn];
vector<ii> v;
vector<int> stuff[maxk];
 
int main(){
	sf("%d%d%d",&n,&k,&m);
	assert(n<=500000&&k<=7000);
    for(int i=0;i<n;++i){
		int l,c;
		sf("%d%d",&l,&c);
		v.pb({l,c});
		++cnt[c];
    }
    sort(all(v));
    for(int i=0;i<n;++i){
	    int c=v[i].se;
	    stuff[c].pb(i);
    }
    for(int i=1;i<=k;++i){
		for(int j:stuff[i]){
			if(v[j].fi*2<=v[stuff[i].back()].fi)++pred[i];
		}
	}
    int ptr=n-1;
    ll ans=0;
    fill(mn,mn+n+1,1);
    for(int i=n-1;i>=0;--i){
		int c=v[i].se;
		if(bad[c])continue;
		while(ptr>=0&&v[i].fi<2*v[ptr].fi){
			--cnt[v[ptr].se];
			--ptr;
		}
		vector<int> undo;
		for(int j=1;j<=k;++j){
			if(c==j)continue;
			int x=bad[j]?cnt[c]:0;
			if(!bad[j]||num[c][j]){
				mn[x+1]*=(cnt[j]+1);
				mn[x+1]%=m;
				undo.pb(x+1);
			}
		}
		ll add=1;
		for(int j=1;j<=cnt[c]+1;++j){
			add*=mn[j];
			add%=m;
			ans+=add;
			ans%=m;
		}
		for(int x:undo)mn[x]=1;
		for(int j=1;j<=k;++j){
			if(pred[j]==cnt[j])num[j][c]=true;
		}
		bad[c]=1;
		//pf("%lld\n",ans);
    }
    pf("%lld\n",ans);
}

Compilation message (stderr)

fish.cpp: In function 'int main()':
fish.cpp:26:4: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   26 |  sf("%d%d%d",&n,&k,&m);
      |    ^
fish.cpp:30:5: warning: ignoring return value of 'int scanf(const char*, ...)' declared with attribute 'warn_unused_result' [-Wunused-result]
   30 |   sf("%d%d",&l,&c);
      |     ^

• Subtask 1: 4 / 4
• Subtask 2: 4 / 4
• Subtask 3: 4 / 4
• Subtask 4: 4 / 4
• Subtask 5: 4 / 4
• Subtask 6: 5 / 5
• Subtask 7: 6 / 6
• Subtask 8: 6 / 6
• Subtask 9: 6 / 6
• Subtask 10: 6 / 6
• Subtask 11: 7 / 7
• Subtask 12: 7 / 7
• Subtask 13: 7 / 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