比赛 五一大礼包 评测结果 AAEEEEEEEE
题目名称 组一辈子乐队 最终得分 20
用户昵称 彭欣越 运行时间 1.227 s
代码语言 C++ 内存使用 4.61 MiB
提交时间 2026-05-04 11:08:47
显示代码纯文本
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1010,mod=998244353;
int n,k,a[N];
ll sum=0,s[N][N],p[N][N];
int main () {
    freopen("MyGo.in","r",stdin);
    freopen("MyGo.out","w",stdout);
    ios::sync_with_stdio(0);
    cin.tie(0),cout.tie(0);
    cin >> n >> k;
    for (int i=1;i<=n;i++) {
        cin >> a[i]; 
    }
    for (int i=1;i<=n;i++) {
        for (int j=i+1;j<=n;j++) {
            p[i][j]=s[i][j]=(a[i]+a[j])%mod;
            sum=(sum+p[i][j])%mod;
        }
    }
    cout << sum <<"\n";
    for (int t=2;t<=k;t++) {
        sum=0;
        for (int i=1;i<=n;i++) {
            for (int j=i+1;j<=n;j++) {
                p[i][j]=s[i][j]*p[i][j]%mod;
                sum=(sum+p[i][j])%mod;
            }
        }
        cout << sum <<"\n";
    }
    return 0;
}