第十八届同济大学程序设计竞赛暨高校网络友谊赛D-平衡的字符串
题目链接
设原串为s。若每个子串都要保持平衡，则对于任意的i都有
1
| s[i]==s[i+k]==s[i+2k]==...
|
也就是说，与i相关的一系列字符中不能同时出现0和1（但可以同时出现数字和问号）。只需要对于所有字符扫一遍，看看它后面的系列字符是否有同时出现0和1，出现了就退出报错，没出现的话就将它们中的问号都改成相同的字符；都没出现的话就再验证原串长度为k的前缀是否平衡，如果这个前缀平衡的话，那么整个串都平衡了。
验证这个前缀是否平衡这一步骤要在看完各个字符后面的系列字符是否有同时出现0和1后进行，不然的话验证的前缀可能带有待更改的问号，影响对串平衡性的判断。
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
| #include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <map>
#include <set>
#include <queue>
#include <stack>
#include <cctype>
using namespace std;
char s[1000005];
int main()
{
int sum0 = 0, sum1 = 0, sumw = 0, n, k, i, j;
scanf("%d%d", &n, &k);
scanf("%s", s + 1);
if (k & 1)
{
printf("No\n");
return 0;
}
for (i = 1; i <= n; i++)
{
char ori = s[i];
for (j = i; j <= n; j += k)
{
if (s[j] != ori)
{
if (ori == '?')
ori = s[j];
else if (s[j] != '?')
break;
}
}
if (j <= n)
break;
for (j = i; j <= n; j += k)
s[j] = ori;
}
if (i <= n)
{
printf("No\n");
return 0;
}
for (i = 1; i <= k; i++)
{
if (s[i] == '0')
sum0++;
else if (s[i] == '1')
sum1++;
else
sumw++;
}
if (abs(sum0 - sum1) > sumw)
{
printf("No\n");
return 0;
}
printf("Yes\n");
return 0;
}
|