Codeforce 128B：String

Codeforce 128B：String

题目链接：http://codeforces.com/problemset/problem/128/B

题目大意：给出一个字符串s，问字典序排第k个的子串是什么.

贪心

我们可以很轻松地求出以prefix为前缀的子串个数，故可以按字典序枚举后一个字母：

• 若以当前prefix+c（'+'为字符串的连接操作）为前缀的子串个数和小于k，则继续枚举下一个前缀prefix+(c+1)；
• 若以当前prefix+c为前缀的子串个数和小于等于k，那么所要求的子串的前缀定为prefix+c.

/*注意：子串和可能会爆int*/

代码如下：

#include <iostream>
#include <vector>
#include <string>
using namespace std;
string s,ans;
int k,len;
void solve(vector<int> a,int &k){
    if(k<=0||(int)a.size()==0)return;
    for(int j='a';j<='z';++j){
        long long pre=0;
        for(int i=0;i<(int)a.size();++i)
            if(s[a[i]]==j)pre+=len-a[i];
        if(pre>=k){
            ans+=j;
            vector<int>t;
            for(int i=0;i<(int)a.size();++i){
                if(s[a[i]]==j){
                    k--;
                    if(a[i]+1<len)t.push_back(a[i]+1);
                }
            }
            solve(t,k);
            return;
        }else k-=pre;
    }
}
int main(void){
    std::ios::sync_with_stdio(false);
    cin>>s>>k;
    vector<int>a;
    len=s.length();
    for(int i=0;i<len;++i)a.push_back(i);
    solve(a,k);
    if(k>0)cout<<"No such line.";
    else cout<<ans;
}