結果
問題 | No.1645 AB's abs |
ユーザー |
![]() |
提出日時 | 2021-08-13 21:44:36 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 6 ms / 2,000 ms |
コード長 | 3,177 bytes |
コンパイル時間 | 2,227 ms |
コンパイル使用メモリ | 206,324 KB |
最終ジャッジ日時 | 2025-01-23 18:38:12 |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 36 |
ソースコード
#include <bits/stdc++.h>#define REP_(i, a_, b_, a, b, ...) \for (int i = (a), END_##i = (b); i < END_##i; ++i)#define REP(i, ...) REP_(i, __VA_ARGS__, __VA_ARGS__, 0, __VA_ARGS__)#define ALL(x) std::begin(x), std::end(x)using i64 = long long;#include <atcoder/modint>using Mint = atcoder::modint998244353;std::ostream &operator<<(std::ostream &os, const Mint &m) {return os << m.val();}template<typename T, typename U>inline bool chmax(T &a, U b) {return a < b and ((a = std::move(b)), true);}template<typename T, typename U>inline bool chmin(T &a, U b) {return a > b and ((a = std::move(b)), true);}template<typename T>inline int ssize(const T &a) {return (int) std::size(a);}template<typename T>std::istream &operator>>(std::istream &is, std::vector<T> &a) {for (auto &x : a) is >> x;return is;}template<typename T, typename U>std::ostream &operator<<(std::ostream &os, const std::pair<T, U> &a) {return os << "(" << a.first << ", " << a.second << ")";}template<typename Container>std::ostream &print_seq(const Container &a, std::string_view sep = " ",std::string_view ends = "\n",std::ostream &os = std::cout) {auto b = std::begin(a), e = std::end(a);for (auto it = std::begin(a); it != e; ++it) {if (it != b) os << sep;os << *it;}return os << ends;}template<typename T, typename = void>struct is_iterable : std::false_type {};template<typename T>struct is_iterable<T, std::void_t<decltype(std::begin(std::declval<T>())),decltype(std::end(std::declval<T>()))>>: std::true_type {};template<typename T, typename = std::enable_if_t<is_iterable<T>::value &&!std::is_same<T, std::string_view>::value &&!std::is_same<T, std::string>::value>>std::ostream &operator<<(std::ostream &os, const T &a) {return print_seq(a, ", ", "", (os << "{")) << "}";}void print() { std::cout << "\n"; }template<class T>void print(const T &x) {std::cout << x << "\n";}template<typename Head, typename... Tail>void print(const Head &head, Tail... tail) {std::cout << head << " ";print(tail...);}void read_from_cin() {}template<typename T, typename... Ts>void read_from_cin(T &value, Ts &...args) {std::cin >> value;read_from_cin(args...);}#define INPUT(type, ...) \type __VA_ARGS__; \read_from_cin(__VA_ARGS__)#ifdef MY_DEBUG#include "debug_dump.hpp"#else#define DUMP(...)#endifusing namespace std;auto solve() {INPUT(int, n);vector<i64> a(n);cin >> a;sort(ALL(a));i64 asum = accumulate(ALL(a), 0LL);auto dp = vector(n + 1, vector(asum + 1, Mint(0)));dp[0][0] = 1;REP(i, n) {i64 val = a[i];REP(j, asum + 1) {dp[i + 1][j] += dp[i][j];if (j + val > asum) break;dp[i + 1][j + val] += dp[i][j];}}Mint ans = 0;REP(j, asum + 1) {ans += abs(asum - j * 2) * dp[n][j];}return ans;}int main() {ios_base::sync_with_stdio(false), cin.tie(nullptr);cout << std::fixed << std::setprecision(15);int t = 1;REP(test_case, t) {auto ans = solve();print(ans);}}