SA-IS后缀数组
Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 512000/512000 K (Java/Others)
Total Submission(s): 2114 Accepted Submission(s): 606
Problem Description
小Q最近阅读了SA-IS算法在线性时间内构造后缀数组的相关论文,面对任何字符串题,都可以想出线性时间的算法。
小T在经历过二分图匹配事件后,再也不相信小Q所说的话。面对小Q,小T又给出了一道字符串题:
给定一个长度为n的小写字符串S[1..n],设sufi表示以i 为开始的后缀,即S[i..n]。
记|X|为字符串X的长度,对于两个字符串X和Y,定义X的字典序比Y小,当且仅当存在非负整数k(k≤min(|X|,|Y|))使得X的前k个字符与Y的前 k个字符对应相同,并且要么满足|X|=k且|Y|>k,要么满足k<min(|X|,|Y|)且X的第k+1个字符比Y的第k+1个字符小。例如aa的字典序比aaa小,ab的字典序比ba小。
请对每个i(1≤i<n),判断sufi和sufi+1的字典序大小关系。
只会吹牛的小Q又不会做了,所以他再一次向你紧急求助。请写一个程序,判断相邻两个后缀的大小关系。
Input
第一行包含一个正整数T(1≤T≤10),表示测试数据的组数。
每组数据第一行包含一个正整数n(2≤n≤1000000),表示字符串S的长度。
第二行包含一个长度为n的小写字符串S。
Output
对于每组数据,输出一行n−1个字符,第i个字符表示sufi和sufi+1的大小关系,若sufi<sufi+1,输出<,否则输出>,显然不存在相等关系。
Sample Input
1 17 quailtyacmbestacm
Sample Output
<><<<<><<><<<><<
Source
解题思路
从最后一组后缀开始往前处理,用栈记录答案,用flg记录上一个不同的地方的状态(是大于还是小于),比较新加入的字符大小,如果不同,则将答案存入栈,并改变flg,如果一样,则将flg对应的答案存入栈。
代码如下
#include <iostream>
#include <stack>
#include <cstdio>
#define maxn 1000005
using namespace std;
char str[maxn];
int main()
{
int T;
cin >> T;
while(T --){
int n;
scanf("%d", &n);
scanf("%s", str);
stack<char> sta;
int flg = 0;
for(int i = n - 2; i >= 0; i --){
if(str[i] < str[i + 1]){
sta.push('<');
flg = -1;
}
else if(str[i] > str[i + 1]){
sta.push('>');
flg = 1;
}
else {
if(flg == 1 || flg == 0) //大于或者是第一个则长度大的比较大
sta.push('>');
else
sta.push('<');
}
}
while(!sta.empty()){
putchar(sta.top());
sta.pop();
}
printf("\n");
}
return 0;
}