結果

問題 No.1645 AB's abs
ユーザー bal4ubal4u
提出日時 2021-08-14 08:03:00
言語 C
(gcc 13.3.0)
結果
AC  
実行時間 4 ms / 2,000 ms
コード長 1,165 bytes
コンパイル時間 585 ms
コンパイル使用メモリ 31,872 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-10-04 17:34:37
合計ジャッジ時間 2,004 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 36
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.c: In function 'in':
main.c:8:14: warning: implicit declaration of function 'getchar_unlocked' [-Wimplicit-function-declaration]
    8 | #define gc() getchar_unlocked()
      |              ^~~~~~~~~~~~~~~~
main.c:12:28: note: in expansion of macro 'gc'
   12 |         int n = 0; int c = gc();
      |                            ^~

ソースコード

diff # 

// yuki 1645 AB's abs
// 2021.8.13

#include <stdio.h>
#include <string.h>

typedef long long ll;
#define gc() getchar_unlocked()
#define pc(c) putchar_unlocked(c)

int in() {   // 非負整数の入力
	int n = 0; int c = gc();
	do n = 10 * n + (c & 0xf); while ((c = gc()) >= '0');
	return n;
}

const int mod = 998244353;
inline static int ABS(int a) { return a >= 0? a: -a; }
inline static void add_mod(int *a, int b) { *a = (*a + b) % mod; }
#define MA 10000
int dp[2][MA+10][2]; int N;

int main()
{
	int i, j, A, s, k0, k1;

	N = in();;
	dp[0][0][0] = 1, k0 = 0, k1 = 1;
	s = 0; for (i = 0; i < N; ++i) {
		A = in();
		memset(dp[k1], 0, sizeof(dp[0]));
		for (j = 0; j <= s; ++j) if (dp[k0][j][0]) {
			add_mod(&dp[k1][j+A][0], dp[k0][j][0]);
			add_mod(&dp[k1][ABS(j-A)][j-A < 0], dp[k0][j][0]);
		}
		for (j = 1; j <= s; ++j) if (dp[k0][j][1]) {
			add_mod(&dp[k1][j+A][1], dp[k0][j][1]);
			add_mod(&dp[k1][ABS(j-A)][j-A > 0], dp[k0][j][1]);
		}
		k1 = k0, k0 = !k0;
		s += A;
	}
	int ans = 0;
	for (i = 0; i < 2; ++i) {
		for (j = 0; j <= s; ++j) if (dp[k0][j][i]) {
			add_mod(&ans, (ll)j * dp[k0][j][i] % mod);
		}
	}
	printf("%d\n", ans);
	return 0;
}
0