首页 > 代码库 > 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,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_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+后缀数组
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。