2018 Wannafly summer camp Day8--连通块计数

#include <cstdio>
#include <cstring>
#include <cmath>
#include<algorithm>
#include<iostream>
using namespace std;
typedef long long ll;
const ll mo=998244353;
const int maxn=1e7+5;
int n;
ll   a[100005],ans=0;
ll fun[maxn];
ll kpow(ll a,ll b)
{
    ll res=1;
    while(b>0)
    {
        if(b&1) res=res*a%mo,b--;
        a=a*a%mo;
        b/=2;
    }
    return res%mo;
}
ll C(ll n,ll m)
{
    if(n<m) return 0;
    return fun[n]*kpow(fun[m]*fun[n-m]%mo,mo-2)%mo;
}
void init(){
    fun[0]=1;
    for(int i=1;i<=maxn;i++)
        fun[i]=fun[i-1]*i%mo;
}
int main()
{
    init();
    cin>>n;
    for(int i=0;i<n;i++){
       cin>>a[i];
       ans+=C(a[i]+1,2);
       ans%=mo;
    }
    ll b=1;
    for(int i=0;i<n;i++){
       b*=(a[i]+1);
       b%=mo;
    }
    cout<<(ans+b)%mo<<endl;
    return 0;
}