lcb-0077

1.000

3/3 tests· algorithms

Proposed solution

```python
import sys
import math

def solution():
    input_data = sys.stdin.read().strip().split()
    N, K = map(int, input_data[:2])
    A = list(map(int, input_data[2:2+N]))

    MOD = 998244353

    prefix_sum = [0] * (N + 1)
    for i in range(N):
        prefix_sum[i + 1] = prefix_sum[i] + A[i]

    result = 0
    for l in range(N):
        for r in range(l, N):
            current_sum = prefix_sum[r + 1] - prefix_sum[l]
            result = (result + current_sum ** K) % MOD

    print(result)

if __name__ == "__main__":
    solution()
```

Test output (stdout)

...                                                                      [100%]
3 passed in 0.05s

System prompt

You are an expert programmer. Solve the task exactly as specified. Output your solution as fenced code blocks using the required file name(s) and the exact function/type signatures requested. Prefer correctness; do not include prose outside code unless asked.