bzoj 3998 第K小子串

【题意】

　　询问排名第k的子串是谁，0代表相同子串不同位置算作相同，1代表相同子串不同位置算作不同。　　

【思路】

　　0的情况和这个题一样每个子串不同位置出现次数算作1；

　　至于1，统计val作为该子串在不同位置出现的次数，就是求一下|right|的大小呗，相应修改一下就可以了。

Sample Input

aabc
0 3

Sample Output

aab

//china no.1#pragma comment(linker, "/STACK:1024000000,1024000000")#include <vector>#include <iostream>#include <string>#include <map>#include <stack>#include <cstring>#include <queue>#include <list>#include <stdio.h>#include <set>#include <algorithm>#include <cstdlib>#include <cmath>#include <iomanip>#include <cctype>#include <sstream>#include <functional>#include <stdlib.h>#include <time.h>#include <bitset>using namespace std;#define pi acos(-1)#define s_1(x) scanf("%d",&x)#define s_2(x,y) scanf("%d%d",&x,&y)#define s_3(x,y,z) scanf("%d%d%d",&x,&y,&z)#define PI acos(-1)#define endl '\n'#define srand() srand(time(0));#define me(x,y) memset(x,y,sizeof(x));#define foreach(it,a) for(__typeof((a).begin()) it=(a).begin();it!=(a).end();it++)#define close() ios::sync_with_stdio(0); cin.tie(0);#define FOR(x,n,i) for(int i=x;i<=n;i++)#define FOr(x,n,i) for(int i=x;i<n;i++)#define fOR(n,x,i) for(int i=n;i>=x;i--)#define fOr(n,x,i) for(int i=n;i>x;i--)#define W while#define sgn(x) ((x) < 0 ? -1 : (x) > 0)#define bug printf("***********\n");#define db double#define ll long long#define mp make_pair#define pb push_backtypedef pair<long long int,long long int> ii;typedef long long LL;const int INF=0x3f3f3f3f;const LL LINF=0x3f3f3f3f3f3f3f3fLL;const int dx[]={-1,0,1,0,1,-1,-1,1};const int dy[]={0,1,0,-1,-1,1,-1,1};const int maxn=5e5+10;const int maxx=600005;const double EPS=1e-8;const double eps=1e-8;const int mod=1e9+7;template<class T>inline T min(T a,T b,T c) { return min(min(a,b),c);}template<class T>inline T max(T a,T b,T c) { return max(max(a,b),c);}template<class T>inline T min(T a,T b,T c,T d) { return min(min(a,b),min(c,d));}template<class T>inline T max(T a,T b,T c,T d) { return max(max(a,b),max(c,d));}template <class T>inline bool scan_d(T &ret){char c;int sgn;if (c = getchar(), c == EOF){return 0;}while (c != '-' && (c < '0' || c > '9')){c = getchar();}sgn = (c == '-') ? -1 : 1;ret = (c == '-') ? 0 : (c - '0');while (c = getchar(), c >= '0' && c <= '9'){ret = ret * 10 + (c - '0');}ret *= sgn;return 1;}inline bool scan_lf(double &num){char in;double Dec=0.1;bool IsN=false,IsD=false;in=getchar();if(in==EOF) return false;while(in!='-'&&in!='.'&&(in<'0'||in>'9'))in=getchar();if(in=='-'){IsN=true;num=0;}else if(in=='.'){IsD=true;num=0;}else num=in-'0';if(!IsD){while(in=getchar(),in>='0'&&in<='9'){num*=10;num+=in-'0';}}if(in!='.'){if(IsN) num=-num;return true;}else{while(in=getchar(),in>='0'&&in<='9'){num+=Dec*(in-'0');Dec*=0.1;}}if(IsN) num=-num;return true;}void Out(LL a){if(a < 0) { putchar('-'); a = -a; }if(a >= 10) Out(a / 10);putchar(a % 10 + '0');}void print(LL a){ Out(a),puts("");}//freopen( "in.txt" , "r" , stdin );//freopen( "data.txt" , "w" , stdout );//cerr << "run time is " << clock() << endl;int root,last;int tots;  //总结点int l;  //字符串长度int sv[maxn*2];struct sam_node{    int fa,son[26];    int len;    void init(int _len)    {        len = _len;        fa = -1;        memset(son,-1,sizeof(son));    }}t[maxn*2];void sam_init(){    tots = 0;    root = last = ++tots;    t[tots].init(0);}void extend(char ch){    int w=ch-'a';    int p=last;    int np=++tots;t[tots].init(t[p].len+1);    sv[l]=np;    int q,nq;    while(p!=-1&&t[p].son[w]==-1){t[p].son[w]=np;p=t[p].fa;}    if (p==-1) t[np].fa=root;    else    {        q=t[p].son[w];        if (t[p].len+1==t[q].len){t[np].fa=q;}        else        {            nq=++tots;t[nq].init(0);            t[nq]=t[q];            t[nq].len=t[p].len+1;            t[q].fa=nq;t[np].fa=nq;            while(p!=-1&&t[p].son[w]==q){t[p].son[w]=nq;p=t[p].fa;}        }    }    last=np;}int w[maxn], r[maxn*2];void topsort(){    for(int i = 0; i <= l; ++i) w[i] = 0;    for(int i = 1; i <= tots; ++i) w[t[i].len]++;    for(int i = 1; i <= l; ++i) w[i] += w[i-1];    for(int i = tots; i >= 1; --i) r[w[t[i].len]--] = i;    r[0] = 0;}int dp[maxn*2];char s[maxn];int op,k,p;int sum[maxn*2];void work(){    if(k>sum[1])    {        puts("-1");        return ;    }    else    {        W(k>dp[p])        {            for(int i=0;i<26;i++)            {                if(t[p].son[i]!=-1)                {                    if(sum[t[p].son[i]]>=k)                    {                        putchar('a'+i);                        k-=dp[p];                        p=t[p].son[i];                        break;                    }                    else                    {                        k-=sum[t[p].son[i]];                    }                }            }        }    }    return ;}int main(){    W(scanf("%s",s)!=EOF)    {        s_2(op,k);        l=strlen(s);        sam_init();        FOr(0,l,i) extend(s[i]);        topsort();        FOR(0,tots,i) dp[i]=0;        int now=root;        FOr(0,l,i)        {            now=t[now].son[s[i]-'a'];            dp[now]++;        }        for(int i=tots;i;i--)        {            p = r[i];            if(op==1)            {                if(t[p].fa != -1) dp[t[p].fa] += dp[p];            }            else dp[p]=1;        }        dp[1]=0;        for(int i=tots;i;i--)        {            p=r[i];            sum[p]=dp[p];            FOr(0,26,j)            {                now=t[p].son[j];                if(t[p].son[j]!=-1)                    sum[p]+=sum[now];            }        }        work();    }}