#465101 (C++14) No.1017 Reiwa Sequence

提出ソース

結果

問題	No.1017 Reiwa Sequence
ユーザー	kcvlex
提出日時	2020-04-17 04:57:55
言語	C++14 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	470 ms / 2,000 ms
コード長	5,888 bytes
コンパイル時間	1,654 ms
コンパイル使用メモリ	149,528 KB
実行使用メモリ	36,288 KB
最終ジャッジ日時	2023-09-16 07:20:58
合計ジャッジ時間	16,851 ms
ジャッジサーバーID （参考情報）	judge11 / judge15

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	1 ms 4,380 KB
testcase_01	AC	2 ms 4,376 KB
testcase_02	AC	2 ms 4,376 KB
testcase_03	AC	2 ms 4,380 KB
testcase_04	AC	2 ms 4,376 KB
testcase_05	AC	2 ms 4,376 KB
testcase_06	AC	2 ms 4,376 KB
testcase_07	AC	2 ms 4,376 KB
testcase_08	AC	2 ms 4,376 KB
testcase_09	AC	2 ms 4,380 KB
testcase_10	AC	2 ms 4,380 KB
testcase_11	AC	2 ms 4,376 KB
testcase_12	AC	8 ms 4,408 KB
testcase_13	AC	2 ms 4,380 KB
testcase_14	AC	2 ms 4,376 KB
testcase_15	AC	3 ms 4,380 KB
testcase_16	AC	2 ms 4,380 KB
testcase_17	AC	7 ms 4,532 KB
testcase_18	AC	2 ms 4,376 KB
testcase_19	AC	470 ms 36,288 KB
testcase_20	AC	431 ms 36,228 KB
testcase_21	AC	450 ms 36,208 KB
testcase_22	AC	435 ms 36,264 KB
testcase_23	AC	450 ms 36,252 KB
testcase_24	AC	422 ms 36,228 KB
testcase_25	AC	384 ms 36,228 KB
testcase_26	AC	297 ms 36,232 KB
testcase_27	AC	64 ms 10,680 KB
testcase_28	AC	85 ms 12,500 KB
testcase_29	AC	36 ms 8,236 KB
testcase_30	AC	15 ms 5,516 KB
testcase_31	AC	30 ms 7,960 KB
testcase_32	AC	29 ms 7,636 KB
testcase_33	AC	2 ms 4,380 KB
testcase_34	AC	2 ms 4,376 KB
testcase_35	AC	2 ms 4,376 KB
testcase_36	AC	2 ms 4,376 KB
testcase_37	AC	2 ms 4,376 KB
testcase_38	AC	175 ms 19,920 KB
testcase_39	AC	15 ms 5,588 KB
testcase_40	AC	50 ms 11,772 KB
testcase_41	AC	47 ms 11,816 KB
testcase_42	AC	47 ms 11,564 KB
testcase_43	AC	47 ms 11,560 KB
testcase_44	AC	50 ms 11,896 KB
testcase_45	AC	48 ms 11,696 KB
testcase_46	AC	47 ms 11,484 KB
testcase_47	AC	47 ms 11,504 KB
testcase_48	AC	24 ms 6,540 KB
testcase_49	AC	88 ms 16,976 KB
testcase_50	AC	24 ms 6,492 KB
testcase_51	AC	2 ms 4,380 KB
testcase_52	AC	68 ms 11,644 KB
testcase_53	AC	422 ms 35,960 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

#include <limits>
#include <initializer_list>
#include <utility>
#include <bitset>
#include <tuple>
#include <type_traits>
#include <functional>
#include <string>
#include <array>
#include <deque>
#include <list>
#include <queue>
#include <stack>
#include <vector>
#include <map>
#include <set>
#include <unordered_map>
#include <unordered_set>
#include <iterator>
#include <algorithm>
#include <complex>
#include <random>
#include <numeric>
#include <iostream>
#include <iomanip>
#include <sstream>
#include <regex>
#include <cassert>
#include <cstddef>
#define endl codeforces
#define ALL(v) std::begin(v), std::end(v)
#define ALLR(v) std::rbegin(v), std::rend(v)
using ll = std::int64_t;
using ull = std::uint64_t;
using pii = std::pair<int, int>;
using tii = std::tuple<int, int, int>;
using pll = std::pair<ll, ll>;
using tll = std::tuple<ll, ll, ll>;
template <typename T> using vec = std::vector<T>;
template <typename T> using vvec = vec<vec<T>>;
template <typename T> const T& var_min(const T &t) { return t; }
template <typename T> const T& var_max(const T &t) { return t; }
template <typename T, typename... Tail> const T& var_min(const T &t, const Tail&... tail) { return std::min(t, var_min(tail...)); }
template <typename T, typename... Tail> const T& var_max(const T &t, const Tail&... tail) { return std::max(t, var_max(tail...)); }
template <typename T, typename... Tail> void chmin(T &t, const Tail&... tail) { t = var_min(t, tail...); }
template <typename T, typename... Tail> void chmax(T &t, const Tail&... tail) { t = var_max(t, tail...); }
template <typename T> const T& clamp(const T &t, const T &low, const T &high) { return std::max(low, std::min(high, t)); }
template <typename T> void chclamp(T &t, const T &low, const T &high) { return t = clamp(t, low, high); }
template <typename T> T make_v(T init) { return init; }
template <typename T, typename... Tail> auto make_v(T init, std::size_t s, Tail... tail) { auto v = std::move(make_v(init, tail...)); return vec<decltype(v)>(s, v); }
template <typename T, std::size_t Head, std::size_t ...Tail> struct multi_dem_array { using type = std::array<typename multi_dem_array<T, Tail...>::type, Head>; };
template <typename T, std::size_t Head> struct multi_dem_array<T, Head> { using type = std::array<T, Head>; };
template <typename T, std::size_t ...Args> using mdarray = typename multi_dem_array<T, Args...>::type;
namespace init__ { struct InitIO { InitIO() { std::cin.tie(nullptr); std::ios_base::sync_with_stdio(false); std::cout << std::fixed << std::setprecision(30); } } init_io; }

struct Solver {
    using bs_type = std::bitset<100>;
    ll n;
    vec<ll> vn, ans;
    std::map<ll, bs_type> sums;

    Solver(ll n) : n(n), vn(n), ans(n) {
        for (ll &e : vn) std::cin >> e;
    }

    bool dfs(ll idx, ll sum, bs_type used) {
        if (idx == n) return false;
        if (vn[idx] == 0) return dfs(idx + 1, sum, used);
        sum += vn[idx];
        used.set(idx, 1);
        auto ite = sums.find(sum);
        if (ite != sums.end()) {
            auto found = ite->second;
            auto band = found & used;
            found ^= band;
            used ^= band;
            for (ll i = 0; i < n; i++) {
                if (found.test(i)) ans[i] = vn[i];
                if (used.test(i)) ans[i] = -vn[i];
            }
            return true;
        }
        sums[sum] = used;
        if (dfs(idx + 1, sum, used)) return true;
        sum -= vn[idx];
        used.set(idx, 0);
        if (dfs(idx + 1, sum, used)) return true;
        return false;
    }

    bool check() {
        std::map<ll, tll> mp;
        tll err(-1, -1, -1);

        auto add_mp = [&](ll key, tll val) {
            auto ite = mp.find(key);
            if (ite != mp.end()) return ite->second;
            mp[key] = val;
            return err;
        };

        auto build = [&](tll s, tll t) {
            ll a, b, c, d, e, f;
            std::tie(a, b, c) = s;
            std::tie(d, e, f) = t;
            vec<ll> sl = { a, b, f, };
            vec<ll> tl = { c, d, e, };
            for (ll i = 0; i < 3; i++) for (ll j = i + 1; j < 3; j++) {
                if (sl[i] == sl[j]) return false;
                if (tl[i] == tl[j]) return false;
            }
            vec<ll> all = { a, b, c, d, e, f, };
            for (ll i = 0; i < sl.size(); i++) if (sl[i] == -1) sl.erase(sl.begin() + i);
            for (ll i = 0; i < tl.size(); i++) if (tl[i] == -1) tl.erase(tl.begin() + i);
            for (ll e : all) {
                auto ites = std::find(ALL(sl), e);
                auto itet = std::find(ALL(tl), e);
                if (ites != sl.end() && itet != tl.end()) {
                    sl.erase(ites);
                    tl.erase(itet);
                }
            }
            for (ll e : sl) ans[e] = vn[e];
            for (ll e : tl) ans[e] = -vn[e];
            return true;
        };

        for (ll i = 0; i < n; i++) for (ll j = i + 1; j < n; j++) {
            if (vn[i] == vn[j]) {
                ans[i] = vn[i];
                ans[j] = -vn[j];
                return true;
            }

            for (ll k = j + 1; k <= n; k++) {
                ll key = vn[i] - vn[j];
                if (k < n) key += vn[k];
                tll val(i, (k < n ? k : -1), j);
                auto res = add_mp(key, val);
                if (res != err) if (build(val, res)) return true;
            }
        }

        return dfs(0, 0, bs_type());
    }

    auto solve() {
        bool res = check();
        return std::make_pair(res, ans);
    }
};

int main() {
    ll n;
    std::cin >> n;
    Solver solver(n);
    bool ok;
    vec<ll> ans;
    std::tie(ok, ans) = std::move(solver.solve());
    if (ok) {
        std::cout << "Yes\n";
        for (ll i = 0; i < n; i++) std::cout << ans[i] << " \n"[i + 1 == n];
    } else {
        std::cout << "No\n";
    }
    return 0;
}

yukicoder

結果

テストケース

ソースコード