luogu P5794 [THUSC2015]解密运算
题面传送门
首先考虑没有重复字符的20pts怎么做,因为每个字符串对应的开头是确定的,所以可以得到最后一个字符是什么。
然后考虑把最后一个字符转到最前面,因为没有重复字符所以在排好序的数组里面的位置是确定的,这样从后往前推出所有字符。
但是如果有重复字符就不能这么做,考虑一个更普适的做法:因为每个字符串的第一个字符都是确定的,我们将最后面的一个字符转到第一个,然后进行排序,排完序之后再把对应的最后一个字符转到前面来继续排序,进行\(O(n)\)轮就可以处理出整个序列。时间复杂度\(O(n^2\log n)\),可以拿到60pts的好成绩。
考虑优化,容易发现我们只对一开始以.
结尾的序列排在第几感兴趣,而计算一个确定顺序的字符串前面加上一个字符是平凡的:前面加上的字符小于当前串的一定字典序小于当前串,等于当前串的只要找到在当前顺序中它前面有多少个开头和它一样的即可。这个可以二分。总时间复杂度\(O(n\log n)\)
code:
#include<bits/stdc++.h>
#define I inline
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
#define abs(x) ((x)>0?(x):-(x))
#define ll long long
#define db double
#define lb long db
#define N (200000+5)
#define M ((1<<20)+5)
#define Ks (12+5)
#define mod 998244353
#define Mod (mod-1)
#define eps (1e-9)
#define ull unsigned ll
#define it iterator
#define Gc() getchar()
#define Me(x,y) memset(x,y,sizeof(x))
#define Mc(x,y) memcpy(x,y,sizeof(x))
#define d(x,y) (n*(x-1)+(y))
#define R(n) (rand()*rand()%(n)+1)
#define Pc(x) putchar(x)
#define LB lower_bound
#define UB upper_bound
#define PB push_back
using namespace std;
int n,m,k,x,y,z,A[N],B[N],Q[N];vector<int> S[N];
int main(){
freopen("1.in","r",stdin);
int i,j;scanf("%d%d",&n,&m);for(i=1;i<=n+1;i++) scanf("%d",&A[i]),!A[i]&&(x=i),S[A[i]].PB(i);
for(i=0;i<=m;i++) Q[i]=(i?Q[i-1]:0)+S[i].size();for(i=n+1;i;i--){
B[i]=A[x];x=(A[x]?Q[A[x]-1]:0)+(LB(S[A[x]].begin(),S[A[x]].end(),x)-S[A[x]].begin()+1);
}for(i=1;i<=n;i++) printf("%d ",B[i]);
}