EOJ 3188 坏掉的彩灯

题目简介

---

Bob 有一串节日彩灯，彩灯共有 4 种颜色（R,B,Y,G 表示），可惜有一些灯泡已经坏掉了（ ! 表示）。

Bob 并不记得彩灯的序列，只记得这串彩灯的任意 4 个连续的灯泡颜色都不相同。

Bob 需要统计每种颜色坏掉的彩灯有多少个。

说明

---

0123下标对应RBYG四种颜色。其中tot和ans数组存储每种颜色的灯应有的总数，cnt为实际的总数，两者相减即可。

#include <stdio.h>
#include <string.h>

int main(void)
{
    int cas, t;
    scanf("%d", &cas);
    for (t = 0; t < cas; ++t){
        char s[101];
        scanf("%s", s);
        int len = strlen(s), i;
        int cnt[4] = {0}, tot[4] = {0}, ans[4];
        for (i = 0; i < len; ++i){
            ++tot[i % 4];
            switch(s[i]){
                case 'R': ++cnt[0]; ans[0] = i % 4; break;
                case 'B': ++cnt[1]; ans[1] = i % 4; break;
                case 'Y': ++cnt[2]; ans[2] = i % 4; break;
                case 'G': ++cnt[3]; ans[3] = i % 4; break;
            }
        }
        printf("case #%d:\n%d %d %d %d\n", t, tot[ans[0]] - cnt[0], tot[ans[1]] - cnt[1], tot[ans[2]] - cnt[2], tot[ans[3]] - cnt[3]);
    }
    return 0;
}