只要sum和x小于等于1就用加,大于等于1就用乘法,因为会对sum取模,所以直接判断不准确,用一个flag记录
#include<iostream>
#include<string>
#define ll long long
using namespace std;
const ll mod=998244353;
int main()
{
int t,n;
ll x;
ll sum=0,ans;
cin>>t;
while(t--)
{
int flag=0;
cin>>n;
sum=0,ans=0;
for(int i=0;i<n;i++)
{
cin>>x;
if(!flag||x<=1)sum=(sum+x)%mod;
else sum=((sum%mod)*(x%mod))%mod;
ans+=x;
if(ans>=2)flag=1;
}
cout<<sum<<endl;
}
return 0;
}