#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+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; }
时间: 2024-11-05 20:31:21