首页 > 代码库 > liberOJ #6173. Samjia 和矩阵 hash+后缀数组

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

#6173. Samjia 和矩阵

题目链接  : 点这里

题目描述

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

输入格式

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

输出格式

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

样例

样例输入

3 3ABABAAAAA

样例输出

22

数据范围与提示

对于 10% 10\%10% 的数据,n,m≤10 n,m \leq 10n,m10;
对于 40% 40\%40% 的数据,n,m≤25 n,m \leq 25n,m25;
对于 70% 70\%70% 的数据,n,m≤60 n,m \leq 60n,m60;
对于 100% 100\%100% 的数据,n,m≤110 n,m \leq 110n,m110。

 

题解:

  技术分享

  技术分享

来自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_pairtypedef long long LL;typedef unsigned long long ULL;const long long INF = 1e18+1LL;const double pi = acos(-1.0);const int N = 2e4+10, M = 1e3+20,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=0;i<m;++i)wm[i]=0;    for(int i=0;i<n;++i)wm[x[i]=r[i]]++;    for(int i=1;i<m;++i)wm[i]+=wm[i-1];    for(int i=n-1;i>=0;--i)sa[--wm[x[i]]]=i;    for(int i=0,j=1,p=0;p<n;j=j*2,m=p){        for(p=0,i=n-j;i<n;++i)y[p++]=i;        for(i=0;i<n;++i)if(sa[i]>=j)y[p++]=sa[i]-j;        for(i=0;i<m;++i)wm[i]=0;        for(i=0;i<n;++i)wm[x[y[i]]]++;        for(i=1;i<m;++i)wm[i]+=wm[i-1];        for(i=n-1;i>=0;--i)sa[--wm[x[y[i]]]]=y[i];        for(t=x,x=y,y=t,i=p=1,x[sa[0]]=0;i<n;++i) {            x[sa[i]]=cmp(y,sa[i],sa[i-1],j)?p-1:p++;        }    }    ran=x;}void Height(int *r,int *sa,int n) {    for(int i=0,j=0,k=0;i<n;height[ran[i++]]=k)    for(k?--k:0,j=sa[ran[i]-1];r[i+k] == r[j+k];++k);}const ULL mod = 10000019ULL;int n,m;ULL sqr[300],has[120][120];char a[210][120];map<ULL,int >mp;int main() {    sqr[0] = 1;    for(int i = 1; i <= 200; ++i) sqr[i] = sqr[i-1] * mod;    scanf("%d%d",&n,&m);    for(int i = 1; i <= n; ++i) {        scanf("%s",a[i]+1);        has[i][0] = 0;        for(int j = 1; j <= m; ++j) {            has[i][j] = has[i][j-1] * mod + a[i][j] - A + 1;        }    }    LL ans = 0;    for(int y = 1; y <= m; ++y) {        int cnt = 0,san = 1;        mp.clear();        for(int j = 1; j + y - 1 <= m; ++j) {             for(int i = 1; i <= n; ++i){                int l = j, rr = j + y - 1;                ULL now = has[i][rr] - has[i][l-1]*sqr[y];                if(mp[now] == 0) mp[now] = san++;                r[cnt++] = mp[now];            }            r[cnt++] = san++;        }        r[cnt] = 0;        SA(r,sa,cnt+1,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+1)/2*(m-y+1);        for(int i = 1; i <= cnt; ++i) {            ans -= height[i];        }    }    printf("%lld\n",ans);    return 0;}

 

 

 

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