Cod sursa(job #3319088)

Utilizator gabi072Sanda Gabriel gabi072 Data 30 octombrie 2025 14:52:39
Problema Abc2 Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.89 kb
#include<bits/stdc++.h>
using namespace std;
ifstream fin("abc2.in");
ofstream fout("abc2.out");
unordered_map<uint64_t,int> mp;
const int base = 1315423911;
uint64_t h[10000000],p;
string sir,c;
inline uint64_t hashd(const string &s) {
    uint64_t h=0;
    for (char i:s) {
        h=h*base+i;
    }
    return h;
}
int main() {
    fin>>sir;
    fin>>c;
    int ws=c.size();
    h[0]=sir[0];
    p=1;
    for (int i=0;i<ws;++i)
        p=p*base;
    for (int i=1;i<(int)sir.size();++i) {
        h[i]=base*h[i-1]+sir[i];
    }
    for (int i=0;i<=(int)sir.size()-ws;++i) {
        uint64_t key;
        if (i==0)
            key=h[ws-1];
        else
            key=h[i+ws-1]-h[i-1]*p;
        mp[key]++;
    }
    long long ap=0;
    uint64_t key=hashd(c);
    ap+=mp[key];
    mp[key]=0;
    while (fin>>c) {
        uint64_t key=hashd(c);
        ap+=mp[key];
        mp[key]=0;
    }
    fout<<ap;
}