题目描述

n个村庄坐落成一条直线,A和B两个部落生活在这里。

每个村庄要么无人居住,要么被两个部落之一所占据。

如果一个无人居住的村庄两侧都是被部落A占据的村庄,那么这个村庄也视作被部落A占据;部落B亦同。

请求出被部落A和B分别占据的村庄个数。


输入格式

第一行为正整数T,表示存在T组测试数据,1≤T≤20。

每组测试数据输入一行,包含一个字符串s,表示每个村庄的状态,|s|≤100000

字符串仅包含A、B、. 三种字符,分别表示被部落A占据、被部落B占据,以及无人居住。

输出格式

每组测试数据输出两个整数,分别表示被部落A和B控制的村庄数量。

输入样例

4

A..A..B...B

..A..

A....A

..B..B..B..

输出样例

4 5

1 0

6 0

0 7

思路: 对于一段连续的A . . . A ,可以直接统计A的数量,同理对于B . . . B也是类似的。对于A . . . B 或者B . . . A 则不需要考虑中间这一段。

所以记录上一个非⋅的下标记为last_idx,对于当前的j满足s[j]≠ ⋅  and  s[last_idx] = s[j] ,则分类讨论(如果s[j]=='A',A控制的村庄增加j - last_idx;否则B控制的村庄增加j - last_idx),统计这一段的长度。

对于s[i]≠ ⋅  and  s[last_idx] ≠ s[j],(如:A..B,前后不一致)只需要统计当前s[j]带来的贡献(+1)即可。

参考程序:


t=int(input())
for i in range(t):
    s=input()
    last_idx=-1
    ansA=0
    ansB=0
    len_s=len(s)
    for j in range(len_s):
        if s[j] != '.':
            if last_idx != -1 and s[last_idx] == s[j]:#与前一个字母比较
                if s[j] == 'A':
                    ansA += j - last_idx
                else:
                    ansB += j - last_idx
            elif s[j] == 'A':
                ansA+=1
            else:
                ansB+=1
            last_idx = j
    print(ansA,ansB)


本站内容未经许可,禁止任何网站及个人进行转载。