Submission #575673

#	Time	Username	Problem	Language	Result	Execution time	Memory
575673		andrei_boaca	Lottery (CEOI18_lot)	C++14	65 / 100	199 ms	24936 KiB

lot

#include <bits/stdc++.h>

using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const ll baza=1e9+7;
const ll mod=1e18;
int cnt[2005][2005],nr[2005][2005];
int n,v[10005],l,q;
map<ll,int> f;
ll myhash[10005];
int main()
{
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    cin>>n>>l;
    for(int i=1;i<=n;i++)
        cin>>v[i];
    if(n<=2000)
    {
        for(int i=1;i<=n;i++)
            for(int j=i+1;j<=n;j++)
                if(v[i]==v[j])
                {
                    int dist=j-i;
                    int lft=max(1,i-l+1);
                    int rgt=j;
                    if(j+l-1>n)
                        rgt=n-l+1;
                    rgt-=dist;
                    if(lft<=rgt)
                    {
                        cnt[dist][lft]++;
                        cnt[dist][rgt+1]--;
                    }
                }
        for(int dist=1;dist<=n;dist++)
            for(int i=1;i+l-1<=n;i++)
            {
                cnt[dist][i]=cnt[dist][i]+cnt[dist][i-1];
                int j=i+dist;
                if(j+l-1>n)
                    continue;
                int val=cnt[dist][i];
                nr[i][val]++;
                nr[j][val]++;
            }
        for(int i=1;i+l-1<=n;i++)
            for(int val=1;val<=l;val++)
                nr[i][val]+=nr[i][val-1];
        cin>>q;
        while(q--)
        {
            int need=0;
            cin>>need;
            need=l-need;
            for(int i=1;i+l-1<=n;i++)
            {
                int val=nr[i][l];
                if(need>0)
                    val-=nr[i][need-1];
                cout<<val<<' ';
            }
            cout<<'\n';
        }
        return 0;
    }
    for(int i=1;i+l-1<=n;i++)
    {
        ll h=0;
        for(int j=i;j<=i+l-1;j++)
            h=((h*baza)%mod+v[j])%mod;
        f[h]++;
        myhash[i]=h;
    }
    for(int i=1;i+l-1<=n;i++)
    {
        int val=f[myhash[i]]-1;
        cout<<val<<' ';
    }
    cout<<'\n';
    return 0;
}

• Subtask 1: 25 / 25
• Subtask 2: 20 / 20
• Subtask 3: 20 / 20
• Subtask 4: 0 / 15
• Subtask 5: 0 / 20