HackerRank Warm-up 4.Repeated String, in C++

題目連結:Repeated String | HackerRank

如果TLE的話請回憶你的數學,相信可以找到解

long repeatedString(string s, long n) {
    long len=s.size(),ct=0;
    for(auto a:s){
        if(a=='a') ct++;
    }

    long ans=0;
    ans+=n/len*ct;
    for(int i=0;i<n%len;i++){
        if(s[i]=='a') ans++;
    }
    return ans;
}

發佈留言