Score : 600 points
Problem Statement
Given are a sequence of N integers A1, A2, …, AN and a positive integer S.
For a pair of integers (L,R) such that 1≤L≤R≤N, let us define f(L,R) as follows:
- f(L,R) is the number of sequences of integers (x1,x2,…,xk) such that L≤x1<x2<⋯<xk≤R and Ax1+Ax2+⋯+Axk=S.
Find the sum of f(L,R) over all pairs of integers (L,R) such that 1≤L≤R≤N. Since this sum can be enormous, print it modulo 998244353.
Constraints
- All values in input are integers.
- 1≤N≤3000
- 1≤S≤3000
- 1≤Ai≤3000
Input is given from Standard Input in the following format:
N S
A1 A2 ... AN
Output
Print the sum of f(L,R), modulo 998244353.
3 4
2 2 4
5
The value of f(L,R) for each pair is as follows, for a total of 5.
- f(1,1)=0
- f(1,2)=1 (for the sequence (1,2))
- f(1,3)=2 (for (1,2) and (3))
- f(2,2)=0
- f(2,3)=1 (for (3))
- f(3,3)=1 (for (3))
5 8
9 9 9 9 9
0
10 10
3 1 4 1 5 9 2 6 5 3
152