Submission #1461678


Source Code Expand

// This amazing code is by Eric Sunli Chen.
#include <algorithm>
#include <bitset>
#include <cmath>
#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <ctime>
#include <iomanip>
#include <iostream>
#include <map>
#include <queue>
#include <set>
#include <string>
#include <utility>
#include <vector>
using namespace std;
template<typename T> void get_int(T &x)
{
	char t=getchar();
	bool neg=false;
	x=0;
	for(; (t>'9'||t<'0')&&t!='-'; t=getchar());
	if(t=='-')neg=true,t=getchar();
	for(; t<='9'&&t>='0'; t=getchar())x=x*10+t-'0';
	if(neg)x=-x;
}
template<typename T> void print_int(T x)
{
	if(x<0)putchar('-'),x=-x;
	short a[20]= {},sz=0;
	while(x>0)a[sz++]=x%10,x/=10;
	if(sz==0)putchar('0');
	for(int i=sz-1; i>=0; i--)putchar('0'+a[i]);
}
#define ff first
#define ss second
#define pb push_back
#define mp make_pair
#define get1(a) get_int(a)
#define get2(a,b) get1(a),get1(b)
#define get3(a,b,c) get1(a),get2(b,c)
#define printendl(a) print_int(a),puts("")
typedef long long LL;
typedef unsigned long long uLL;
typedef pair<int,int> pii;
const int inf=0x3f3f3f3f;
const LL Linf=1ll<<61;
const double pi=acos(-1.0);

const int mod=1e9+7;

int power(int x,int y)
{
	int ret=1;
	while(y)
	{
		if(y&1)ret=1ll*ret*x%mod;
		x=1ll*x*x%mod;
		y>>=1;
	}
	return ret;
}
int n,x,y,c[200111],w[200111];
vector<pii> v[200111];
map<int,int> cnt[200111];
int sz[200111],f[200111],fac[200111],invf[200111];
int gf(int x){return x==f[x]?x:f[x]=gf(f[x]);}
void un(int x,int y)
{
	x=gf(x);y=gf(y);
	if(x==y)return;
	if(sz[x]<sz[y])swap(x,y);
	f[y]=x;
	sz[x]+=sz[y];
}
int main()
{
	fac[0]=invf[0]=1;
	for(int i=1;i<200111;i++)
	{
		fac[i]=1ll*fac[i-1]*i%mod;
		invf[i]=power(fac[i],mod-2);
	}
	
	get3(n,x,y);
	for(int i=1;i<=n;i++)
	{
		f[i]=i;sz[i]=1;
		get2(c[i],w[i]);
		v[c[i]].pb(mp(w[i],i));
	}
	int mn=0,mn2=0;
	for(int i=1;i<=n;i++)
	{
		if((int)v[i].size()==0)continue;
		sort(v[i].begin(),v[i].end());
		for(int j=1;j<(int)v[i].size();j++)
			if(v[i][0].ff+v[i][j].ff<=x)
				un(v[i][0].ss,v[i][j].ss);
		if(mn==0||v[mn][0].ff>v[i][0].ff)
		{
			mn2=mn;
			mn=i;
		}
		else if(mn2==0||v[mn2][0].ff>v[i][0].ff)mn2=i;
	}
	for(int i=1;i<=n;i++)
	{
		if((int)v[i].size()==0)continue;
		if(i!=mn2&&mn2>0&&v[i][0].ff+v[mn2][0].ff<=y)un(v[i][0].ss,v[mn2][0].ss);
		if(i!=mn&&mn>0&&v[i][0].ff+v[mn][0].ff<=y)un(v[i][0].ss,v[mn][0].ss);
	}
	for(int i=1;i<=n;i++)cnt[gf(i)][c[i]]++;
	int ans=1;
	for(int i=1;i<=n;i++)if(f[i]==i)
	{
		ans=1ll*ans*fac[sz[i]]%mod;
		for(map<int,int>::iterator it=cnt[i].begin();it!=cnt[i].end();it++)
			ans=1ll*ans*invf[it->ss]%mod;
	}
	printendl(ans);
	return 0;
}

Submission Info

Submission Time
Task D - Colorful Balls
User OhWeOnFire
Language C++14 (GCC 5.4.1)
Score 0
Code Size 2721 Byte
Status WA
Exec Time 158 ms
Memory 32896 KB

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 0 / 1000
Status
AC × 3
AC × 21
WA × 36
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 38 ms 18176 KB
00_example_02.txt AC 39 ms 18176 KB
00_example_03.txt AC 39 ms 18176 KB
01.txt WA 39 ms 18176 KB
02.txt WA 39 ms 18176 KB
03.txt WA 41 ms 18560 KB
04.txt WA 39 ms 18176 KB
05.txt WA 40 ms 18432 KB
06.txt AC 39 ms 18176 KB
07.txt WA 80 ms 23424 KB
08.txt AC 39 ms 18176 KB
09.txt WA 52 ms 20224 KB
10.txt WA 50 ms 19456 KB
11.txt WA 39 ms 18176 KB
12.txt WA 39 ms 18176 KB
13.txt WA 41 ms 18688 KB
14.txt WA 38 ms 18176 KB
15.txt WA 57 ms 20608 KB
16.txt WA 73 ms 24960 KB
17.txt WA 43 ms 18816 KB
18.txt WA 51 ms 19840 KB
19.txt WA 68 ms 23296 KB
20.txt AC 140 ms 30592 KB
21.txt WA 132 ms 32640 KB
22.txt AC 118 ms 31616 KB
23.txt WA 154 ms 32384 KB
24.txt WA 116 ms 32512 KB
25.txt AC 140 ms 30080 KB
26.txt WA 152 ms 32640 KB
27.txt AC 110 ms 29312 KB
28.txt WA 155 ms 30080 KB
29.txt AC 146 ms 30336 KB
30.txt WA 112 ms 32512 KB
31.txt WA 158 ms 31360 KB
32.txt AC 129 ms 30976 KB
33.txt WA 154 ms 30336 KB
34.txt AC 138 ms 28928 KB
35.txt WA 96 ms 27904 KB
36.txt AC 112 ms 23680 KB
37.txt WA 96 ms 28160 KB
38.txt WA 97 ms 27520 KB
39.txt AC 96 ms 23168 KB
40.txt AC 120 ms 30080 KB
41.txt AC 122 ms 31232 KB
42.txt WA 117 ms 32896 KB
43.txt AC 109 ms 32128 KB
44.txt WA 130 ms 32768 KB
45.txt WA 106 ms 32896 KB
46.txt AC 122 ms 30848 KB
47.txt AC 87 ms 23664 KB
48.txt AC 93 ms 26608 KB
49.txt AC 89 ms 24688 KB
50.txt WA 87 ms 32244 KB
51.txt WA 85 ms 32244 KB
52.txt WA 80 ms 25460 KB
53.txt WA 79 ms 25460 KB
54.txt WA 80 ms 25460 KB