結果
問題 | 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; }