結果

問題 No.1977 Extracting at Constant Intervals
コンテスト
ユーザー SSRS
提出日時 2022-06-10 21:50:07
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 137 ms / 2,000 ms
コード長 1,349 bytes
コンパイル時間 2,081 ms
コンパイル使用メモリ 176,672 KB
実行使用メモリ 63,804 KB
最終ジャッジ日時 2024-10-09 09:41:21
合計ジャッジ時間 5,034 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 31
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
const int LOG = 50;
const long long INF = 2000000000000000000;
int main(){
  int N, M;
  long long L;
  cin >> N >> M >> L;
  vector<int> A(N);
  for (int i = 0; i < N; i++){
    cin >> A[i];
  }
  vector<vector<int>> nxt(LOG, vector<int>(N, 0));
  vector<vector<long long>> sum(LOG, vector<long long>(N, 0));
  for (int i = 0; i < N; i++){
    nxt[0][i] = (i + L) % N;
    sum[0][i] = A[i];
  }
  for (int i = 0; i < LOG - 1; i++){
    for (int j = 0; j < N; j++){
      nxt[i + 1][j] = nxt[i][nxt[i][j]];
      sum[i + 1][j] = sum[i][j] + sum[i][nxt[i][j]];
    }
  }
  vector<long long> a(N, 0), b(N, 0);
  for (int i = 0; i < N; i++){
    long long R = (long long) N * M / L;
    int p = i;
    for (int j = 0; j < LOG; j++){
      if ((R >> j & 1) == 1){
        a[i] += sum[j][p];
        p = nxt[j][p];
      }
    }
  }
  for (int i = 0; i < N; i++){
    long long R = (long long) N * M / L + 1;
    int p = i;
    for (int j = 0; j < LOG; j++){
      if ((R >> j & 1) == 1){
        b[i] += sum[j][p];
        p = nxt[j][p];
      }
    }
  }
  long long r = (long long) M * N % L;
  long long ans = -INF;
  for (int i = 0; i < min(r, (long long) N); i++){
    ans = max(ans, b[i]);
  }
  for (long long i = max(r, L - N); i < L; i++){
    ans = max(ans, a[i % N]);
  }
  cout << ans << endl;
}
0