Submission #1679648


Source Code Expand

#include <cstdio>
#include <cctype>
#include <algorithm>
#define repu(i,x,y) for (int i=x; i<=y; ++i)
#define repe(i,x) for (edge *i=fst[x]; i; i=i->nxt)
using namespace std;

typedef long long LL;
const int p=1e9+7;
int n,x,y,c[200100],w[200100],fa[200100],s[200100],minw[200100],min1,min2,cnt[200100],inv[200100];
struct edge
{
    int v;
    edge *nxt;
} pool[200100],*tp=pool,*fst[200100];

int getint()
{
    char ch;
    while (!isdigit(ch=getchar()));
    int x=ch-'0';
    for (; isdigit(ch=getchar()); x=x*10+ch-'0');
    return x;
}

int find(int x)
{
    if (fa[x]==x)
        return x;
    return fa[x]=find(fa[x]);
}

void merge(int u,int v)
{
    if ((u=find(u))==(v=find(v)))
        return;
    if (s[u]>s[v])
        swap(u,v);
    fa[u]=v,s[v]+=s[u];
}

int main()
{
    scanf("%d%d%d",&n,&x,&y),w[0]=1<<30;
    repu(i,1,n)
    {
        c[i]=getint(),w[i]=getint(),s[fa[i]=i]=1;
        if (w[i]<w[minw[c[i]]])
            minw[c[i]]=i;
        if (w[i]<w[min1])
        {
            if (c[i]!=c[min1])
                min2=min1;
            min1=i;
        }
        else
            if (w[i]<w[min2] && c[i]!=c[min1])
                min2=i;
    }
    repu(i,1,n)
    {
        if (w[i]+w[minw[c[i]]]<=x)
            merge(i,minw[c[i]]);
        int t=c[i]==c[min1]?min2:min1;
        if (w[i]+w[t]<=y)
            merge(i,t);
    }
    int ans=1;
    repu(i,1,n)
    {
        if (find(i)==i)
            repu(j,2,s[i])
                ans=LL(ans)*j%p;
        *tp=(edge){i,fst[fa[i]]},fst[fa[i]]=tp++;
    }
    inv[0]=inv[1]=1;
    repu(i,2,n)
        inv[i]=LL(p-p/i)*inv[p%i]%p;
    repu(i,1,n)
        if (fa[i]==i)
        {
            repe(j,i)
                ans=LL(ans)*inv[++cnt[c[j->v]]]%p;
            repe(j,i)
                cnt[c[j->v]]=0;
        }
    printf("%d\n",ans);
    return 0;
}

Submission Info

Submission Time
Task D - Colorful Balls
User dwjshift
Language C++14 (GCC 5.4.1)
Score 1000
Code Size 1918 Byte
Status AC
Exec Time 45 ms
Memory 10368 KB

Compile Error

./Main.cpp: In function ‘int main()’:
./Main.cpp:44:40: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
     scanf("%d%d%d",&n,&x,&y),w[0]=1<<30;
                                        ^

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 1 ms 4224 KB
00_example_02.txt AC 1 ms 4224 KB
00_example_03.txt AC 1 ms 4224 KB
01.txt AC 1 ms 4224 KB
02.txt AC 1 ms 4224 KB
03.txt AC 3 ms 4480 KB
04.txt AC 1 ms 4224 KB
05.txt AC 2 ms 4352 KB
06.txt AC 1 ms 4352 KB
07.txt AC 20 ms 8320 KB
08.txt AC 2 ms 4352 KB
09.txt AC 11 ms 7808 KB
10.txt AC 7 ms 7296 KB
11.txt AC 1 ms 4224 KB
12.txt AC 1 ms 4224 KB
13.txt AC 3 ms 4480 KB
14.txt AC 1 ms 4224 KB
15.txt AC 9 ms 7424 KB
16.txt AC 18 ms 8320 KB
17.txt AC 4 ms 4736 KB
18.txt AC 9 ms 7424 KB
19.txt AC 16 ms 8192 KB
20.txt AC 40 ms 10368 KB
21.txt AC 36 ms 10368 KB
22.txt AC 37 ms 10368 KB
23.txt AC 39 ms 10368 KB
24.txt AC 35 ms 10368 KB
25.txt AC 41 ms 10368 KB
26.txt AC 38 ms 10368 KB
27.txt AC 40 ms 10368 KB
28.txt AC 42 ms 10368 KB
29.txt AC 43 ms 10368 KB
30.txt AC 37 ms 10368 KB
31.txt AC 41 ms 10368 KB
32.txt AC 43 ms 10368 KB
33.txt AC 45 ms 10368 KB
34.txt AC 36 ms 9472 KB
35.txt AC 36 ms 10368 KB
36.txt AC 39 ms 10240 KB
37.txt AC 40 ms 10368 KB
38.txt AC 38 ms 10368 KB
39.txt AC 39 ms 10368 KB
40.txt AC 36 ms 10368 KB
41.txt AC 38 ms 10368 KB
42.txt AC 35 ms 10368 KB
43.txt AC 37 ms 10368 KB
44.txt AC 36 ms 10368 KB
45.txt AC 33 ms 10368 KB
46.txt AC 36 ms 10368 KB
47.txt AC 32 ms 10368 KB
48.txt AC 32 ms 10368 KB
49.txt AC 34 ms 10368 KB
50.txt AC 28 ms 9600 KB
51.txt AC 28 ms 9600 KB
52.txt AC 30 ms 9600 KB
53.txt AC 28 ms 9600 KB
54.txt AC 31 ms 9600 KB