牛客小白月赛76 DMoonLight的运算问题

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