牛客多校第一场 I Substring (不同子串的个数)

题目链接：嘤嘤嘤

题目大意，求一个字符串内不能通过完全映射对应到的子串的数量

因为字符串中只存在a,b,c三种字符，所以我们可以通过枚举(3!)个全排列，然后通过后缀数组常用搞法

将字符串通过不同的映射得到的串通过特殊字符连接起来

因为按照题意，如果两个字符串同构，那么他们两个一定能够通过一种映射转化为两个串相等

然后我们就可以把问题转化为一个字符串有多少个不同的子串的问题

对于原字符串每一个后缀

因为我们通过后缀数组排过序，这个时候所有通过映射的得到的字典序最小的后缀一定在后缀数组的前面

这个时候我们首先找出每个后缀通过映射后得到的字典序最小串

然后这个时候每两个相邻串的lcp就是重复的串，减去即可(可以在原串通过映射变得相同)

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstring>
using namespace std;
//后缀数组裸题
const int maxn=5e4*6*3;
typedef long long llt;
#define F(x) ((x)/3+((x)%3==1?0:tb))
#define G(x) ((x)<tb?(x)*3+1:((x)-tb)*3+2)
int wa[maxn],wb[maxn],wv[maxn],wss[maxn];
int sa[maxn],ranks[maxn],height[maxn];
int c0(int *r,int a,int b)
{
    return r[a]==r[b]&&r[a+1]==r[b+1]&&r[a+2]==r[b+2];
}
int c12(int k,int *r,int a,int b)
{
    if(k==2) return r[a]<r[b]||r[a]==r[b]&&c12(1,r,a+1,b+1);
    else return r[a]<r[b]||r[a]==r[b]&&wv[a+1]<wv[b+1];
}
void sort(int *r,int *a,int *b,int n,int m)
{
    int i;
    for(i=0; i<n; i++) wv[i]=r[a[i]];
    for(i=0; i<m; i++) wss[i]=0;
    for(i=0; i<n; i++) wss[wv[i]]++;
    for(i=1; i<m; i++) wss[i]+=wss[i-1];
    for(i=n-1; i>=0; i--) b[--wss[wv[i]]]=a[i];
    return;
}
void dc3(int *r,int *sa,int n,int m)
{
    int i,j,*rn=r+n,*san=sa+n,ta=0,tb=(n+1)/3,tbc=0,p;
    r[n]=r[n+1]=0;
    for(i=0; i<n; i++) if(i%3!=0) wa[tbc++]=i;
    sort(r+2,wa,wb,tbc,m);
    sort(r+1,wb,wa,tbc,m);
    sort(r,wa,wb,tbc,m);
    for(p=1,rn[F(wb[0])]=0,i=1; i<tbc; i++)
        rn[F(wb[i])]=c0(r,wb[i-1],wb[i])?p-1:p++;
    if(p<tbc) dc3(rn,san,tbc,p);
    else for(i=0; i<tbc; i++) san[rn[i]]=i;
    for(i=0; i<tbc; i++) if(san[i]<tb) wb[ta++]=san[i]*3;
    if(n%3==1) wb[ta++]=n-1;
    sort(r,wb,wa,ta,m);
    for(i=0; i<tbc; i++) wv[wb[i]=G(san[i])]=i;
    for(i=0,j=0,p=0; i<ta && j<tbc; p++)
        sa[p]=c12(wb[j]%3,r,wa[i],wb[j])?wa[i++]:wb[j++];
    for(; i<ta; p++) sa[p]=wa[i++];
    for(; j<tbc; p++) sa[p]=wb[j++];
    return;
}
void calheight(int *tmp,int n)
{
    int i,j;
    for(i=0;i<n;i++) ranks[sa[i]]=i;
    int k=0;
    for(i=0;i<n-1;i++)
    {
        if(k) k--;
        j=sa[ranks[i]-1];
        while(tmp[i+k]==tmp[j+k]) k++;
        height[ranks[i]]=k;
    }
}
char str[maxn];
//暴力枚举六种映射
const int inf=0x3f3f3f3f;
const int mp[][3]={{0,1,2},{0,2,1},{1,0,2},{1,2,0},{2,0,1},{2,1,0}};
int arr[maxn];
void print(int *r,int sz){
    for(int i=0;i<sz;i++) printf("%d ",r[i]);
    putchar('\n');
}
bool vis[maxn];
int main()
{
    int n;
    while(scanf("%d%s",&n,str)!=EOF)
    {
        int tot=0;
        for(int i=0;i<6;i++){
            for(int j=0;j<n;j++){
                arr[tot++]=mp[i][str[j]-'a']+1;
            }
            //用特殊字符隔开
            arr[tot++]='&';
        }
        //print(arr,tot);
        //然后用后缀数组排个序
        arr[tot-1]=0;
        dc3(arr,sa,tot,128);
        calheight(arr,tot);
        memset(vis,0,sizeof(vis));
        llt ret=1LL*(n+1)*n/2;
        int last=inf;
        for(int i=1;i<=tot;i++){
            int pos=sa[i]%(n+1);
            //printf("pos->%d %d\n",pos,height[i]);
            last=min(last,height[i]);
            if(vis[pos]||pos==n) continue;
            ret-=last;
            last=height[i+1];
            vis[pos]=true;
        }
        printf("%lld\n",ret);
    }
    return 0;
}