Exercise(14)：小小故事

/* 小小故事: 在天朝的鱼C总基地，大厅的墙壁上有N盏灯，分别按顺序从1到N编号。 每盏灯都可以每打开或关上。在每一秒钟，如果第I+1盏灯是亮的， 那么第I盏灯在下一秒会变化它的状态；特别的，如果第1盏灯是亮的， 那么第N盏灯在下一秒会变化它的状态。小甲鱼老师给你某一时刻所有灯的状态， 希望你能求出它们在M秒之后的状态。 输入描述： 输入有N+1行。第一行有两个数N,M。此后N行，每行是一个数，不是0就是1。第I+1行的数描述初始时第I盏灯的状态。如果这个数为0，表示第I盏灯是灭的；如果这个数为1，表示第I盏灯是亮的。 输出描述： 输出有N行，每行一个数，不是0就是1。第I行的数描述M秒后第I盏灯的状态。如果这个数为0，表示第I盏灯是灭的；如果这个数为1，表示第I盏灯是亮的。 输入范例： 3 1 0 0 1 输出范例： 0 1 1 问题分析： 题解可分为两步： 1、先判断灯1是否亮， 是则灯N 改变状态 2、使用for判断灯i+1(0<=i<=N-2)是否亮 若是 则灯i改变状态 若否 则灯i状态不变 */
#include <iostream>

int main()
{
    int i,j,n,m;                // i,j为循环变量 n为灯泡数 m为秒数 
    std::cin>>n>>m;             // 

    int bulbs[100];             // 表示灯泡状态 0 or 1 

    for(i=0;i<n;i++)            // 输入各灯泡的状态 
        std::cin>>bulbs[i];

    for(i=0;i<m;i++)            // 循环m秒 
    {
        if(bulbs[0]==1)         // 情况1 灯泡1亮 灯泡n-1改变状态 
        {
            if(bulbs[n-1]==1)
                bulbs[n-1] = 0;
            else
                bulbs[n-1] = 1;
        }

        for(j=0;j<n-1;j++)      // 情况2 判断其他灯泡 
        {
            if(bulbs[j+1]==1)
            {
                if(bulbs[j]==1)
                    bulbs[j] = 0;
                else
                    bulbs[j] = 1;
            }
        }
    }

    for(i=0;i<n;i++)            // 输出 
        std::cout<<bulbs[i]<<std::endl;
}