第八届“图灵杯”NEUQ-ACM程序设计竞赛个人赛——H题 数羊

题目描述

憨憨小杨晚上睡不着觉，就开始数羊，她觉得一只一只数太慢了，突发奇想出了一种新的数羊方式，羊羊数量A(n,m)由两个整形变量n和m决定，计算方式如下:

现在给出n和m的值，请你帮小杨数数一共有多少只羊。

输入描述

多组输入。

第一行包含一个整数$T(1\le T\le 1000)$，表示有T组测试数据。

每组测试数据包含一行，包含两个整数$n(1\le n\le 10^9)$和$m(0\le m\le 2)$.

输出描述

对每一组输入，在一行中输出A(n,m)的值,由于输出的结果可能会很大，答案对998244353取模

示例

输入

3
3 0
3 1
3 2

输出

5
6
8

AC的C++代码

#include<iostream>
using namespace std;

int main()
{
	int n,m,T,i,ans=0;
	cin >> T;
	while(T--)
	{
		ans=0;
		cin>>n>>m;
		if(m == 0)
			ans=(n+2)%998244353;
		else if(m == 1)
			ans=2*n%998244353;
		else
		{	ans=1;
			for(int j=0; j<n; j++)
				ans=ans*2%998244353;
		}
		cout<<ans<<endl;
	}
	return 0;
 }