結果

問題 No.2860 Heal Slimes
コンテスト
ユーザー テナガザル
提出日時 2024-08-25 16:29:36
言語 C++23
(gcc 13.3.0 + boost 1.87.0)
結果
WA  
実行時間 -
コード長 1,508 bytes
コンパイル時間 939 ms
コンパイル使用メモリ 97,628 KB
実行使用メモリ 6,944 KB
最終ジャッジ日時 2024-08-25 16:29:50
合計ジャッジ時間 8,289 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 49 WA * 11
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>

using namespace std;

void solve()
{
  int n, k, x;
  cin >> n >> k >> x;
  vector<int> h(n), cnt(n);
  for (int i = 0; i < n; ++i) cin >> h[i];
  int lid = 0, rid = n - 1;
  for (int i = 0; (i + 1) * k < n; ++i)
  {
    int s = i * k, flag = 1;
    for (int j = 0; j < k; ++j)
    {
      if (h[s + j] != h[s])
      {
        flag = 0;
        break;
      }
    }
    if (flag) lid = (i + 1) * k;
    else break;
  }
  for (int i = 0; (i + 1) * k < n; ++i)
  {
    int s = n - 1 - i * k, flag = 1;
    for (int j = 0; j < k; ++j)
    {
      if (h[s - j] != h[s])
      {
        flag = 0;
        break;
      }
    }
    if (flag) rid = s - k;
    else break;
  }
  if (lid > rid)
  {
    cout << "Yes" << endl;
    return;
  }
  if (rid - lid + 1 < k)
  {
    cout << "No" << endl;
    return;
  }
  int mi = 1e9;
  for (int i = lid; i <= rid; ++i) mi = min(mi, h[i]);
  for (int i = 0; i < k - 1; ++i)
  {
    if (h[lid + i] < h[lid + i + 1])
    {
      cout << "No" << endl;
      return;
    }
  }
  for (int i = 0; i < k - 1; ++i)
  {
    if (h[rid - i] < h[rid - i - 1])
    {
      cout << "No" << endl;
      return;
    }
  }
  long long sum = 0;
  for (int i = 0; i < n; ++i)
  {
    int d = h[i] - mi;
    if (d % x)
    {
      cout << "No" << endl;
      return;
    }
    sum += (h[i] - mi) / x;
  }
  int g = __gcd(n, k);
  cout << (sum % g ? "No" : "Yes") << endl;
}

int main()
{
  int t;
  cin >> t;
  while (t--) solve();
}
0