Submission #2107175


Source Code Expand

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <vector>
#define int LL
using namespace std;
typedef long long LL;
const LL mo=1e9+7;
const int INF=1e18;
LL fac[500001],inv[500001];
int n,x,y,fa[500001],p[500001];
vector <int> v[500001],id[500001],ans[500001];
struct data {
    int c,w;
}a[500001];
bool cmp(int x,int y) {
    int ux,vx;
    if (v[x].size()==0) ux=INF;else ux=v[x][0];
    if (v[y].size()==0) vx=INF;else vx=v[y][0];
    return ux<vx;
}
LL mul(LL x,LL y) {
    if (y==0) return 1;
    LL now=mul(x,y/2);
    now=now*now%mo;
    if (y%2) return now*x%mo;
    return now;
}
void pre() {
    fac[0]=1;
    for (int i=1;i<=n;i++) fac[i]=fac[i-1]*i%mo;
    inv[n]=mul(fac[n],mo-2);
    for (int i=n-1;i>=0;i--) inv[i]=inv[i+1]*(i+1)%mo;
}
int getf(int x) {
    if (fa[x]==x) return x;
    fa[x]=getf(fa[x]);
    return fa[x];
}
void unio(int u,int v) {
    int fu=getf(u),fv=getf(v);
    if (fu==fv) return;
    fa[fu]=fv;
}
main() {
    scanf("%lld%lld%lld",&n,&x,&y);
    pre();
    for (int i=1;i<=n;i++) {
        scanf("%lld%lld",&a[i].c,&a[i].w);
        v[a[i].c].push_back(a[i].w);
        id[a[i].c].push_back(i);
    }
    for (int i=1;i<=n;i++)
        sort(v[i].begin(),v[i].end());
    for (int i=1;i<=n;i++) fa[i]=p[i]=i;
    sort(p+1,p+n+1,cmp);
    for (int i=1;i<=n;i++) {
        if (v[i].size()==0) continue;
        int wx=max(x-v[i][0],y-v[p[1]][0]);
        int lx=0,rx=v[i].size()-1;
        while (lx<rx) {
            int mid=(lx+rx+1)/2;
            if (v[i][mid]<=wx) lx=mid;
            else rx=mid-1;
        }
        for (int j=0;j<lx;j++) unio(id[i][j],id[i][j+1]);
    }
    int wx=y-v[p[1]][0];
    int lx=1,rx=n;
    while (lx<rx) {
        int mid=(lx+rx+1)/2;
        int now;
        if (!v[p[mid]].size()) now=INF;
        else now=v[p[mid]][0];
        if (now<=wx) lx=mid;
        else rx=mid-1;
    }
    for (int i=1;i<lx;i++) unio(id[p[i]][0],id[p[i+1]][0]);
    for (int i=1;i<=n;i++) ans[getf(i)].push_back(a[i].c);
    LL ansx=1;
    for (int i=1;i<=n;i++) {
        if (getf(i)!=i) continue;
        sort(ans[i].begin(),ans[i].end());
        ansx=ansx*fac[ans[i].size()]%mo;
        for (int j=0;j<ans[i].size();) {
            int kx=j;
            while (ans[i][j]==ans[i][kx]&&kx<ans[i].size()) kx++;
            ansx=ansx*inv[kx-j]%mo;
            j=kx;
        }
    }
    printf("%lld\n",ansx);
    return 0;
}

Submission Info

Submission Time
Task D - Colorful Balls
User langsike
Language C++14 (GCC 5.4.1)
Score 0
Code Size 2529 Byte
Status WA
Exec Time 179 ms
Memory 70268 KB

Compile Error

./Main.cpp: In function ‘int main()’:
./Main.cpp:48:35: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
     scanf("%lld%lld%lld",&n,&x,&y);
                                   ^
./Main.cpp:51:42: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
         scanf("%lld%lld",&a[i].c,&a[i].w);
                                          ^

Judge Result

Set Name Sample All
Score / Max Score 0 / 0 0 / 1000
Status
AC × 3
AC × 54
WA × 3
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 14 ms 43264 KB
00_example_02.txt AC 14 ms 43264 KB
00_example_03.txt AC 14 ms 43264 KB
01.txt AC 14 ms 43264 KB
02.txt AC 14 ms 43264 KB
03.txt AC 18 ms 45824 KB
04.txt AC 14 ms 43392 KB
05.txt AC 15 ms 43520 KB
06.txt AC 14 ms 43264 KB
07.txt AC 73 ms 57852 KB
08.txt AC 14 ms 43264 KB
09.txt AC 38 ms 52484 KB
10.txt AC 32 ms 49280 KB
11.txt AC 15 ms 43264 KB
12.txt AC 14 ms 43264 KB
13.txt AC 19 ms 45952 KB
14.txt AC 14 ms 43264 KB
15.txt AC 41 ms 50584 KB
16.txt AC 73 ms 58752 KB
17.txt AC 22 ms 46336 KB
18.txt AC 34 ms 49788 KB
19.txt AC 58 ms 56696 KB
20.txt AC 170 ms 68596 KB
21.txt AC 179 ms 68468 KB
22.txt AC 167 ms 69240 KB
23.txt AC 172 ms 67368 KB
24.txt AC 161 ms 69756 KB
25.txt AC 171 ms 68084 KB
26.txt AC 175 ms 67828 KB
27.txt AC 174 ms 68352 KB
28.txt AC 176 ms 68244 KB
29.txt AC 172 ms 68852 KB
30.txt AC 164 ms 70016 KB
31.txt AC 172 ms 67604 KB
32.txt AC 167 ms 68728 KB
33.txt AC 178 ms 67360 KB
34.txt AC 150 ms 67576 KB
35.txt WA 93 ms 62324 KB
36.txt AC 122 ms 63100 KB
37.txt WA 96 ms 62984 KB
38.txt WA 92 ms 61812 KB
39.txt AC 101 ms 62456 KB
40.txt AC 155 ms 68468 KB
41.txt AC 154 ms 68724 KB
42.txt AC 153 ms 69236 KB
43.txt AC 152 ms 70136 KB
44.txt AC 158 ms 67956 KB
45.txt AC 155 ms 70268 KB
46.txt AC 168 ms 68724 KB
47.txt AC 91 ms 63844 KB
48.txt AC 93 ms 64100 KB
49.txt AC 92 ms 63972 KB
50.txt AC 109 ms 68332 KB
51.txt AC 110 ms 68844 KB
52.txt AC 90 ms 64240 KB
53.txt AC 89 ms 64240 KB
54.txt AC 89 ms 64240 KB