MoonLight的运算问题

MoonLight的运算问题
 

月色哥哥手中有一个数字 x，最初 x=0。

给出一个长度为 n的序列 a，月色哥哥会从序列的第一个元素 a1按顺序看到序列的最后一个元素 an。对于序列的第 i个元素 ai​，月色哥哥可以进行下面的操作之一：
 

•  令 x=x⋅ai；
•  令 x=x+ai​。

请求出 x 的最大值，并输出这个最大值除 998244353的余数。

输入描述:

第一行包含一个整数 T(1≤T≤105)，表示测试用例的组数。

对于每组测试用例：

第一行包含一个整数 n(1≤n≤2⋅105)，表示序列的长度。

第二行包含 n 个整数 a1…an(0≤ai≤109），表示该序列。

保证对于所有的测试用例，n 的总和不超过 2⋅105。

输出描述:

对于每组测试用例：

仅输出一行，包含一个整数，表示答案。

示例1

输入

3
2
1 1
1
0
1
998244353

输出

2
0
0

思路：每次操作，对于数x来说，大于1就乘，否则加，另外需加上一个条件，就是当初始值大于1之后，每次操作除0和1外都是乘

AC代码：

#include<bits/stdc++.h>
using namespace std;
#define p 998244353
int main()
{
     int t;
     cin>>t;
     while(t--)
     {
     	int n;
     	cin>>n;
     	long long x,y=0,flag=0;
     	for(int i=0;i<n;i++)
     	{
     		cin>>x;
            if(flag&&x>1)
            y*=x;
            else
            y+=x;
            if(y>1)
            flag=1;
            y%=p;
		}
		cout<<y<<endl;
	 }
	 return 0;
}