MOOC 查找字符串

题目内容：

给定一个字符串，在字符串中找到第一个连续出现至少k次的字符。

输入格式:

第一行包含一个正整数k，表示至少需要连续出现的次数。1 <= k <= 100。

第二行包含需要查找的字符串。字符串长度在1到100之间，且不包含任何空白符。

输出格式：

若存在连续出现至少k次的字符，输出该字符；否则输出NO。

输入样例1：

3

abcccaaab

输出样例1：

c

输入样例2：

3

abccaab

输出样例2：

NO

代码如下：

#include<stdio.h>
int main() {
	int num;
	scanf("%d", &num);
	getchar();
	char s[99];
	gets(s);
	int judge = 0;
	int i=1, cnt = 1;
	while (s[i] != '\0') {
		if (s[i] == s[i - 1]) {
			cnt++;
		}
		else {
			cnt = 1;
		}
		if (cnt >= num) {
			printf("%c", s[i]);
			judge = 1;
			break;
		}
		i++;
	}
	if (judge == 0) {
		printf("NO");
	}
	return 0;
}

若有帮助，请不要吝啬您的点赞