1. 程式人生 > >Complete the Word-一道字元統計,解決很簡單,優化有點難

Complete the Word-一道字元統計,解決很簡單,優化有點難

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;
}