Cod sursa(job #2563461)

Utilizator anamariatoaderAna Toader anamariatoader Data 1 martie 2020 11:45:25
Problema Substr Scor 10
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.38 kb
#include <fstream>
#include <unordered_map>
#define mod 666013

using namespace std;
ifstream fin("substr.in");
ofstream fout("substr.out");

int n,K,st,i,dr,mij,p1[16390],p2[16390],h1,h2,sol,mx,k;
char s[16390];
unordered_map <int,int> h;

int caracter(char c){
    if(c>='a' && c<='z')
        return c-'a'+1;
    if(c>='A' && c<='Z')
        return c-'A'+27;
    return c-'0'+53;
}

int main(){
    fin>>n>>K>>(s+1);
    p1[0]=p2[0]=1;
    for(i=1;i<=n;i++){
        p1[i]=(1ll*p1[i-1]*67)%mod;
        p2[i]=(1ll*p2[i-1]*73)%mod;
    }
    st=1,dr=n; sol=0;
    while(st<=dr){
        mij=(st+dr)/2;
        h1=h2=0;
        for(i=1;i<=mij;i++){
            int x=caracter(s[i]);
            h1=(1ll*h1*67+x)%mod;
            h2=(1ll*h2*73+x)%mod;
        }
        k=(1ll*h1*31+h2)%mod;
        h[k]++;
        mx=1;
        for(i=mij+1;i<=n;i++){
            int x=caracter(s[i]);
            int c=caracter(s[i-mij]);
            h1=((1ll*h1-1ll*c*p1[mij-1])*67+x)%mod;
            if(h1<0)
                h1+=mod;
            h2=((1ll*h2-1ll*c*p2[mij-1])*73+x)%mod;
            if(h2<0)
                h2+=mod;
            k=(1ll*h1*31+h2)%mod;
            h[k]++;
            mx=(mx,h[k]);
        }
        if(mx>=K){
            sol=mij;
            st=mij+1;
        }
        else
            dr=mij-1;
    }
    fout<<sol;
    return 0;
}