Submission #17222478

---

Source Code Expand

#include <bits/stdc++.h>

typedef long long int64;

const int N = 3e2, MOD = 998244353;

int n, m, s, a[N + 5], f[N + 5][N + 5][N + 5], g[N + 5][N + 5][N + 5], h[N + 5][N + 5][N + 5];
char str[N + 5];

inline void add(int &x, const int &y) {
	(x += y) >= MOD && (x -= MOD);
}
inline void sub(int &x, const int &y) {
	(x -= y) < 0 && (x += MOD);
}
inline int add(const int &x) {
	return x >= MOD ? x - MOD : x;
}
inline int sub(const int &x) {
	return x < 0 ? x + MOD : x;
}
int main() {
	scanf("%s%d", str + 1, &s);
	n = strlen(str + 1);
	str[n + 1] = '0';
	int sum = 0;
	for (int cnt = 0, i = 1; i <= n + 1; i++) {
		if (str[i] == '0') {
			a[++m] = cnt;
			cnt = 0;
		} else {
			cnt++;
			sum++;
		}
	}
	s = std::min(s, sum - a[1]);
	f[0][0][0] = 1;
	std::fill(g[0][0], g[0][0] + n + 1, 1);
	std::fill(h[0][0], h[0][0] + n + 1, 1);
	for (int pre = 0, i = 1; i <= m; i++) {
		pre += a[i];
		for (int j = 0; j <= s; j++) {
			for (int k = 0; k <= j && k + pre <= sum; k++) {
				auto query = [](int *a, int l, int r) {
					return l == 0 ? a[r] : sub(a[r] - a[l - 1]);
				};
				// for (int p = 1; p <= k; p++) {
				// 	f[i][j][k] += f[i - 1][j - p][k - p];
				// }
				if (k > 0) {
					add(f[i][j][k], query(h[i - 1][j - k], 0, k - 1)); // k >= 1
				}
				// for (int p = 0; p <= a[i]; p++) {
				// 	f[i][j][k] += f[i - 1][j][k + p];
				// }
				add(f[i][j][k], query(g[i - 1][j], k, std::min(k + a[i], n)));
			}
			std::partial_sum(f[i][j], f[i][j] + n + 1, g[i][j], [](const int &x, const int &y) {
				return add(x + y);
			});
		}
		for (int j = 0; j <= s; j++) {
			for (int k = 0; k <= n; k++) {
				h[i][j][k] = add(h[i][j][k - 1] + (k + j <= n ? f[i][k + j][k] : 0));
			}
		}
	}
	int ans = 0;
	for (int i = 0; i <= s; i++) {
		add(ans, f[m][i][0]);
	}
	printf("%d\n", ans);
	return 0;
}

Submission Info

Compile Error

./Main.cpp: In function ‘int main()’:
./Main.cpp:23:7: warning: ignoring return value of ‘int scanf(const char*, ...)’, declared with attribute warn_unused_result [-Wunused-result]
   23 |  scanf("%s%d", str + 1, &s);
      |  ~~~~~^~~~~~~~~~~~~~~~~~~~~

Judge Result