标签:set count ems amp 常用 rank 相同 void 开头
又是一个学了n遍还没学会的算法……
后缀数组是一种常用的处理字符串问题的数据结构,主要由\(sa\)和\(rank\)两个数组组成。以下给出一些定义:
\(str\)表示处理的字符串,长度为\(len\)。(下标从\(0\)开始)
\([i,j)\)表示\(str\)从\(i\)到\(j - 1\)的字串。
后缀\(i\)表示子串\([i,len)\),以字典序排序。
\(sa[i]\)表示排名为\(i\)的后缀的起始位置(即后缀\(sa[i]\)是第\(i\)名)
\(rank[i]\)表示后缀\(i\)的排名(从\(0\)开始)。显然\(rank[sa[i]]=i\)。
先简单介绍一下后缀数组的前置技能:基数排序。
以对整数数组\(arr\)排序为例。从低到高遍历每一个十进制位,对于每个位:
\(1.\)\(arr\)数组已经按照前\(i-1\)位排好序,(\(i=0\)时忽略这句),现在我们将把它变为按前\(i\)位排好序。脑补以下整数的比较方式,现在应该把第\(i\)位作为第一关键字,前\(i-1\)位作为第二关键字。
\(2.\)统计第\(i\)位为数字\(a\)的数的数量,存入\(count[a]\)。
\(3.\)对\(count\)数组求前缀和,算出最后一个第\(i\)位为\(a\)的数在按照前\(i\)位排序后数组中的位置的下一个。这句表达比较鬼畜,看下面的例子。
比如,\(i\)位为\(0\)的有\(2\)个,为\(1\)的有\(1\)个,为\(2\)的有\(3\)个,第\(3\)步以后\(count\)位\(\{2,3,6\}\),那么排序后\(arr[0]\)和\(arr[1]\)的第\(i\)位为\(0\),\(arr[2]\)的第\(i\)位为\(1\),\(arr[3]\)到\(arr[5]\)的第\(i\)位为\(2\)。
\(4.\)逆序遍历\(arr\),按照上一步中算出的第\(i\)位为\(a\)的数排序后的位置逆序填充临时数组。两个均逆序保证了对于第\(i\)位相同的数按照最初在\(arr\)中的位置排序。
\(5.\)最后,把临时数组复制给\(arr\),此时\(arr\)按照前\(i\)位有序。
int count[10];
for(int i = 1; i <= 10; i++, ra *= 10)
{
memset(count, 0, sizeof(count));
for (int j = 1; j <= n; j++)
++count[arr[j] / ra % 10];//step 2
for (int j = 1; j < 10; j++)
count[j] += count[j - 1];//step 3
for(int j = n - 1; j >= 0; j--)
buc[--count[arr[j] / ra % 10]] = arr[j];
memcpy(arr, buc, sizeof(int[n]));
}
考虑我们现在有了对所有形如\([i,min(i+tmp,len))\)的子串排序的数组\(sa\)和\(rank\)(对于相同的子串,它们的\(rank\)值相同,在\(sa\)中顺序任意),我们现在要构造对所有形如\([i,min(i+2tmp,len))\)的子串排序。最坏情况下,当\(2tmp\geq len\)时就得到了答案。
可以发现此时很类似于基数排序时排到某一位时的情况。此时,第一关键字是\([i,i+tmp)\),第二关键字是\([i+tmp, i+2tmp)\)。并且,现在已经按照第二关键字排好序了。
于是我们先看看此处的基数排序。其中\(kind\)是\(rank\)中不同值的种数(由于\(rank\)从\(0\)开始,也可以看成\(rank\)中最大值加\(1\)),\(tp[i]\)表示哪个串的第二关键字在所有第二关键字中的排名是\(i\)。
void radix_sort()
{
static int count[N];
memset(count, 0, sizeof(int[kind]);
for (int i = 0; i < len; i++)
count[rank[tp[i]]]++;
for (int i = 1; i < kind; i++)
count[i] += count[i - 1];
for (int i = len - 1; i >= 0; i--)
sa[--count[rank[tp[i]]]] = tp[i];
}
然后我们来构造\(tp\)数组。首先,对于起点在\([len-tmp,len)\)中的串,它们的第二关键字都是空串,排名是最低的。所以它们应当在\(tp\)的开头:
for (int i = len - tmp; i < len; i++)
tp[cnt++] = i;
然后,按照\(sa\)加入剩下的串。注意只有起点在\(tmp\)及以后的串才能作为第二关键字。
for(int i=0;i<len;i++)
if(sa[i]>=tmp)
tp[cnt++]=sa[i]-tmp;
至此,\(tp\)数组构造完毕,可以进行基数排序。排序后,我们要按照新的\(sa\)和旧的\(rank\)构造新的\(rank\)。首先,把旧的\(rank\)进行拷贝。为了优化常数可以这样写:
swap(rank,tp)
记住,此后\(tp\)就只是旧的\(rank\)的一份拷贝了,没有更多实际意义。更新\(rank\)的过程比较显然。
rank[sa[0]] = 0;
kind = 1;
for (int i = 1; i < len; i++)
{
if (tp[sa[i]] == tp[sa[i - 1]] &&
(sa[i] + tmp < len && sa[i - 1] + tmp < len) &&
(tp[sa[i] + tmp] == tp[sa[i - 1] + tmp]))
rank[sa[i]] = rank[sa[i - 1]];
else
rank[sa[i]] = kind++;
}
最后,如果\(kind=len\),即\(rank\)已经两两不同,则说明已经得出了答案。
我不会,你开心不qwq
int sa[N], rank[N], tp[N], kind, len;
void radix_sort()
{
static int count[N];
memset(count, 0, sizeof(int) * kind);
for (int i = 0; i < len; i++)
count[rank[tp[i]]]++;
for (int i = 1; i < kind; i++)
count[i] += count[i - 1];
for (int i = len - 1; i >= 0; i--)
sa[--count[rank[tp[i]]]] = tp[i];
}
void build(const string &s)
{
len = s.size();
for (int i = 0; i < len; i++)
rank[i] = s[i], tp[i] = i;
kind = CH;
radix_sort();
for (int tmp = 1; tmp < len; tmp *= 2)
{
int cnt = 0;
for (int i = len - tmp; i < len; i++)
tp[cnt++] = i;
for (int i = 0; i < len; i++)
if (sa[i] >= tmp)
tp[cnt++] = sa[i] - tmp;
radix_sort();
swap(rank, tp);
rank[sa[0]] = 0;
kind = 1;
for (int i = 1; i < len; i++)
{
if (tp[sa[i]] == tp[sa[i - 1]] &&
(sa[i] + tmp < len && sa[i - 1] + tmp < len) &&
(tp[sa[i] + tmp] == tp[sa[i - 1] + tmp]))
rank[sa[i]] = rank[sa[i - 1]];
else
rank[sa[i]] = kind++;
}
if (kind == len)
break;
}
}
标签:set count ems amp 常用 rank 相同 void 开头
原文地址:https://www.cnblogs.com/zyt1253679098/p/10115485.html