Submission #2207396


Source Code Expand

#include<bits/stdc++.h>
using namespace std;
#define mod 1000000007
#define MN 200005
#define ll long long
inline int read(){
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
	return x*f;
}
ll fac[MN],inv[MN];
int n,x,y,cnt=0,c[MN],w[MN],ge[MN];bool vis[MN];
vector<pair<int, int> >r[MN];
vector<pair<int, int> >p;
vector<int> v[MN],tmp;
void dfs(int u){
    vis[u]=1;
    ++cnt,++ge[c[u]],tmp.push_back(c[u]);
    for(int i=0;i<v[u].size();i++)
        if(!vis[v[u][i]]) dfs(v[u][i]);
}
int main(){
	n=read(),x=read(),y=read();
	fac[0]=fac[1]=inv[0]=inv[1]=1;
    for(ll i=2;i<=n;i++){
        fac[i]=fac[i-1]*i%mod;
        inv[i]=(mod-mod/i)*inv[mod%i]%mod;
    }
    for(int i=2;i<=n;i++) inv[i]=inv[i-1]*inv[i]%mod;
    for(int i=1;i<=n;i++) c[i]=read(),w[i]=read(),r[c[i]].push_back(make_pair(w[i],i));
    for(int i=1;i<=n;i++)if(r[i].size()){
    	sort(r[i].begin(),r[i].end());p.push_back(r[i][0]);
    	for(int j=1;j<r[i].size();j++)
    		if(r[i][j].first+r[i][0].first<=x)
    			v[r[i][0].second].push_back(r[i][j].second),
    			v[r[i][j].second].push_back(r[i][0].second);
	}
    sort(p.begin(),p.end());
    int id=p[0].second;
    for(int i=1;i<=n;i++){
    	if(c[i]==c[id]||w[i]+w[id]>y) continue;
    	v[id].push_back(i),v[i].push_back(id);
	}
    if(p.size()>1){
    	id=p[1].second;int cmin=c[p[0].second];
    	for(int i=0;i<r[cmin].size();i++)
    		if(r[cmin][i].first+w[id]<=y)
			v[r[cmin][i].second].push_back(id),v[id].push_back(r[cmin][i].second);
	}
	dfs(p[0].second);
    ll ans=fac[cnt];
    for(int i=0;i<tmp.size();i++)
        ans=(ans*inv[ge[tmp[i]]])%mod,ge[tmp[i]]=0;
    printf("%lld\n",ans);
    return 0;
}

Submission Info

Submission Time
Task D - Colorful Balls
User vjudge4
Language C++14 (GCC 5.4.1)
Score 1000
Code Size 1739 Byte
Status AC
Exec Time 106 ms
Memory 28660 KB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 1000 / 1000
Status
AC × 3
AC × 57
Set Name Test Cases
Sample 00_example_01.txt, 00_example_02.txt, 00_example_03.txt
All 00_example_01.txt, 00_example_02.txt, 00_example_03.txt, 01.txt, 02.txt, 03.txt, 04.txt, 05.txt, 06.txt, 07.txt, 08.txt, 09.txt, 10.txt, 11.txt, 12.txt, 13.txt, 14.txt, 15.txt, 16.txt, 17.txt, 18.txt, 19.txt, 20.txt, 21.txt, 22.txt, 23.txt, 24.txt, 25.txt, 26.txt, 27.txt, 28.txt, 29.txt, 30.txt, 31.txt, 32.txt, 33.txt, 34.txt, 35.txt, 36.txt, 37.txt, 38.txt, 39.txt, 40.txt, 41.txt, 42.txt, 43.txt, 44.txt, 45.txt, 46.txt, 47.txt, 48.txt, 49.txt, 50.txt, 51.txt, 52.txt, 53.txt, 54.txt
Case Name Status Exec Time Memory
00_example_01.txt AC 5 ms 14592 KB
00_example_02.txt AC 5 ms 14592 KB
00_example_03.txt AC 6 ms 14592 KB
01.txt AC 5 ms 14592 KB
02.txt AC 5 ms 14592 KB
03.txt AC 8 ms 14976 KB
04.txt AC 5 ms 14592 KB
05.txt AC 6 ms 14720 KB
06.txt AC 6 ms 14592 KB
07.txt AC 40 ms 19704 KB
08.txt AC 5 ms 14592 KB
09.txt AC 26 ms 17916 KB
10.txt AC 19 ms 16512 KB
11.txt AC 5 ms 14592 KB
12.txt AC 6 ms 14592 KB
13.txt AC 8 ms 15104 KB
14.txt AC 5 ms 14592 KB
15.txt AC 22 ms 17404 KB
16.txt AC 33 ms 17656 KB
17.txt AC 11 ms 15488 KB
18.txt AC 20 ms 16512 KB
19.txt AC 32 ms 18044 KB
20.txt AC 97 ms 25460 KB
21.txt AC 81 ms 24568 KB
22.txt AC 83 ms 22772 KB
23.txt AC 89 ms 27812 KB
24.txt AC 75 ms 21876 KB
25.txt AC 95 ms 25844 KB
26.txt AC 86 ms 28148 KB
27.txt AC 86 ms 24180 KB
28.txt AC 105 ms 28276 KB
29.txt AC 106 ms 26996 KB
30.txt AC 80 ms 21120 KB
31.txt AC 94 ms 28276 KB
32.txt AC 84 ms 23668 KB
33.txt AC 105 ms 28020 KB
34.txt AC 94 ms 27380 KB
35.txt AC 75 ms 24820 KB
36.txt AC 91 ms 25368 KB
37.txt AC 75 ms 25080 KB
38.txt AC 73 ms 24564 KB
39.txt AC 84 ms 24840 KB
40.txt AC 88 ms 28660 KB
41.txt AC 83 ms 26608 KB
42.txt AC 75 ms 24948 KB
43.txt AC 74 ms 23028 KB
44.txt AC 83 ms 28312 KB
45.txt AC 68 ms 22272 KB
46.txt AC 81 ms 25968 KB
47.txt AC 76 ms 23280 KB
48.txt AC 61 ms 20208 KB
49.txt AC 68 ms 21744 KB
50.txt AC 58 ms 26484 KB
51.txt AC 56 ms 24436 KB
52.txt AC 61 ms 22772 KB
53.txt AC 62 ms 22772 KB
54.txt AC 66 ms 22772 KB