結果

問題 No.1612 I hate Construct a Palindrome
ユーザー risujiroh
提出日時 2021-07-21 23:27:29
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 102 ms / 2,000 ms
コード長 1,937 bytes
コンパイル時間 2,064 ms
コンパイル使用メモリ 210,260 KB
最終ジャッジ日時 2025-01-23 05:13:45
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 36
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>

int main() {
  using namespace std;
  cin.tie(nullptr)->sync_with_stdio(false);
  int n, m;
  cin >> n >> m;
  vector<int> a(m), b(m);
  vector<vector<tuple<int, int, char>>> g(n);
  string s;
  for (int id = 0; id < m; ++id) {
    char c;
    cin >> a[id] >> b[id] >> c;
    --a[id], --b[id];
    g[a[id]].emplace_back(id, b[id], c);
    g[b[id]].emplace_back(id, a[id], c);
    s += c;
  }
  auto bfs = [&](int start) {
    vector<int> que{start};
    vector<int> d(n, -1), pe(n, -1);
    d[start] = 0;
    for (int z = 0; z < int(size(que)); ++z) {
      int v = que[z];
      for (auto&& [id, u, c] : g[v])
        if (d[u] == -1) {
          d[u] = d[v] + 1;
          pe[u] = id;
          que.push_back(u);
        }
    }
    return pair{d, pe};
  };
  int x = get<1>(g[0][0]);
  auto [dx, pex] = bfs(x);
  auto [dn, pen] = bfs(n - 1);
  for (int id = 0; id < m; ++id)
    if (s[id] != get<2>(g[0][0])) {
      string t;
      vector<int> ans;
      t += get<2>(g[0][0]);
      ans.push_back(get<0>(g[0][0]));
      for (int v = a[id]; v != x;) {
        int cur_id = pex[v];
        t += s[cur_id];
        ans.push_back(cur_id);
        v ^= a[cur_id] ^ b[cur_id];
      }
      reverse(begin(t) + 1, end(t));
      reverse(begin(ans) + 1, end(ans));
      t += s[id];
      ans.push_back(id);
      for (int v = b[id]; v != n - 1;) {
        int cur_id = pen[v];
        t += s[cur_id];
        ans.push_back(cur_id);
        v ^= a[cur_id] ^ b[cur_id];
      }
      if (t != string(rbegin(t), rend(t))) {
        cout << size(ans) << '\n';
        for (auto&& e : ans) cout << e + 1 << '\n';
      } else {
        cout << size(ans) + 2 << '\n';
        for (auto&& e : ans) cout << e + 1 << '\n';
        cout << ans.back() + 1 << '\n';
        cout << ans.back() + 1 << '\n';
      }
      exit(0);
    }
  cout << "-1\n";
}
0