Complete the Word-一道字元統計,解決很簡單,優化有點難
阿新 • • 發佈:2018-12-24
Codeforces Round #372 (Div. 2)B. Complete the Word
解決很簡單,直接列舉每一個26長度的字串看符不符合就行了,符合的話先把這個子字串’?’替換成缺失的字母,這個字串以後的’?’用任意字母填充完就可以輸出。這樣做的複雜度是O(26*26*|s|)
可以優化成O(|s|),原理是用一個計數器counter,先統計第一個26長度的字串字母的出現情況,然後從第一個26長度的字串開始往後移一個字元,把第一個字元代表的字母減一(如果減為零counter減一),如果是問號的話counte也減一,然後把新加進來的字母加一(如果是一的話counter加一),如果是問號的話counte也加一,然後檢查counter是否為26就可以了,26的話就填充輸出。
先貼沒有優化過的程式碼
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 50050;
char s[maxn];
int a[26];
int main()
{
scanf("%s",s);
int len = strlen(s);
int ok = 0;
for(int i=0;i<len;++i){
for (int j=0;j<26;++j)a[j] = -1;
int letter = 0,q = 0;
for(int j=i;j<i+26;++j){
if(s[j]=='?')++q;
else{
if(a[ s[j]-'A' ] == -1){
a[s[j]-'A'] = j;
++letter;
}
else break;
}
}
if (letter + q == 26){
for(int j=i;j<i+26;++j){
if(s[j]=='?'){
for(int k=0;k<26;++k){
if(a[k] == -1){
a[k] = j;
s[j] = k + 'A';
break;
}
}
}
}
ok = 1;
break;
}
}
if(ok){
for(int i=0;i<len;++i)if(s[i]=='?')s[i]='A';
printf("%s\n",s);
}else printf("-1\n");
return 0;
}
優化為O(|s|)的版
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn = 50050;
char s[maxn];
int cnt[26],len;
void fulfill(int pos){
int x = 0;
while(cnt[x]>0)x++;
for(int i=pos;i<pos+26;++i){
if(s[i]=='?'){
s[i] = 'A' + x;
x++;
while(cnt[x]>0)x++;
}
}
for(int i=0;i<len;++i)if(s[i]=='?')s[i]='A';
}
int main()
{
scanf("%s",s);
len = strlen(s);
if(len<26){ printf("-1\n");return 0;}
memset(cnt,0,sizeof(cnt));
int counter = 0;
for(int i=0;i<26;++i){
if(s[i]=='?'){++counter;continue;}
else {
cnt[s[i]-'A']++;
if(cnt[s[i]-'A']==1)++counter;
}
}
if(counter==26){
fulfill(0);
printf("%s\n",s);
return 0;
}
for(int i=26;i<len;++i){
if(s[i-26]=='?')--counter;
else{
cnt[s[i-26]-'A']--;
if(cnt[s[i-26]-'A']==0)--counter;
}
if(s[i]=='?')++counter;
else{
cnt[s[i]-'A']++;
if(cnt[s[i]-'A']==1)++counter;
}
if(counter==26){
fulfill(i-25);
printf("%s\n",s);
return 0;
}
}
printf("-1\n");
return 0;
}