「日常训练」Phone Numbers (CFR466D2C)

时间:2023-03-09 18:06:17
「日常训练」Phone Numbers  (CFR466D2C)

题意(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;
}