牛客真题（35）-青草游戏

今天继续刷牛客真题，青草游戏

分析：
找规律，通过规律，判断游戏的输赢。

问题：
1、Python中if else 的简写方式；

方式C++代码：

#include<iostream>
using namespace std;
int main()
{
    int n;
    cin>>n;
    int a[n];
    for(int i=0;i<n;i++)
    {
        int x;
        cin>>x;
        a[i]=x;
    }
    for(int j=0;j<n;j++)
    {
        if(a[j]%5==1||a[j]%5==3||a[j]%5==4)
            cout<<"niu"<<endl;
        else
            cout<<"yang"<<endl;
    }
    return 0;
}

附上Python代码：

for i in range(int(input())):
    print('niu' if int(input())%5 in [1,3,4] else 'yang')