結果
問題 | No.1694 ZerOne |
ユーザー |
![]() |
提出日時 | 2021-10-01 22:38:49 |
言語 | C++23 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 574 ms / 2,000 ms |
コード長 | 1,766 bytes |
コンパイル時間 | 2,591 ms |
コンパイル使用メモリ | 252,656 KB |
実行使用メモリ | 5,632 KB |
最終ジャッジ日時 | 2024-07-19 14:42:34 |
合計ジャッジ時間 | 7,889 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 31 |
ソースコード
#include <bits/stdc++.h>using i32 = std::int32_t;using u32 = std::uint32_t;using i64 = std::int64_t;using u64 = std::uint64_t;using isize = std::ptrdiff_t;using usize = std::size_t;class rep {struct Iter {usize itr;constexpr Iter(const usize pos) noexcept : itr(pos) {}constexpr void operator++() noexcept { ++itr; }constexpr bool operator!=(const Iter& other) const noexcept { return itr != other.itr; }constexpr usize operator*() const noexcept { return itr; }};const Iter first, last;public:explicit constexpr rep(const usize first, const usize last) noexcept : first(first), last(std::max(first, last)) {}constexpr Iter begin() const noexcept { return first; }constexpr Iter end() const noexcept { return last; }};template <class T> using Vec = std::vector<T>;void main_() {std::string S;std::cin >> S;usize cnt = 0, sum = 0;for (const auto i : rep(0, S.size())) {if (S[i] == '1') {cnt += 1;sum += i;}}std::map<std::pair<usize, usize>, u64> dp;dp[{0, 0}] += 1;while (cnt--) {std::map<std::pair<usize, usize>, u64> next;for (const auto& [p, c] : dp) {const auto& [n, s] = p;for (const auto i : rep(n, S.size())) {if (s + i <= sum) {next[{i + 1, s + i}] += c;}}}dp = std::move(next);}u64 ans = 0;for (const auto& [p, c] : dp) {if (p.second == sum) {ans += c;}}std::cout << ans << '\n';}int main() {std::ios_base::sync_with_stdio(false);std::cin.tie(nullptr);main_();return 0;}