【十三届蓝桥杯C++国赛试题】电线上的小鸟

题目描述

输入

第一行输入一个正整数N（3≤N≤1000），表示有N只小鸟落在电线上

第二行输入N个字符，字符只能为“q”和“p”，“q”表示小鸟头向左看，“p”表示小鸟头向右看，字符之间以一个空格隔开

输出

一行输出三个整数，分别为有几只小鸟被0只小鸟看到；有几只小鸟被1只小鸟看到；有几只小鸟被2只小鸟看到；整数之间以一个空格隔开

#include <iostream>
#include <algorithm>
using namespace std;
const int N = 1007;
int n, a1, a2, a3;
char a[N];
int main() {
    cin >>n;
    for (int i = 0; i < n; i++)
        cin >> a[i];
    for (int i = 0; i < n; i++) {
        if (i == 0) {
            if (a[i+1] == 'p') a1++;
            else a2++;
        } else if (i == n-1) {
            if (a[i-1] == 'q') a1++;
            else a2++;
        } else {
            if (a[i-1]=='q' && a[i+1]=='p') a1++;
            else if (a[i-1]=='p' && a[i+1]=='q') a3++;
            else a2++;
        }
    }
    cout << a1 << " "<< a2 << " "<< a3;
    return 0;
}