「日常訓練」Phone Numbers (CFR466D2C)

SamHX發表於2018-04-19

題意（Codeforces 940C）

給定一字串，求比它字典序大的字串。限定其長度，並且只能用原串的字母。

分析

考慮原串長度lorigin與給定的長度lgiven。若給定長度大於原串長度，直接在後面加字母就可以了（甚至不論什麼字母都可以）。如果給定長度小於等於原串長度，那麼最簡單的操作是求原串的lg長度子串的後繼即可。後繼的求法很簡單：從後往前，對單位+“1”，若加滿了，回覆到“0”，對上一位+“1”（具體實現見程式碼，資料保證該串非字典序最大的串，這樣做也就不會報錯）。這裡的1/0用set實現。
這題的重點是把握字典序是個啥意思。

程式碼

#include<bits/stdc++.h>

#define inf 0x3f3f3f3f
#define PB push_back
#define MP make_pair
#define fi first
#define se second
#define lowbit(x) (x&(-x))
#define rep(i, a, b) for(int i = (a); i <= (b); i++)
#define per(i, a, b) for(int i = (a); i >= (b); i--)
#define pr(x) cout << #x << " = " << x << " ";
#define prl(x) cout << #x << " = " << x << endl;
#define ZERO(X) memset((X),0,sizeof(X))
#define ALL(X) X.begin(),X.end()
#define SZ(x) (int)x.size()

using namespace std;

typedef pair<int,int> PI;
typedef pair<pair<int,int>, int> PII;
typedef pair<pair<pair<int,int>, int>, int> PIII; 
typedef unsigned long long ull;
typedef long long ll;
typedef long double lb;
#define quickio ios::sync_with_stdio(false); cin.tie(0); cout.tie(0)
#define debug(...) fprintf(stderr, __VA_ARGS__), fflush(stderr)
/*      debug("Precalc: %.3f\n", (double)(clock()) / CLOCKS_PER_SEC);
clock_t z = clock();
        solve();
        //debug("Test: %.3f\n", (double)(clock() - z) / CLOCKS_PER_SEC);
*/
template<typename T = int>
inline T read() {
    T val=0, sign=1;
    char ch;
    for (ch=getchar();ch<'0'||ch>'9';ch=getchar())
        if (ch=='-') sign=-1;
    for (;ch>='0'&&ch<='9';ch=getchar())
        val=val*10+ch-'0';
    return sign*val;
}
int main()
{
    int n,k;string str; cin>>n>>k>>str;
    set<char> s;
    for(auto &chr:str) s.insert(chr); 
    string ansstr=str.substr(0,k);
    if(k>str.length())
    {
        cout<<ansstr;
        char chr=*(s.begin());
        for(int i=0;i!=k-str.length();++i) cout<<chr;
        cout<<endl;
    } 
    else
    {
        for(int i=ansstr.length()-1;i>=0;--i)
        {
            auto it=s.find(ansstr[i]);//,it_cpy=it;
            if(++it==s.end())
            {
                ansstr[i]=*(s.begin());
            }
            else
            {
                ansstr[i]=*it;
                break;
            }
        }
        cout<<ansstr<<endl;
    }
    return 0;
}

相關文章