liberOJ #6173. Samjia 和矩阵 hash+后缀数组

#6173. Samjia 和矩阵

题目链接  ： 点这里

题目描述

给你一个只包含大写字母的矩阵，求有多少本质不同的子矩阵。

输入格式

第一行包含两个整数 nnn , mmm ,表示矩阵 nnn 行 mmm 列 。
接下来 nnn 行描述这个矩阵。

输出格式

只含一个整数，为本质不同的子矩阵个数。

样例

样例输入

3 3
ABA
BAA
AAA

样例输出

22

数据范围与提示

对于 10% 10\%10% 的数据，n,m≤10 n,m \leq 10n,m≤10；
对于 40% 40\%40% 的数据，n,m≤25 n,m \leq 25n,m≤25；
对于 70% 70\%70% 的数据，n,m≤60 n,m \leq 60n,m≤60；
对于 100% 100\%100% 的数据，n,m≤110 n,m \leq 110n,m≤110。

题解：

　　

　　

来自wannafly

#include<bits/stdc++.h>
using namespace std;
#pragma comment(linker, "/STACK:102400000,102400000")
#define ls i<<1
#define rs ls | 1
#define pii pair<int,int>
#define MP make_pair
typedef long long LL;
typedef unsigned long long ULL;
const long long INF = 1e18+1LL;
const double pi = acos(-1.0);
const int N = 2e4+, M = 1e3+,inf = 2e9;

int *ran,r[N],sa[N],height[N],wa[N],wb[N],wm[N];
bool cmp(int *r,int a,int b,int l) {
    return r[a] == r[b] && r[a+l] == r[b+l];
}
void SA(int *r,int *sa,int n,int m) {
    int *x=wa,*y=wb,*t;
    for(int i=;i<m;++i)wm[i]=;
    for(int i=;i<n;++i)wm[x[i]=r[i]]++;
    for(int i=;i<m;++i)wm[i]+=wm[i-];
    for(int i=n-;i>=;--i)sa[--wm[x[i]]]=i;
    for(int i=,j=,p=;p<n;j=j*,m=p){
        for(p=,i=n-j;i<n;++i)y[p++]=i;
        for(i=;i<n;++i)if(sa[i]>=j)y[p++]=sa[i]-j;
        for(i=;i<m;++i)wm[i]=;
        for(i=;i<n;++i)wm[x[y[i]]]++;
        for(i=;i<m;++i)wm[i]+=wm[i-];
        for(i=n-;i>=;--i)sa[--wm[x[y[i]]]]=y[i];
        for(t=x,x=y,y=t,i=p=,x[sa[]]=;i<n;++i) {
            x[sa[i]]=cmp(y,sa[i],sa[i-],j)?p-:p++;
        }
    }
    ran=x;
}
void Height(int *r,int *sa,int n) {
    for(int i=,j=,k=;i<n;height[ran[i++]]=k)
    for(k?--k:,j=sa[ran[i]-];r[i+k] == r[j+k];++k);
}

const ULL mod = 10000019ULL;
int n,m;
ULL sqr[],has[][];
char a[][];
map<ULL,int >mp;
int main() {
    sqr[] = ;
    for(int i = ; i <= ; ++i) sqr[i] = sqr[i-] * mod;
    scanf("%d%d",&n,&m);
    for(int i = ; i <= n; ++i) {
        scanf("%s",a[i]+);
        has[i][] = ;
        for(int j = ; j <= m; ++j) {
            has[i][j] = has[i][j-] * mod + a[i][j] - 'A' + ;
        }
    }
    LL ans = ;
    for(int y = ; y <= m; ++y) {
        int cnt = ,san = ;
        mp.clear();
        for(int j = ; j + y -  <= m; ++j) {
             for(int i = ; i <= n; ++i){
                int l = j, rr = j + y - ;
                ULL now = has[i][rr] - has[i][l-]*sqr[y];
                if(mp[now] == ) mp[now] = san++;
                r[cnt++] = mp[now];
            }
            r[cnt++] = san++;
        }
        r[cnt] = ;

        SA(r,sa,cnt+,san);
        Height(r,sa,cnt);
        //for(int i = 0; i <= cnt; ++i) cout<<sa[i]<<" "<<ran[i]<<" "<<height[i]<<endl;
       // return 0;
        ans += n*(n+)/*(m-y+);

        for(int i = ; i <= cnt; ++i) {
            ans -= height[i];
        }
    }
    printf("%lld\n",ans);
    return ;
}