Submission #53375058

Source Code Expand

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
#include<bits/stdc++.h>
using namespace std;
#define int long long
const int mod=998244353;
#define ll long long
#include<array>
struct Shash{
    const ll base[2]={29,31};
    const ll hashmod[2]={(ll)1e9,998244353};
    #define fi first
    #define se second
    array<vector<ll>,2>hsh,pwMod;
    void init(string s){
        int n=s.size();s=' '+s;
        hsh[0].resize(n+1),hsh[1].resize(n+1);
        pwMod[0].resize(n+1),pwMod[1].resize(n+1);
        for(int i=0;i<2;i++){
            pwMod[i][0]=1;
            for(int j=1;j<=n;j++){
                pwMod[i][j]=pwMod[i][j-1]*base[i]%hashmod[i];
                hsh[i][j]=(hsh[i][j-1]*base[i]+s[j])%hashmod[i];
 
 
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX

#include<bits/stdc++.h>
using namespace std;
#define int long long
const int mod=998244353;
#define ll long long
#include<array>
struct Shash{
    const ll base[2]={29,31};
    const ll hashmod[2]={(ll)1e9,998244353};
    #define fi first
    #define se second
    array<vector<ll>,2>hsh,pwMod;
    void init(string s){
        int n=s.size();s=' '+s;
        hsh[0].resize(n+1),hsh[1].resize(n+1);
        pwMod[0].resize(n+1),pwMod[1].resize(n+1);
        for(int i=0;i<2;i++){
            pwMod[i][0]=1;
            for(int j=1;j<=n;j++){
                pwMod[i][j]=pwMod[i][j-1]*base[i]%hashmod[i];
                hsh[i][j]=(hsh[i][j-1]*base[i]+s[j])%hashmod[i];
            }
        }
    }
    pair<ll,ll>get(int l,int r){
        pair<ll,ll>ans;
        ans.fi=(hsh[0][r]-hsh[0][l-1]*pwMod[0][r-l+1])%hashmod[0];
        ans.se=(hsh[1][r]-hsh[1][l-1]*pwMod[1][r-l+1])%hashmod[1];
        ans.fi=(ans.fi+hashmod[0])%hashmod[0];
        ans.se=(ans.se+hashmod[1])%hashmod[1];
        return ans;
    }
    bool same(int la,int ra,int lb,int rb){
        return get(la,ra)==get(lb,rb);
    }
};
const int N=3e5+10;
map<pair<int,int>,set<int>>mp[N];
void solve(){
    int n;cin>>n;
    vector<string>a(n);
    for(auto&i:a)cin>>i;
    Shash h;
    for(int j=0;j<n;j++){
        int len=a[j].size();
        h.init(a[j]);
        for(int i=0;i<len;i++){
            mp[i+1][h.get(1,i+1)].insert(j);
        }
    }
    int ans=0;
    for(int j=0;j<n-1;j++){
        int pre=ans;
        int len=a[j].size();
        h.init(a[j]);
        vector<int>st(len+2);
        for(int i=0;i<len;i++){
            mp[i+1][h.get(1,i+1)].erase(j);
            st[i+1]=mp[i+1][h.get(1,i+1)].size();
        }
        for(int i=0;i<=len;i++){
            st[i]-=st[i+1];
        }
        for(int i=1;i<=len;i++){
            ans+=i*st[i];
            // cout<<i<<" "<<st[i]<<endl;
        }
        // cout<<ans-pre<<endl;
    }
    cout<<ans;
}
signed main(){
    ios::sync_with_stdio(0);cin.tie(0);
    solve();
}

Submission Info

Compile Error

Main.cpp: In function ‘void solve()’:
Main.cpp:53:13: warning: unused variable ‘pre’ [-Wunused-variable]
   53 |         int pre=ans;
      |             ^~~

Judge Result