結果

問題 No.1017 Reiwa Sequence
ユーザー noshi91noshi91
提出日時 2020-04-03 23:05:50
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 5,310 bytes
コンパイル時間 940 ms
コンパイル使用メモリ 88,280 KB
実行使用メモリ 23,064 KB
最終ジャッジ日時 2023-09-16 04:19:28
合計ジャッジ時間 12,290 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge13
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 WA -
testcase_01 AC 1 ms
4,376 KB
testcase_02 WA -
testcase_03 WA -
testcase_04 WA -
testcase_05 WA -
testcase_06 WA -
testcase_07 WA -
testcase_08 WA -
testcase_09 WA -
testcase_10 WA -
testcase_11 WA -
testcase_12 WA -
testcase_13 WA -
testcase_14 AC 9 ms
12,684 KB
testcase_15 AC 9 ms
13,328 KB
testcase_16 AC 9 ms
13,176 KB
testcase_17 AC 9 ms
12,900 KB
testcase_18 AC 9 ms
12,392 KB
testcase_19 AC 31 ms
23,064 KB
testcase_20 AC 31 ms
22,672 KB
testcase_21 AC 29 ms
21,584 KB
testcase_22 AC 29 ms
21,988 KB
testcase_23 AC 27 ms
21,808 KB
testcase_24 AC 28 ms
21,648 KB
testcase_25 AC 26 ms
21,180 KB
testcase_26 AC 25 ms
20,796 KB
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 WA -
testcase_31 WA -
testcase_32 WA -
testcase_33 WA -
testcase_34 WA -
testcase_35 WA -
testcase_36 WA -
testcase_37 WA -
testcase_38 WA -
testcase_39 WA -
testcase_40 TLE -
testcase_41 -- -
testcase_42 -- -
testcase_43 -- -
testcase_44 -- -
testcase_45 -- -
testcase_46 -- -
testcase_47 -- -
testcase_48 -- -
testcase_49 -- -
testcase_50 -- -
testcase_51 -- -
testcase_52 -- -
testcase_53 -- -
権限があれば一括ダウンロードができます

ソースコード

diff # 

//#define NDEBUG
#include <algorithm>
#include <cstddef>
#include <cstdint>
#include <iostream>
#include <utility>
#include <vector>

namespace n91 {

using i8 = std::int_fast8_t;
using i32 = std::int_fast32_t;
using i64 = std::int_fast64_t;
using u8 = std::uint_fast8_t;
using u32 = std::uint_fast32_t;
using u64 = std::uint_fast64_t;
using isize = std::ptrdiff_t;
using usize = std::size_t;

struct rep {
  struct itr {
    usize i;
    constexpr itr(const usize i) noexcept : i(i) {}
    void operator++() noexcept { ++i; }
    constexpr usize operator*() const noexcept { return i; }
    constexpr bool operator!=(const itr x) const noexcept { return i != x.i; }
  };
  const itr f, l;
  constexpr rep(const usize f, const usize l) noexcept
      : f(std::min(f, l)), l(l) {}
  constexpr auto begin() const noexcept { return f; }
  constexpr auto end() const noexcept { return l; }
};
struct revrep {
  struct itr {
    usize i;
    constexpr itr(const usize i) noexcept : i(i) {}
    void operator++() noexcept { --i; }
    constexpr usize operator*() const noexcept { return i; }
    constexpr bool operator!=(const itr x) const noexcept { return i != x.i; }
  };
  const itr f, l;
  constexpr revrep(const usize f, const usize l) noexcept
      : f(l - 1), l(std::min(f, l) - 1) {}
  constexpr auto begin() const noexcept { return f; }
  constexpr auto end() const noexcept { return l; }
};
template <class T> auto md_vec(const usize n, const T &value) {
  return std::vector<T>(n, value);
}
template <class... Args> auto md_vec(const usize n, Args... args) {
  return std::vector<decltype(md_vec(args...))>(n, md_vec(args...));
}
template <class T> constexpr T difference(const T &a, const T &b) noexcept {
  return a < b ? b - a : a - b;
}
template <class T> void chmin(T &a, const T &b) noexcept {
  if (b < a)
    a = b;
}
template <class T> void chmax(T &a, const T &b) noexcept {
  if (a < b)
    a = b;
}
template <class F> class rec_lambda {
  F f;

public:
  rec_lambda(F &&f) : f(std::move(f)) {}
  template <class... Args> auto operator()(Args &&... args) const {
    return f(*this, std::forward<Args>(args)...);
  }
};
template <class F> auto make_rec(F &&f) { return rec_lambda<F>(std::move(f)); }
template <class T> T scan() {
  T ret;
  std::cin >> ret;
  return ret;
}

} // namespace n91

#include <deque>

namespace n91 {

void main_() {
  /*
  std::ios::sync_with_stdio(false);
  std::cin.tie(nullptr);
  //*/
  const usize n = scan<usize>();
  std::vector<u64> a(n + 1);
  for (const usize i : rep(0, n)) {
    std::cin >> a[i];
  }
  a[n] = 0;
  const usize A = *std::max_element(a.begin(), a.end());
  for (const usize i : revrep(0, n)) {
    a[i] += a[i + 1];
  }
  const auto next = [&](std::deque<usize> &rle, u64 &sum) -> bool {
    if (rle.size() == 2) {
      return false;
    }
    usize t0 = rle[0];
    usize t1 = rle[1];
    sum -= a[t0] - a[t0 + t1];
    sum += a[t0 + t1] - a[t0 + t1 + 1];
    if (rle.size() >= 4 && rle[2] == 1) {
      rle[3] += 1;
      rle.pop_front();
      rle.pop_front();
      rle.pop_front();
      rle.push_front(t0 + 1);
      if (t1 != 1) {
        sum += a[0] - a[t1 - 1];
        rle.push_front(t1 - 1);
        rle.push_front(0);
      }
    } else {
      rle.pop_front();
      rle.pop_front();
      rle[0] -= 1;
      if (rle[0] == 0) {
        rle.pop_front();
      }
      rle.push_front(1);
      rle.push_front(t0 + 1);
      if (t1 != 1) {
        sum += a[0] - a[t1 - 1];
        rle.push_front(t1 - 1);
        rle.push_front(0);
      }
    }
    return true;
  };
  std::vector<int> oc;
  u64 ans = -static_cast<u64>(1);
  for (const usize k : rep(1, n + 1)) {
    std::deque<usize> deq;
    if (k != n) {
      deq.push_front(n - k);
    }
    deq.push_front(k);
    deq.push_front(0);
    u64 sum = a[0] - a[k];
    oc.resize(A * k + 1);
    do {
      if (oc[sum] != 0) {
        ans = sum;
        break;
      }
      oc[sum] = 1;
    } while (next(deq, sum));
    if (ans != -static_cast<u64>(1)) {
      break;
    }
  }
  if (ans == -static_cast<u64>(1)) {
    std::cout << "No\n";
    return;
  }
  std::vector<int> b(n, 0);
  int state = 0;
  for (const usize k : rep(1, n + 1)) {
    std::deque<usize> deq;
    if (k != n) {
      deq.push_front(n - k);
    }
    deq.push_front(k);
    deq.push_front(0);
    u64 sum = a[0] - a[k];
    do {
      if (sum == ans) {
        if (state == 0) {
          state = 1;
          usize i = 0;
          bool f = false;
          for (const auto e : deq) {
            if (f) {
              for (const usize j : rep(i, i + e)) {
                b[j] += 1;
              }
            }
            i += e;
            f = !f;
          }
        } else if (state == 1) {
          state = 2;
          usize i = 0;
          bool f = false;
          for (const auto e : deq) {
            if (f) {
              for (const usize j : rep(i, i + e)) {
                b[j] -= 1;
              }
            }
            i += e;
            f = !f;
          }
        }
      }
    } while (next(deq, sum));
    if (state == 2) {
      break;
    }
  }
  for (const usize i : rep(0, n)) {
    if (i)
      std::cout << " ";
    std::cout << i64(b[i]) * i64(a[i] - a[i + 1]);
  }
  std::cout << "\n";
}

} // namespace n91

int main() {
  n91::main_();
  return 0;
}
0