#628572 (C++17) No.1424 Ultrapalindrome

提出ソース

結果

問題	No.1424 Ultrapalindrome
ユーザー	👑 null
提出日時	2021-03-12 22:15:35
言語	C++17 (gcc 13.2.0 + boost 1.83.0)
結果	WA (最新) AC (最初)
実行時間	-
コード長	6,442 bytes
コンパイル時間	8,503 ms
コンパイル使用メモリ	446,596 KB
実行使用メモリ	12,664 KB
最終ジャッジ日時	2024-04-22 17:10:26
合計ジャッジ時間	10,265 ms
ジャッジサーバーID （参考情報）	judge3 / judge4

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	2 ms 5,248 KB
testcase_01	AC	1 ms 5,248 KB
testcase_02	AC	1 ms 5,248 KB
testcase_03	AC	2 ms 5,248 KB
testcase_04	AC	2 ms 5,376 KB
testcase_05	AC	1 ms 5,376 KB
testcase_06	AC	2 ms 5,376 KB
testcase_07	AC	2 ms 5,376 KB
testcase_08	AC	2 ms 5,376 KB
testcase_09	AC	45 ms 8,960 KB
testcase_10	AC	44 ms 8,960 KB
testcase_11	AC	32 ms 7,424 KB
testcase_12	AC	42 ms 8,704 KB
testcase_13	AC	12 ms 5,376 KB
testcase_14	AC	34 ms 7,808 KB
testcase_15	AC	2 ms 5,376 KB
testcase_16	AC	28 ms 6,784 KB
testcase_17	AC	33 ms 7,552 KB
testcase_18	AC	23 ms 6,144 KB
testcase_19	AC	36 ms 6,912 KB
testcase_20	AC	9 ms 5,376 KB
testcase_21	AC	28 ms 6,016 KB
testcase_22	AC	18 ms 5,376 KB
testcase_23	AC	5 ms 5,376 KB
testcase_24	AC	7 ms 5,376 KB
testcase_25	AC	7 ms 5,376 KB
testcase_26	AC	43 ms 7,552 KB
testcase_27	WA	-
testcase_28	AC	36 ms 9,344 KB
testcase_29	AC	34 ms 9,216 KB
testcase_30	AC	39 ms 12,664 KB
testcase_31	AC	40 ms 12,664 KB

権限があれば一括ダウンロードができます

ソースコード

raw source code

/*
このコード、と～おれ!
Be accepted!
∧＿∧　
（｡･ω･｡)つ━☆・*。
⊂　　 ノ 　　　・゜+.
　しーＪ　　　°。+ *´¨)
 　　　　　　　　　.· ´¸.·*´¨) ¸.·*¨)
		  　　　　　　　　　　(¸.·´ (¸.·'* ☆
*/

#include <cstdio>
#include <algorithm>
#include <string>
#include <cmath>
#include <cstring>
#include <vector>
#include <numeric>
#include <iostream>
#include <random>
#include <map>
#include <unordered_map>
#include <queue>
#include <regex>
#include <functional>
#include <complex>
#include <list>
#include <cassert>
#include <iomanip>
#include <set>
#include <stack>
#include <bitset>
#include <array>
#include <chrono>

//#pragma GCC target("arch=skylake-avx512")
//#pragma GCC target("avx2")
//#pragma GCC optimize("O3")
//#pragma GCC optimize("Ofast")
//#pragma GCC target("sse4")
//#pragma GCC optimize("unroll-loops")
//#pragma GCC target("sse,sse2,sse3,ssse3,sse4,popcnt,abm,mmx,avx,tune=native")
#define repeat(i, n, m) for(int i = n; i < (m); ++i)
#define rep(i, n) for(int i = 0; i < (n); ++i)
#define printynl(a) printf(a ? "yes\n" : "no\n")
#define printyn(a) printf(a ? "Yes\n" : "No\n")
#define printYN(a) printf(a ? "YES\n" : "NO\n")
#define printim(a) printf(a ? "possible\n" : "imposible\n")
#define printdb(a) printf("%.50lf\n", a)
#define printLdb(a) printf("%.50Lf\n", a)
#define printdbd(a) printf("%.16lf\n", a)
#define prints(s) printf("%s\n", s.c_str())
#define all(x) (x).begin(), (x).end()
#define deg_to_rad(deg) (((deg)/360.0L)*2.0L*PI)
#define rad_to_deg(rad) (((rad)/2.0L/PI)*360.0L)
#define Please return
#define AC 0
#define manhattan_dist(a, b, c, d) (abs(a - c) + abs(b - d))

using ll = long long;
using ull = unsigned long long;

constexpr int INF = 1073741823;
constexpr int MINF = -1073741823;
constexpr ll LINF = ll(4661686018427387903);
constexpr ll MOD = 1e9 + 7;
constexpr ll mod = 998244353;
constexpr long double eps = 1e-6;
const long double PI = acosl(-1.0L);

using namespace std;

void scans(string& str) {
	char c;
	str = "";
	scanf("%c", &c);
	if (c == '\n')scanf("%c", &c);
	while (c != '\n' && c != -1 && c != ' ') {
		str += c;
		scanf("%c", &c);
	}
}

void scanc(char& str) {
	char c;
	scanf("%c", &c);
	if (c == -1)return;
	while (c == '\n') {
		scanf("%c", &c);
	}
	str = c;
}

double acot(double x) {
	return PI / 2 - atan(x);
}

ll LSB(ll n) { return (n & (-n)); }

template<typename T>
inline T chmin(T& a, const T& b) {
	if (a > b)a = b;
	return a;
}

template<typename T>
inline T chmax(T& a, const T& b) {
	if (a < b)a = b;
	return a;
}

//cpp_int
#if __has_include(<boost/multiprecision/cpp_int.hpp>)
#include <boost/multiprecision/cpp_int.hpp>
#include <boost/multiprecision/cpp_dec_float.hpp>
using namespace boost::multiprecision;
#else
using cpp_int = ll;
#endif

//atcoder library
#if __has_include(<atcoder/all>)
#include <atcoder/all>
//using namespace atcoder;
#endif

/*
	random_device seed_gen;
	mt19937 engine(seed_gen());
	uniform_int_distribution dist(1, 100);
*/


/*----------------------------------------------------------------------------------*/

/*
* @title template(graph)
* @docs kyopro/docs/graph_template.md
*/

template<typename T>
struct edge {
	T cost;
	int from, to;

	edge(int from, int to) : from(from), to(to), cost(T(1)) {}
	edge(int from, int to, T cost) : from(from), to(to), cost(cost) {}
};

template<typename T = int>
struct graph {

	int n;
	bool directed, weighted;

	vector<vector<edge<T>>> g;

	graph(int n, bool directed, bool weighted) : g(n), n(n), directed(directed), weighted(weighted) {}

	void add_edge(int from, int to, T cost = T(1)) {
		g[from].emplace_back(from, to, cost);
		if (not directed) {
			g[to].emplace_back(to, from, cost);
		}
	}

	vector<edge<T>>& operator[](const int& idx) {
		return g[idx];
	}

	void read(int e, bool one_indexed) {
		int a, b, c = 1;
		while (e--) {
			scanf("%d%d", &a, &b);
			if (weighted) {
				scanf("%d", &c);
			}
			if (one_indexed)--a, --b;
			add_edge(a, b, c);
		}
	}

	void read(int e, bool one_indexed, const string& format) {
		int a, b;
		T c = T(1);
		while (e--) {
			scanf("%d%d", &a, &b);
			if (weighted) {
				scanf(format.c_str(), &c);
			}
			if (one_indexed)--a, --b;
			add_edge(a, b, c);
		}
	}

};

/*
* @title dijkstra
* @docs kyopro/docs/dijkstra.md
*/


template<typename T = int>
vector<T> dijkstra(graph<T>& g, const int& v, const int& n, const T Inf) {
	priority_queue<pair<T, int>, vector<pair<T, int>>, greater<pair<T, int>>> priq;
	vector<T> res(n);
	fill(all(res), Inf);
	priq.push({ 0, v });
	res[v] = 0;
	int top;
	while (!priq.empty()) {
		top = priq.top().second;
		T cost = priq.top().first;
		priq.pop();
		if (cost > res[top])continue;
		for (const auto& aa : g[top]) {
			if (res[top] + aa.cost >= res[aa.to])continue;
			res[aa.to] = aa.cost + res[top];
			priq.push({ res[aa.to], aa.to });
		}
	}
	return res;
}


int main() {

	int n;
	scanf("%d", &n);
	graph g(n, false, false);
	g.read(n - 1, true);
	int fst = -1, sec = -1, one = 0;
	vector<int> p;
	rep(i, n) {
		if (fst == -1 and g[i].size() == 1)fst = i;
		else if(g[i].size() == 1)sec = i;
		if (g[i].size() == 1) {
			++one;
			p.push_back(i);
		}
	}
	auto dist = dijkstra(g, fst, n, INF);
	int d = dist[sec];
	queue<pair<int, int>> que;
	vector<bool> visited(n);
	--one;
	int two = one;
	que.push({ fst, d });
	while (not que.empty()) {
		const auto [cur, cnt] = que.front();
		que.pop();
		visited[cur] = true;
		if (cnt == 0) {
			if (g[cur].size() == 1)--one;
			continue;
		}
		for (const auto& aa : g[cur])if (not visited[aa.to])que.push({ aa.to, cnt - 1 });
	}
	fill(all(visited), false);
	if (one) {
		puts("No");
		return 0;
	}
	que.push({ sec, d });
	one = two;
	while (not que.empty()) {
		const auto [cur, cnt] = que.front();
		que.pop();
		visited[cur] = true;
		if (cnt == 0) {
			if (g[cur].size() == 1)--one;
			continue;
		}
		for (const auto& aa : g[cur])if (not visited[aa.to])que.push({ aa.to, cnt - 1 });
	}
	fill(all(visited), false);
	if (one) {
		puts("No");
		return 0;
	}
	random_device seed_gen;
	mt19937 engine(seed_gen());
	shuffle(all(p), engine);
	que.push({ p.back(), d });
	one = two;
	while (not que.empty()) {
		const auto [cur, cnt] = que.front();
		que.pop();
		visited[cur] = true;
		if (cnt == 0) {
			if (g[cur].size() == 1)--one;
			continue;
		}
		for (const auto& aa : g[cur])if (not visited[aa.to])que.push({ aa.to, cnt - 1 });
	}
	printyn(!one);

	Please AC;
}

yukicoder

結果

テストケース

ソースコード