CF1537E2 Erase and Extend 題解
分析
透過觀察題目,可以證明結果一定是由多次字首複製得來的。
題目要求你進行刪和複製的操作,與其交替著操作,不如直接先刪到最優的字首再進行復制。
現在就是要找最優的字首。從頭一位一位往後遍歷。用 \(l\) 來儲存目前最優字首的長度,第 \(i\) 位對應複製後的 \(k\) 位最優字首就是第 \(i\bmod l\) 位。在往後遍歷時,如果下一位的 ASCII 碼比最優字首第 \(i\bmod l\) 位的大,就說明後面沒有最優解了,直接 break
。如果小於的話,說明當前的解更優,更新一下長度即可。
複雜度 \(\mathcal O(n)\)。
程式碼
#include<bits/stdc++.h>
using namespace std;
namespace Raiden
{
int work()
{
string s;
int n,k;
int l=1;
cin>>n>>k>>s;
for(int i=0;i<n;i++)
{
if(s[i]>s[i%l])
break;
else if(s[i]<s[i%l])
l=i+1;
}
for(int i=0;i<k;i++)
cout<<s[i%l];
return 0;
}
}
int main()
{
ios::sync_with_stdio(false);
cin.tie(0);
return Raiden::work();
}