結果

問題 No.1645 AB's abs
ユーザー tarattata1tarattata1
提出日時 2021-08-13 22:18:45
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 11 ms / 2,000 ms
コード長 1,583 bytes
コンパイル時間 2,621 ms
コンパイル使用メモリ 157,248 KB
実行使用メモリ 11,904 KB
最終ジャッジ日時 2024-10-03 19:45:53
合計ジャッジ時間 4,012 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <string>
#include <cstring>
#include <cstdlib>
#include <cmath>
#include <algorithm>
#include <vector>
#include <set>
#include <map>
#include <queue>
#include <stack>
#include <list>
#include <iterator>
#include <cassert>
#include <numeric>
#include <functional>
#include <ctime>
#include <bitset>
#pragma warning(disable:4996) 

#define ATCODER
#ifdef ATCODER
#include <atcoder/all>
#endif

typedef long long ll;
typedef unsigned long long ull;
#define LINF  9223300000000000000
#define LINF2 1223300000000000000
#define LINF3 1000000000000
#define INF 2140000000
//const long long MOD = 1000000007;
const long long MOD = 998244353;

using namespace std;
#ifdef ATCODER
using namespace atcoder;
#endif

const int M = 10005;
ll dp[105][M * 2];


void solve()
{
    int n;
    scanf("%d", &n);
    vector<int> a(n);
    for (int i = 0; i < n; i++) {
        scanf("%d", &a[i]);
    }
    dp[0][M] = 1;
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < M * 2; j++) {
            if (dp[i][j] == 0) continue;

            int j2 = j + a[i];
            dp[i + 1][j2] = (dp[i + 1][j2] + dp[i][j]) % MOD;
            j2 = j - a[i];
            dp[i + 1][j2] = (dp[i + 1][j2] + dp[i][j]) % MOD;
        }
    }
    ll ans = 0;
    for (int j = 0; j < M * 2; j++) {
        ans = (ans + abs(j-M)*dp[n][j]) % MOD;
    }
    printf("%lld\n", ans);
    return;
}

int main()
{
#if 1
    solve();
#else
    int T, t;
    scanf("%d", &T);
    for (t = 0; t < T; t++) {
        //printf("Case #%d: ", t + 1);
        solve();
    }
#endif
    return 0;
}
0