結果

問題 No.1612 I hate Construct a Palindrome
ユーザー nok0nok0
提出日時 2021-06-18 00:08:42
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 511 ms / 2,000 ms
コード長 2,066 bytes
コンパイル時間 2,744 ms
コンパイル使用メモリ 212,268 KB
最終ジャッジ日時 2025-01-22 08:38:52
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 36
権限があれば一括ダウンロードができます
コンパイルメッセージ
main.cpp: In function ‘int main()’:
main.cpp:86:25: warning: ‘sc’ may be used uninitialized [-Wmaybe-uninitialized]
   86 |                 if(c != sc) {
      |                         ^~
main.cpp:74:14: note: ‘sc’ was declared here
   74 |         char sc;
      |              ^~
main.cpp:94:32: warning: ‘x’ may be used uninitialized [-Wmaybe-uninitialized]
   94 |         auto tmp = restore(s, x);
      |                                ^
main.cpp:73:20: note: ‘x’ was declared here
   73 |         int s = 0, x, y, xy_id;
      |                    ^

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
#define rep(i, x) for(int i = 0; i < (x); i++)
constexpr int inf = 1 << 30;
using T = tuple<int, int, int>;

int bit;
int main() {
	int n, m;
	cin >> n >> m;
	vector<T> edges;
	vector<vector<pair<int, int>>> g(n);

	auto is_palindrome = [](const string &s) {
		rep(i, s.size()) {
			if(s[i] != s[s.size() - 1 - i]) return false;
		}
		return true;
	};

	auto bfs = [&](int s) {
		vector dist(n, inf);
		dist[s] = 0;
		queue<int> que;
		que.push(s);
		while(que.size()) {
			int now = que.front();
			que.pop();
			for(auto [to, ignored] : g[now]) {
				if(dist[to] <= dist[now] + 1) continue;
				dist[to] = dist[now] + 1;
				que.push(to);
			}
		}
		return dist;
	};

	auto restore = [&](int s, int t) {
		vector<int> res;
		auto dist = bfs(s);
		int now = t;
		while(now != s) {
			for(auto [to, id] : g[now]) {
				if(dist[to] + 1 == dist[now]) {
					now = to;
					res.push_back(id);
					break;
				}
			}
		}
		reverse(res.begin(), res.end());
		return res;
	};

	rep(i, m) {
		int a, b;
		char c;
		cin >> a >> b >> c;
		--a, --b;
		g[a].emplace_back(b, i);
		g[b].emplace_back(a, i);
		edges.push_back({a, b, c - 'a'});
		bit |= 1 << (c - 'a');
	}
	if(__builtin_popcount(bit) == 1) {
		puts("-1");
		return 0;
	}

	//方針
	//0-s s-x x-y y-x x-(n-1)
	vector<int> res;
	int s = 0, x, y, xy_id;
	char sc;
	rep(i, m) {
		auto [u, v, ig] = edges[i];
		if(!u or !v) {
			s = max(u, v);
			res.push_back(i);
			sc = get<2>(edges[i]);
			break;
		}
	}
	rep(i, m) {
		auto [u, v, c] = edges[i];
		if(c != sc) {
			auto dist = bfs(s);
			if(dist[u] > dist[v]) swap(u, v);
			x = u, y = v, xy_id = i;
			break;
		}
	}

	auto tmp = restore(s, x);
	res.insert(res.end(), tmp.begin(), tmp.end());
	rep(i, 2) res.push_back(xy_id);
	tmp = restore(x, n - 1);
	res.insert(res.end(), tmp.begin(), tmp.end());

	string str = "";
	for(auto v : res) str.push_back(get<2>(edges[v]));
	if(is_palindrome(str)) rep(i, 2) res.push_back(res.back());

	cout << res.size() << endl;
	for(auto &v : res) cout << v + 1 << endl;

	return 0;
}
0