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;
}