ในปัญหานี้ เราได้รับสตริง str ที่มีเพียง a และ b และจำนวนเต็ม N เพื่อให้สตริงถูกสร้างขึ้นโดยการต่อท้าย str n ครั้ง งานของเราคือการพิมพ์จำนวนสตริงย่อยทั้งหมดที่การนับของมากกว่าการนับ b
มาดูตัวอย่างทำความเข้าใจปัญหากัน
Input: aab 2 Output: 9 Explanation: created string is aabaab. Substrings with count(a) > count(b) : ‘a’ , ‘aa’, ‘aab’, ‘aaba’, ‘aabaa’, ‘aabaab’, ‘aba’, ‘baa’, ‘abaa’.
เพื่อแก้ปัญหานี้ เราจะต้องตรวจสอบว่าสตริงมีชุดย่อยคำนำหน้าที่จำเป็นหรือไม่ ที่นี่ เราจะตรวจสอบสตริง str ไม่ใช่เวอร์ชันเต็ม ที่นี่ w จะตรวจสอบสตริงตามคำนำหน้าและจำนวนการเกิดของ a และ b
โปรแกรมนี้จะแสดงการใช้งานโซลูชันของเรา
ตัวอย่าง
#include <iostream>
#include <string.h>
using namespace std;
int prefixCount(string str, int n){
int a = 0, b = 0, count = 0;
int i = 0;
int len = str.size();
for (i = 0; i < len; i++) {
if (str[i] == 'a')
a++;
if (str[i] == 'b')
b++;
if (a > b) {
count++;
}
}
if (count == 0 || n == 1) {
cout<<count;
return 0;
}
if (count == len || a - b == 0) {
cout<<(count*n);
return 0;
}
int n2 = n - 1, count2 = 0;
while (n2 != 0) {
for (i = 0; i < len; i++) {
if (str[i] == 'a')
a++;
if (str[i] == 'b')
b++;
if (a > b)
count2++;
}
count += count2;
n2--;
if (count2 == 0)
break;
if (count2 == len) {
count += (n2 * count2);
break;
}
count2 = 0;
}
return count;
}
int main() {
string str = "aba";
int N = 2;
cout<<"The string created by using '"<<str<<"' "<<N<<" times has ";
cout<<prefixCount(str, N)<<" substring with count of a greater than count of b";
return 0;
} ผลลัพธ์
The string created by using 'aba' 2 times has 5 substring with count of a greater than count of b
