题目链接:https://loj.ac/problem/3095。
假设我们当前要比较第$x$个和第$y$个的大小,假设$x<y$,那么其实这两个串是非常相似的,更具体的说,从第$x$个字符开始,第一个串的$i+1$对应第二个串的$i$,所以找到从$x$开始第一对相邻的不一样的字符比较一下就行了。
所以直接写个比较函数跑一遍$\rm sort$就行了。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58
| #include<bits/stdc++.h> using namespace std;
void read(int &x) { x=0;int f=1;char ch=getchar(); for(;!isdigit(ch);ch=getchar()) if(ch=='-') f=-f; for(;isdigit(ch);ch=getchar()) x=x*10+ch-'0';x*=f; }
void print(int x) { if(x<0) putchar('-'),x=-x; if(!x) return ;print(x/10),putchar(x%10+48); } void write(int x) {if(!x) putchar('0');else print(x);putchar('\n');}
#define lf double #define ll long long
#define pii pair<int,int > #define vec vector<int >
#define pb push_back #define mp make_pair #define fr first #define sc second
#define FOR(i,l,r) for(int i=l,i##_r=r;i<=i##_r;i++)
const int maxn = 1e6+10; const int inf = 1e9; const lf eps = 1e-8; const int mod = 1e9+7;
char s[maxn]; int n,p[maxn],nxt[maxn];
void gen() { nxt[n]=n; for(int i=n-1;i;i--) if(s[i]!=s[i+1]) nxt[i]=i; else nxt[i]=nxt[i+1]; }
int cmp(int x,int y) { int t=nxt[min(x,y)]; if(t<max(x,y)) { if(x<y) return s[t+1]<s[t]; else return s[t]<s[t+1]; } else return x<y; }
int main() { read(n);scanf("%s",s+1);gen(); for(int i=1;i<=n;i++) p[i]=i; sort(p+1,p+n+1,cmp); for(int i=1;i<=n;i++) printf("%d ",p[i]);puts(""); return 0; }
|