Search
 
SCRIPT & CODE EXAMPLE
 

CPP

SUMOFPROD2 codechef solution

/*
Sum of product 2:
*/
/*
Python code:
a = [0]*1000001
b = [0]*1000001
d = [0]*1000001
m = 998244353

def f_1(x,y):
    if x<0 or y>x:
        return 0
    return a[x]*d[y]%m*d[x-y]%m

a[0] = b[0] = b[1] = d[0] = d[1] = 1

for i in range(1,1000001):
    a[i] = a[i-1]*i%m

for i in range(2,1000001):
    b[i] = m-m//i*b[m%i]%m

for i in range(2,1000001):
    d[i] = d[i-1]*b[i]%m

for _ in range(int(input())):
    n = int(input())
    arr = list(map(int,input().split()))[:n]
    c = c1 = r = 0
    for i in range(n):
        if arr[i] == 0:
            c += 1
        else:
            c1 += 1
    for i in range(c1+1):
        r = (r+i*f_1(c1+c-i,c))%m
    print((((r*(c+1)-f_1(c1+c-2,c-1))%m+m)%m+f_1(c1+c-2,c-1))*a[c1]%m*a[c]%m)
*/
/* C++ code:
*/
#include<bits/stdc++.h>
#define MOD 998244353
using namespace std;
long long fac[1000010], inv[1000010], finv[1000010];
long long C(long long x, long long y) {
  if (x < 0 || y > x)
    return 0;
  return fac[x] * finv[y] % MOD * finv[x - y] % MOD;
}
#define int long long
void solve() {
  int n,x,c0=0,c1=0,t=0;
  cin >> n;
  for(int i=1; i<=n; ++i) {
    cin >> x;
    c0+=x==0;
    c1+=x==1;
  }
  for(int i=0; i<=c1; ++i)
    t=(t+i*C(c1+c0-i,c0))%MOD;
  cout << (((t*(c0+1)-C(c1+c0-2,c0-1))%MOD+MOD)%MOD+C(c1+c0-2,c0-1))*fac[c1]%MOD*fac[c0]%MOD << endl;
}
signed main() {
  fac[0] = inv[0] = inv[1] = finv[0] = finv[1] = 1;
  for (long long i = 1; i <= 1000000; ++i)
    fac[i] = fac[i - 1] * i % MOD;
  for (long long i = 2; i <= 1000000; ++i)
    inv[i] = MOD - MOD / i * inv[MOD % i] % MOD;
  for (long long i = 2; i <= 1000000; ++i)
    finv[i] = finv[i - 1] * inv[i] % MOD;
  int T;
  cin >> T;
  while(T--) solve();
}
Comment

SUMOFPROD1 codechef solution

# Sum of Product 1:
# First system:

for _ in range(int(input())):
    n = int(input())
    arr = list(map(int,input().split()))[:n]
    res = 0
    c = 0
    for i in range(n):
        if arr[i] == 1:
            c += 1
        else:
            res += c*(c+1)//2
            c = 0
    res += c*(c+1)//2
    print(res)
    
# Second system:

'''
for _ in range(int(input())):
    n = int(input())
    arr = list(map(int,input().split()))[:n]
    res = 0
    c = 0
    for i in range(n):
        if arr[i] == 0:
            c = 0
        else:
            c += 1
        res += c
    print(res)
'''
Comment

PREVIOUS NEXT
Code Example
Cpp :: factorial in c++ using recursion 
Cpp :: std::count() in C++ STL 
Cpp :: how to remove maximum number of characters in c++ cin,ignore 
Cpp :: count number of prime numbers in a given range in c 
Cpp :: vector iterating 
Cpp :: getline() 
Cpp :: c++ get last element in vector 
Cpp :: map in cpp 
Cpp :: map count function c++ 
Cpp :: convert all strings in vector to lowercase or uppercase c++ 
Cpp :: cpp get exception type 
Cpp :: c++ array pointer 
Cpp :: resize vector c++ 
Cpp :: how to sort array in c++ stockoverflow 
Cpp :: range based for loop c++ 
Cpp :: long long int range c++ 
Cpp :: print Colored text in C++ 
Cpp :: print fps sfml 
Cpp :: how to make loop in c++ 
Cpp :: c++ set intersection 
Cpp :: c++ cin 
Cpp :: c++98 check if character is integer 
Cpp :: tabeau pseudo dynamique sur c++ 
Cpp :: get function in cpp. 
Cpp :: string copy in cpp 
Cpp :: std::string substr 
Cpp :: Shell-Sort C++ 
Cpp :: c++ last element of vector 
Cpp :: 83. remove duplicates from sorted list solution in c++ 
Cpp :: array bubble sort c++ static 
ADD CONTENT
Topic
Content
Source link
Name
4+5 =