oj.uz

Submission #48941

# Submission time Handle Problem Language Result Execution time Memory
48941 2018-05-20T06:21:50 Z leehosu01 백신 (KOI13_vaccine) C++17
24 / 24
 31 ms 2800 KB 
#include<bits/stdc++.h>
using namespace std;
bitset<2500>ext;
typedef long long ll;
ll p1=10061,p2=100000007,n1=396,n2=117;
ll PW(ll X,ll Y,ll P)
{
    ll I=1ll<<62,cs=1;
    for(;I;I>>=1)
    {
        cs=cs*cs%P;
        if(I&Y)cs=cs*X%P;
    }
    return cs;
}
int N,K;
ll PS1,PS2;
vector<int>V;
struct C{
ll H1,H2;
int from;
C RG(C&X)
{
    return {((H1-X.H1*PS1)%p1+p1)%p1,((H2-X.H2*PS2)%p2+p2)%p2};
}
bool operator<(const C&X)const{return H1==X.H1?H2<X.H2:H1<X.H1;}
bool operator==(const C&X)const{return H1==X.H1&&H2==X.H2;}
};
vector<C>VC,TV,TTV;
void BH(int n)
{
    int i;
    TV.resize(n+1);
    TV[0]={0,0,0};
    for(i=0;i<n;i++)
        TV[i+1]={(TV[i].H1*n1+V[i])%p1,(TV[i].H2*n2+V[i])%p2};
}
int in()
{
    int M;
    cin>>M;
    V.resize(M);
    for(int&I:V)cin>>I;
    return M;
}
void getH()
{
    TTV.clear();
    for(int i=K;i<TV.size();i++)
    {
        TTV.push_back(TV[i].RG(TV[i-K]));
    }
}
void first()
{
    VC.clear();
    int L=in();
    BH(L);
    getH();
    VC.insert(VC.end(),TTV.begin(),TTV.end());
    sort(VC.begin(),VC.end());
    VC.resize(distance(VC.begin(),unique(VC.begin(),VC.end())) );
    for(auto&I:VC)I.from=(&I-&VC[0]);
}
void second()
{
    ext.reset();
    int L=in();
    BH(L);
    getH();
    for(auto&I:TTV)
    {
        if(lower_bound(VC.begin(),VC.end(),I)!=VC.end()&&I==*lower_bound(VC.begin(),VC.end(),I))
            ext.set((*lower_bound(VC.begin(),VC.end(),I)).from);
    }
    reverse(V.begin(), V.end());
    BH(L);
    getH();
    for(auto&I:TTV)
    {
        if(lower_bound(VC.begin(),VC.end(),I)!=VC.end()&&I==*lower_bound(VC.begin(),VC.end(),I))
            ext.set((*lower_bound(VC.begin(),VC.end(),I)).from);
    }
    int w=0;
    for(int i=0;i<VC.size();i++)
    {
        if(ext[VC[i].from])
            VC[w++]=VC[i];
    }
    VC.resize(w);
}
int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    cin>>N>>K;
    PS1=PW(n1,K,p1),PS2=PW(n2,K,p2);
    first();
    int i;
    for(i=1;i<N&&!VC.empty();i++)
        second();
    printf("%s",VC.empty()?  "NO":"YES");
}

Compilation message

vaccine.cpp: In function 'void getH()':
vaccine.cpp:49:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i=K;i<TV.size();i++)
                 ~^~~~~~~~~~
vaccine.cpp: In function 'void second()':
vaccine.cpp:85:18: warning: comparison between signed and unsigned integer expressions [-Wsign-compare]
     for(int i=0;i<VC.size();i++)
                 ~^~~~~~~~~~

Subtask #1
2.64 / 2.64

# Verdict Execution time Memory Grader output
1 Correct 2 ms 376 KB Output is correct
2 Correct 2 ms 488 KB Output is correct
3 Correct 2 ms 524 KB Output is correct
4 Correct 3 ms 524 KB Output is correct
5 Correct 2 ms 540 KB Output is correct
6 Correct 3 ms 540 KB Output is correct
7 Correct 3 ms 576 KB Output is correct
8 Correct 2 ms 688 KB Output is correct
9 Correct 3 ms 840 KB Output is correct
10 Correct 3 ms 840 KB Output is correct

Subtask #2
3.12 / 3.12

# Verdict Execution time Memory Grader output
1 Correct 2 ms 840 KB Output is correct
2 Correct 3 ms 840 KB Output is correct
3 Correct 2 ms 840 KB Output is correct
4 Correct 2 ms 840 KB Output is correct
5 Correct 2 ms 840 KB Output is correct
6 Correct 3 ms 840 KB Output is correct
7 Correct 8 ms 840 KB Output is correct
8 Correct 3 ms 840 KB Output is correct
9 Correct 3 ms 840 KB Output is correct
10 Correct 3 ms 840 KB Output is correct

Subtask #3
3.6 / 3.6

# Verdict Execution time Memory Grader output
1 Correct 2 ms 840 KB Output is correct
2 Correct 3 ms 840 KB Output is correct
3 Correct 4 ms 840 KB Output is correct
4 Correct 3 ms 840 KB Output is correct
5 Correct 3 ms 840 KB Output is correct
6 Correct 3 ms 840 KB Output is correct
7 Correct 3 ms 840 KB Output is correct
8 Correct 3 ms 840 KB Output is correct
9 Correct 4 ms 840 KB Output is correct
10 Correct 2 ms 880 KB Output is correct

Subtask #4
4.32 / 4.32

# Verdict Execution time Memory Grader output
1 Correct 3 ms 912 KB Output is correct
2 Correct 3 ms 912 KB Output is correct
3 Correct 3 ms 912 KB Output is correct
4 Correct 2 ms 912 KB Output is correct
5 Correct 3 ms 912 KB Output is correct
6 Correct 2 ms 924 KB Output is correct
7 Correct 3 ms 940 KB Output is correct
8 Correct 4 ms 940 KB Output is correct
9 Correct 3 ms 1100 KB Output is correct
10 Correct 3 ms 1100 KB Output is correct

Subtask #5
4.8 / 4.8

# Verdict Execution time Memory Grader output
1 Correct 6 ms 1100 KB Output is correct
2 Correct 9 ms 1100 KB Output is correct
3 Correct 11 ms 1100 KB Output is correct
4 Correct 17 ms 1100 KB Output is correct
5 Correct 2 ms 1100 KB Output is correct
6 Correct 4 ms 1100 KB Output is correct
7 Correct 3 ms 1100 KB Output is correct
8 Correct 15 ms 1328 KB Output is correct
9 Correct 28 ms 1888 KB Output is correct
10 Correct 3 ms 1888 KB Output is correct

Subtask #6
5.52 / 5.52

# Verdict Execution time Memory Grader output
1 Correct 15 ms 1888 KB Output is correct
2 Correct 29 ms 1888 KB Output is correct
3 Correct 31 ms 1888 KB Output is correct
4 Correct 2 ms 1888 KB Output is correct
5 Correct 3 ms 1888 KB Output is correct
6 Correct 3 ms 1888 KB Output is correct
7 Correct 18 ms 2448 KB Output is correct
8 Correct 20 ms 2800 KB Output is correct
9 Correct 4 ms 2800 KB Output is correct
10 Correct 4 ms 2800 KB Output is correct