#941965 (C++17) No.2566 美しい整数列

提出ソース

結果

問題	No.2566 美しい整数列
ユーザー	rrrriki
提出日時	2024-01-05 18:40:36
言語	C++17 (gcc 12.3.0 + boost 1.83.0)
結果	AC
実行時間	137 ms / 2,000 ms
コード長	9,906 bytes
コンパイル時間	4,973 ms
コンパイル使用メモリ	287,828 KB
実行使用メモリ	20,480 KB
最終ジャッジ日時	2024-09-27 19:08:54
合計ジャッジ時間	8,270 ms
ジャッジサーバーID （参考情報）	judge1 / judge3

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 6,816 KB
testcase_01	AC	3 ms 6,940 KB
testcase_02	AC	2 ms 6,940 KB
testcase_03	AC	1 ms 6,940 KB
testcase_04	AC	126 ms 20,480 KB
testcase_05	AC	137 ms 20,480 KB
testcase_06	AC	64 ms 7,936 KB
testcase_07	AC	65 ms 8,064 KB
testcase_08	AC	63 ms 7,936 KB
testcase_09	AC	64 ms 7,936 KB
testcase_10	AC	64 ms 7,936 KB
testcase_11	AC	80 ms 11,008 KB
testcase_12	AC	82 ms 11,008 KB
testcase_13	AC	83 ms 10,880 KB
testcase_14	AC	82 ms 11,136 KB
testcase_15	AC	81 ms 11,008 KB
testcase_16	AC	109 ms 19,584 KB
testcase_17	AC	112 ms 19,840 KB
testcase_18	AC	129 ms 19,968 KB
testcase_19	AC	84 ms 14,336 KB
testcase_20	AC	106 ms 18,688 KB
testcase_21	AC	116 ms 19,200 KB
testcase_22	AC	73 ms 13,312 KB
testcase_23	AC	97 ms 17,152 KB

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

ソースコード

raw source code

/**
 *    author:  rrrriki
 *    created: 02.12.2023 20:22:41
 */
#define USE_ACL
#if !__INCLUDE_LEVEL__
#include __FILE__

int main() {
  cin.tie(0);
  ios_base::sync_with_stdio(false);
  ll N, M, sum_cost = 0;
  cin >> N >> M;
  vector<ll> A(N), B(M), C(N);
  for (int i = 0; i < N; i++) {
    cin >> A[i];
  }
  for (int i = 0; i < M; i++) {
    cin >> B[i];
  }
  for (int i = 0; i < N; i++) {
    cin >> C[i];
    sum_cost += C[i];
  }
  map<ll, ll> m;  // a0の値, グループのコスト
  // 総コストからグループのコストを引いたものが必要なコストとなる

  m[A[0]] = C[0];
  ll sum = 0;
  for (int i = 1; i < N; i++) {
    sum += B[(i - 1) % M];
    m[A[i] - sum] += C[i];
  }
  //out_c_pairs(m);
  ll ans = sum_cost - max_element(all(m), [](const pair<ll, ll>& p1,
                                             const pair<ll, ll>& p2) {
                        return p1.second < p2.second;
                      })->second;
  cout << ans << "\n";
  return 0;
}

#else

// clang-format off
#include <bits/stdc++.h>
#define all(x) x.begin(), x.end()
#define YES cout << "Yes\n"
#define NO cout << "No\n"
using namespace std;
#ifdef USE_ACL
#include <atcoder/all>
using namespace atcoder;
using mint = modint998244353;
//using mint = modint1000000007;
#endif
#ifdef LOCAL
#include "debug.h"
#else
#define dbg(...) 42
#endif
using ll = long long;
using vl = vector<ll>;
// コンテナの全出力 outC(A);
template <class T> void out_c(T &A, string gap=" ") {auto itr = A.begin(); if (itr != A.end()) {cout << *itr; itr++;} while (itr != A.end()) {cout << gap << *itr; itr++;}cout << "\n"; return;}
template <class T> void out_c_pairs(T &A, string gap_inside=" ", string gap_outside = " ") {auto itr = A.begin();if (itr != A.end()) {cout << itr->first << gap_inside << itr->second;itr++;}while (itr != A.end()) {cout << gap_outside << itr->first << gap_inside << itr->second;itr++;}cout << "\n";return;}
ll _pow(ll x, ll n) {if (n == 0) return 1; ll val = _pow(x, n / 2); val *= val; if (n & 1) val *= x; return val;} // べき乗
// マンハッタン距離
template <class T> T mnht(T a, T b, T c, T d) {return abs(a - c) + abs(b - d);}
// ランレングス圧縮
void rle(string s, vector<pair<char, int>> &vec){int cnt = 1; for(int i = 1; i < (int)s.size(); i++) {if(s[i] != s[i-1]){vec.emplace_back(s[i-1], cnt); cnt = 0;} cnt++;} vec.emplace_back(s.back(), cnt);}
// 素数
bool is_prime(ll x){for (ll i=2; i*i<=x; i++){if(x%i==0)return false;}return true;}
map<ll,int> prime_factor(ll n) {map<ll,int> ret; for(ll i=2; i*i <= n; i++) {while(n%i == 0) {ret[i]++; n /= i;}} if(n != 1) ret[n]=1;return ret;}
// 組み合わせ全探索
template <typename T> bool next_combination(const T first, const T last, int k) {
    const T subset = first + k;
    // empty container | k = 0 | k == n 
    if (first == last || first == subset || last == subset) {
        return false;
    }
    T src = subset;
    while (first != src) {
        src--;
        if (*src < *(last - 1)) {
            T dest = subset;
            while (*src >= *dest) {
                dest++;
            }
            iter_swap(src, dest);
            rotate(src + 1, dest + 1, last);
            rotate(subset, subset + (last - dest) - 1, last);
            return true;
        }
    }
    // restore
    rotate(first, subset, last);
    return false;
}
// グラフ

/**
 * @brief Edgeクラスはグラフのエッジ（辺）を表します。
 *
 * @tparam T エッジの重みの型（デフォルトはint）
 */
template <typename T = int> struct Edge {
  int from, to;  // エッジの始点と終点
  T cost;        // エッジの重み
  int idx;       // エッジのインデックス（オプション）

  // デフォルトコンストラクタ
  Edge() = default;

  // エッジをコストに基づいて比較するための演算子オーバーロード
  bool operator<(const Edge &other) const { return cost < other.cost; }
  bool operator>(const Edge& other) const { return cost > other.cost; }
  friend std::ostream& operator<<(std::ostream& os, const Edge& edge) { os << edge.to; return os; }
  
  // コンストラクタ
  Edge(int from, int to, T cost = 1, int idx = -1)
      : from(from), to(to), cost(cost), idx(idx) {}

  // エッジの終点をintとして取得するためのキャスト演算子
  operator int() const { return to; }
};

/**
 * @brief Graphクラスはグラフのデータ構造を表します。
 *
 * @tparam T エッジの重みの型（デフォルトはint）
 */
template <typename T = int> struct Graph {
  vector<vector<Edge<T>>> g;  // 各ノードから出ているエッジのリスト
  int es;                     // エッジの数

  // デフォルトコンストラクタ
  Graph() = default;

  // ノード数nを指定するコンストラクタ
  explicit Graph(int n) : g(n), es(0) {}

  // グラフのサイズ（ノードの数）を返す
  size_t size() const { return g.size(); }

  // 有向エッジを追加する関数
  void add_directed_edge(int from, int to, T cost = 1) {
    g[from].emplace_back(from, to, cost, es++);
  }

  // 無向エッジを追加する関数
  void add_edge(int from, int to, T cost = 1) {
    g[from].emplace_back(from, to, cost, es);
    g[to].emplace_back(to, from, cost, es++);
  }

  // エッジを読み込む関数
  // M: エッジの数, padding: インデックスのオフセット, weighted: 重み付きかどうか, directed: 有向かどうか
  void read(int M, int padding = -1, bool weighted = false,
            bool directed = false) {
    for (int i = 0; i < M; i++) {
      int a, b;
      cin >> a >> b;
      a += padding;
      b += padding;
      T c = T(1);
      if (weighted) cin >> c;
      if (directed)
        add_directed_edge(a, b, c);
      else
        add_edge(a, b, c);
    }
  }

  // 演算子オーバーロード：インデックスによるエッジのリストへのアクセス
  inline vector<Edge<T>> &operator[](const int &k) { return g[k]; }

  // 演算子オーバーロード（const版）：インデックスによるエッジのリストへのアクセス
  inline const vector<Edge<T>> &operator[](const int &k) const { return g[k]; }
};

// Edgesテンプレートの別名定義
template <typename T = int> using Edges = vector<Edge<T>>;

/**
 * @brief NQueenクラスはN-Queen問題を解くためのクラスです。
 *
 */
struct NQueen {
 public:
  explicit NQueen(int n) : N(n) { assert(n > 0);}

  bool add_queen(int x, int y) {  // クイーンを置く
    if (row.count(x) || col.count(y) || diag1.count(x + y) || diag2.count(x - y)) return false;
    assert(x < N && y < N);
    assert(x >= 0 && y >= 0);
    queens.emplace(x, y);
    row.insert(x);        // x が同じ行にある
    col.insert(y);        // y が同じ列にある
    diag1.insert(x + y);  // x + y が同じ斜めの利き筋にある
    diag2.insert(x - y);  // x - y が同じ斜めの利き筋にある
    return true;
  }

  bool remove_queen(int x, int y) {  // クイーンを取り除く
    if (!row.count(x) || !col.count(y) || !diag1.count(x + y) || !diag2.count(x - y)) return false;
    assert(x < N && y < N);
    assert(x >= 0 && y >= 0);
    queens.erase({x, y});
    if (added_queens.count({x, y})) added_queens.erase({x, y});
    row.erase(x);
    col.erase(y);
    diag1.erase(x + y);
    diag2.erase(x - y);
    return true;
  }
  // x, y にクイーンを置けるかどうか
  bool is_valid(int x, int y) { return !row.count(x) && !col.count(y) && !diag1.count(x + y) && !diag2.count(x - y);}
  // クイーンが全てのマスを利き筋に置いているかどうか
  bool is_valid() { return (int)row.size() == N;}
  bool solve(int x = 0) {  // x行目以降のクイーンを置く
    if (x == N) return true;
    if (is_valid()) return true;
    if (row.count(x)) return solve(x + 1);
    for (int y = 0; y < N; y++) {
      if (is_valid(x, y)) {
        add_queen(x, y);
        added_queens.emplace(x, y);
        if (solve(x + 1)) return true;
        added_queens.erase({x, y});
        remove_queen(x, y);
      }
    }
    return false;
  }
  int size() { return N; }  // チェス盤のサイズを返す
  friend std::ostream& operator<<(std::ostream& os, const NQueen& nqueen) {  // クイーンの位置を出力する
    for (auto [x, y] : nqueen.queens) os << x << " " << y << "\n";
    return os;
  }
  // クイーンの位置をvector<pair<int,int>>に出力する
  vector<pair<int, int> > get_queens() { return vector<pair<int, int> >(queens.begin(), queens.end());}
  // 追加したクイーンの位置をvector<pair<int,int>>に出力する
  vector<pair<int, int> > get_added_queens() { return vector<pair<int, int> >(added_queens.begin(), added_queens.end());}
  void print(char c = 'Q', char d = '.') {  // 盤面を出力する
    vector<vector<char> > board(N, vector<char>(N, d));
    for (auto [x, y] : queens) {
      board[x][y] = c;
    }
    for (auto& row : board) {
      for (auto& c : row) {
        cout << c;
      }
      cout << "\n";
    }
  }
 private:
  int N;  // チェス盤のサイズ
  unordered_set<int> row, col, diag1, diag2;  // それぞれの行、列、斜めの利き筋にクイーンがあるかどうか
  set<pair<int, int> > queens;        // クイーンの位置
  set<pair<int, int> > added_queens;  // 追加したクイーンの位置
};
// clang-format on

#endif

/*
 *******  神龜雖壽  *******
 *******  猶有竟時  *******
 *******  騰蛇乘霧  *******
 *******  終爲土灰  *******
 *******  老驥伏櫪  *******
 *******  志在千里  *******
 *******  烈士暮年  *******
 *******  壯心不已  *******
 *******  盈縮之期  *******
 *******  不但在天  *******
 *******  養怡之福  *******
 *******  可得永年  *******
 *******  幸甚至哉  *******
 *******  歌以詠志  *******
 */

yukicoder

結果

テストケース

ソースコード