結果

問題 No.3426 Mod K Graph Increments (Hard)
コンテスト
ユーザー Ohuton_Racing
提出日時 2026-01-11 14:18:37
言語 C++17
(gcc 15.2.0 + boost 1.89.0)
結果
AC  
実行時間 255 ms / 2,000 ms
コード長 1,234 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 1,042 ms
コンパイル使用メモリ 104,808 KB
実行使用メモリ 15,812 KB
最終ジャッジ日時 2026-01-11 14:18:41
合計ジャッジ時間 2,962 ms
ジャッジサーバーID
（参考情報） 		judge6 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 10
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

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

using namespace std;

void solve()
{
  int n, m, k;
  cin >> n >> m >> k;
  vector<vector<int>> g(n);
  for (int i = 0; i < m; ++i)
  {
    int u, v;
    cin >> u >> v;
    --u, --v;
    g[u].push_back(v);
    g[v].push_back(u);
  }
  vector<int> b(n);
  for (int i = 0; i < n; ++i) cin >> b[i];
  vector<int> dis(n, 1e9);
  dis[0] = 0;
  auto dfs = [&](auto f, int now, int p) -> void
  {
    for (auto to : g[now])
    {
      if (to == p) continue;
      if (dis[to] > dis[now] + 1)
      {
        dis[to] = dis[now] + 1;
        f(f, to, now);
      }
    }
  };
  dfs(dfs, 0, -1);
  int isb = 1;
  for (int i = 0; i < n; ++i)
  {
    for (auto to : g[i])
    {
      if (dis[i] % 2 == dis[to] % 2) isb = 0;
    }
  }
  if (isb)
  {
    long long sum = 0;
    for (int i = 0; i < n; ++i)
    {
      if (dis[i] % 2) sum += b[i];
      else sum -= b[i];
    }
    cout << (sum % k == 0 ? "Yes" : "No") << endl;
  }
  else
  {
    long long sum[2] = {0};
    for (int i = 0; i < n; ++i) sum[dis[i] % 2] += b[i];
    int g = __gcd(k, 2);
    cout << (abs(sum[0] - sum[1]) % g == 0 ? "Yes" : "No") << endl;
  }
}

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