提出 #60963980


ソースコード 拡げる

#include <algorithm>
#include <atcoder/all>
#include <cassert>
#include <chrono>
#include <cmath>
#include <complex>
#include <cstdint>
#include <fstream>
#include <functional>
#include <iomanip>
#include <iostream>
#include <iterator>
#include <limits>
#include <list>
#include <map>
#include <queue>
#include <random>
#include <regex>
#include <set>
#include <stack>
#include <string>
#include <tuple>
#include <unordered_map>
#include <unordered_set>
#include <utility>
#include <vector>
using namespace std;
using namespace atcoder;
using ll = long long;
using ld = long double;
using pii = pair<int, int>;
using pll = pair<ll, ll>;
using vll = vector<ll>;
using vvll = vector<vector<ll>>;
using vvvll = vector<vector<vector<ll>>>;
using vii = vector<int>;
using vvii = vector<vector<int>>;
using vvvii = vector<vector<vector<int>>>;
using vdd = vector<ld>;
using vvdd = vector<vector<ld>>;
using vvvdd = vector<vector<vector<ld>>>;
using vbb = vector<bool>;
using vvbb = vector<vector<bool>>;
using vvvbb = vector<vector<vector<bool>>>;
using vpll = vector<pll>;
using vvpll = vector<vector<pll>>;
using vvvpll = vector<vector<vector<pll>>>;
using vmm1 = vector<modint1000000007>;
using vvmm1 = vector<vector<modint1000000007>>;
using vvvmm1 = vector<vector<vector<modint1000000007>>>;
using vmm2 = vector<modint998244353>;
using vvmm2 = vector<vector<modint998244353>>;
using vvvmm2 = vector<vector<vector<modint998244353>>>;
#define pb push_back
#define mp make_pair
#define sc second
#define fr first
#define endl '\n'
#define stpr std::fixed << setprecision
#define cYES cout << "YES" << endl
#define cNO cout << "NO" << endl
#define cYes cout << "Yes" << endl
#define cNo cout << "No" << endl
#define cerr cout << -1 << endl
#define rep(i, n) for (ll i = 0; i < (n); ++i)
#define drep(i, a, b, d) for (ll i = (a); i <= (b); i += d)
#define Rep(i, a, b) for (ll i = (a); i < (b); ++i)
#define rrep(i, n) for (ll i = n - 1; i >= 0; i--)
#define drrep(i, a, b, d) for (ll i = (a); i >= (b); i -= d)
#define rRep(i, a, b) for (ll i = a; i >= b; i--)
#define crep(i) for (char i = 'a'; i <= 'z'; ++i)
#define Crep(i) for (char i = 'A'; i <= 'Z'; ++i)
#define ALL(x) (x).begin(), (x).end()
#define rALL(x) (x).rbegin(), (x).rend()
#define sort2(A, N) \
  sort(A, A + N,    \
       [](const pii &a, const pii &b) { return a.second < b.second; });
#define debug(v)      \
  cout << #v << ":";  \
  for (auto x : v) {  \
    cout << x << ' '; \
  }                   \
  cout << endl;
int ctoi(const char c) {
  if ('0' <= c && c <= '9') return (c - '0');
  return -1;
}
ll gcd(ll a, ll b) { return (b == 0 ? a : gcd(b, a % b)); }
ll lcm(ll a, ll b) { return a * b / gcd(a, b); }
ll rup(ll a, ll b) { return a + (b - a % b) % b; }
ll rdc(ll a, ll b) { return a / b * b; }
constexpr ll INF = 1000000011;
constexpr ll MOD = 1000000007;
constexpr ll MOD2 = 998244353;
constexpr ll LINF = 1001002003004005006ll;
constexpr ld EPS = 10e-15;
using mint = modint998244353;
using mint2 = modint1000000007;
template <class T, class U>
inline bool chmax(T &lhs, const U &rhs) {
  if (lhs < rhs) {
    lhs = rhs;
    return 1;
  }
  return 0;
}
template <class T, class U>
inline bool chmin(T &lhs, const U &rhs) {
  if (lhs > rhs) {
    lhs = rhs;
    return 1;
  }
  return 0;
}
template <typename T>
istream &operator>>(istream &is, vector<T> &v) {
  for (auto &&x : v) is >> x;
  return is;
}
template <typename T, typename U>
istream &operator>>(istream &is, pair<T, U> &p) {
  is >> p.first;
  is >> p.second;
  return is;
}
template <typename T, typename U>
ostream &operator<<(ostream &os, const pair<T, U> &p) {
  os << p.first << ' ' << p.second;
  return os;
}
template <class T>
ostream &operator<<(ostream &os, vector<T> &v) {
  for (auto i = begin(v); i != end(v); ++i) {
    if (i != begin(v)) os << ' ';
    os << *i;
  }
  return os;
}

int main() {
  ios::sync_with_stdio(false);
  cin.tie(0);
  ll n;cin >> n;
  vll h(n);
  rep(i,n) cin >> h[i];

    map<ll,vll> pos; // 高さと位置
    rep(i,n) pos[h[i]].pb(i);

    ll ans = 1; // 1つのビル
    for(auto p : pos) {
      auto v = p.second; // 同じ高さのビルの位置
      ll m = v.size();

      if(m == 1) continue; // ビル一つ

      // dp[i][d]:= v[i]を末尾の時の公差dの等差数列の長さ
      vector<map<ll,ll>> dp(m);
      
      rep(i,m) { // 各ビル
        rep(j,i) { // それより前のビル
          ll diff = v[i] - v[j];
          if(dp[j].count(diff)) { // v[j]が末尾でdiffの等差数列があれば延長 なければその2つ
            dp[i][diff] = dp[j][diff] + 1;
          } else {
            dp[i][diff] = 2;
          }
          chmax(ans,dp[i][diff]);
        }
      }
    }

    cout << ans << "\n";
}

提出情報

提出日時
問題 C - Illuminate Buildings
ユーザ N00b
言語 C++ 20 (gcc 12.2)
得点 350
コード長 4915 Byte
結果 AC
実行時間 1699 ms
メモリ 284632 KiB

ジャッジ結果

セット名 Sample All
得点 / 配点 0 / 0 350 / 350
結果
AC × 3
AC × 28
セット名 テストケース
Sample sample_01.txt, sample_02.txt, sample_03.txt
All random_01.txt, random_02.txt, random_03.txt, random_04.txt, random_05.txt, random_06.txt, random_07.txt, random_08.txt, random_09.txt, random_10.txt, random_11.txt, random_12.txt, random_13.txt, random_14.txt, random_15.txt, random_16.txt, random_17.txt, random_18.txt, random_19.txt, random_20.txt, random_21.txt, random_22.txt, random_23.txt, random_24.txt, random_25.txt, sample_01.txt, sample_02.txt, sample_03.txt
ケース名 結果 実行時間 メモリ
random_01.txt AC 7 ms 3640 KiB
random_02.txt AC 3 ms 3492 KiB
random_03.txt AC 2 ms 3752 KiB
random_04.txt AC 1 ms 3536 KiB
random_05.txt AC 7 ms 3636 KiB
random_06.txt AC 3 ms 3592 KiB
random_07.txt AC 2 ms 3768 KiB
random_08.txt AC 1 ms 3592 KiB
random_09.txt AC 7 ms 3732 KiB
random_10.txt AC 5 ms 3664 KiB
random_11.txt AC 2 ms 3684 KiB
random_12.txt AC 1 ms 3580 KiB
random_13.txt AC 7 ms 3728 KiB
random_14.txt AC 5 ms 3732 KiB
random_15.txt AC 3 ms 4100 KiB
random_16.txt AC 2 ms 3624 KiB
random_17.txt AC 8 ms 4056 KiB
random_18.txt AC 6 ms 4052 KiB
random_19.txt AC 2 ms 3740 KiB
random_20.txt AC 2 ms 3780 KiB
random_21.txt AC 1699 ms 284632 KiB
random_22.txt AC 1624 ms 273380 KiB
random_23.txt AC 1674 ms 284260 KiB
random_24.txt AC 2 ms 3880 KiB
random_25.txt AC 1 ms 3516 KiB
sample_01.txt AC 1 ms 3532 KiB
sample_02.txt AC 1 ms 3468 KiB
sample_03.txt AC 1 ms 3436 KiB