Cho một xâu S và số nguyên dương K. Hãy viết chương trình kiểm tra xem trong xâu S có K kí tự liền nhau giống nhau hay không? Input Specification Dòng 1 là chuỗi kí tự S có độ dài nhỏ hơn 1000. Dòng 2 là số nguyên k. Output Specification In ra YES nếu thoả mãn yêu cầu bài toán, in ra NO nếu ngược lại Sample Input abcaab 2 Sample Output YES

Giải:

Chúng ta sẽ tách xâu ra thành ít phần nhất sao cho mỗi phần chỉ có đúng 1 loại chữ cái tạo thành

Ví dụ: abcaab -> tách thành {"a","b","c","aa","b"}

Sau đó, ta sẽ duyệt qua dãy tạo được và tìm xem có phần tử nào có độ dài lớn hơn hoặc bằng k hay không. Nếu có thì in YES, nếu không thì NO.

Code

#include <bits/stdc++.h>
using namespace std;

string s;
char curChar = '!';
int k,curLen;

int main(){
    ios_base::sync_with_stdio(0);cin.tie(0);cout.tie(0);

    cin >> s >> k;
    s += '!';
    for(char c: s){
        if(c == curChar) curLen++;
        else{
            if(curLen >= k){
                cout << "YES";
                return 0;
            }
            curLen = 1;
            curChar = c;
        }
    }
    cout << "NO";
    return 0;
}

$#Nguyenchuong$