結果
| 問題 |
No.1295 木と駒
|
| コンテスト | |
| ユーザー |
 りあん
|
| 提出日時 | 2020-11-20 23:07:51 |
| 言語 | C#(csc) (csc 3.9.0) |
| 結果 |
WA
|
| 実行時間 | - |
| コード長 | 12,113 bytes |
| コンパイル時間 | 3,435 ms |
| コンパイル使用メモリ | 122,388 KB |
| 実行使用メモリ | 85,688 KB |
| 最終ジャッジ日時 | 2024-07-23 13:44:00 |
| 合計ジャッジ時間 | 19,900 ms |
|
ジャッジサーバーID （参考情報） |
judge5 / judge1 |
（要ログイン）
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 1 |
| other | AC * 20 WA * 28 |
コンパイルメッセージ
Microsoft (R) Visual C# Compiler version 3.9.0-6.21124.20 (db94f4cc) Copyright (C) Microsoft Corporation. All rights reserved.
ソースコード
using System;
using System.Collections.Generic;
using System.Linq;
using System.IO;
using System.Threading;
using System.Text;
using System.Text.RegularExpressions;
using System.Diagnostics;
using static util;
using P = pair<int, int>;
class Program {
static void Main(string[] args) {
var sw = new StreamWriter(Console.OpenStandardOutput()) { AutoFlush = false };
var solver = new Solver(sw);
// var t = new Thread(solver.solve, 1 << 28); // 256 MB
// t.Start();
// t.Join();
solver.solve();
sw.Flush();
}
}
class Solver {
StreamWriter sw;
Scan sc;
void Prt(string a) => sw.WriteLine(a);
void Prt<T>(IEnumerable<T> a) => Prt(string.Join(" ", a));
void Prt(params object[] a) => Prt(string.Join(" ", a));
public Solver(StreamWriter sw) {
this.sw = sw;
this.sc = new Scan();
}
public void solve() {
int n = sc.Int;
var edges = new int[n - 1][];
for (int i = 0; i < n - 1; i++)
{
edges[i] = sc.IntArr.Select(x => x - 1).ToArray();
}
var reRooting = new ReRooting<node>(n, edges, node.identity, node.operate, node.operateNode);
for (int i = 0; i < n; i++)
{
Prt(reRooting.Query(i).status == 2 ? "No" : "Yes");
}
}
struct node {
// 0: can return
// 1: cannot return
// 2: cannot tour subtree
public int status;
public int maxvalue;
public int minvalue;
public int childminexceptmaxv;
public int childminonlymaxv;
public node(int st, int ma, int mi, int cme, int cmo) {
status = st;
maxvalue = ma;
minvalue = mi;
childminexceptmaxv = cme;
childminonlymaxv = cmo;
}
public static node identity = new node(0, -M, M, M, M);
public static node operate(node a, node b) {
int ma = Math.Max(a.maxvalue, b.maxvalue);
int mi = Math.Min(a.minvalue, b.minvalue);
int cme, cmo;
int st = 0;
if (a.maxvalue < b.maxvalue) {
cme = Math.Min(Math.Min(a.childminexceptmaxv, a.childminonlymaxv), b.childminexceptmaxv);
cmo = b.childminonlymaxv;
if (a.status >= 1 || b.status == 2) {
st = 2;
}
else if (b.status == 1) {
st = 1;
}
}
else {
cme = Math.Min(Math.Min(b.childminexceptmaxv, b.childminonlymaxv), a.childminexceptmaxv);
cmo = a.childminonlymaxv;
if (b.status >= 1 || a.status == 2) {
st = 2;
}
else if (a.status == 1) {
st = 1;
}
}
return new node(st, ma, mi, cme, cmo);
}
public static node operateNode(node a, int index) {
int st = a.status;
if (index > a.childminexceptmaxv) {
st = 2;
}
else if (st == 0 && index > a.childminonlymaxv) {
st = 1;
}
return new node(st, index, index, M, a.minvalue);
}
}
}
class pair<T, U> : IComparable<pair<T, U>> {
public T v1;
public U v2;
public pair() : this(default(T), default(U)) {}
public pair(T v1, U v2) { this.v1 = v1; this.v2 = v2; }
public int CompareTo(pair<T, U> a) {
int c = Comparer<T>.Default.Compare(v1, a.v1);
return c != 0 ? c : Comparer<U>.Default.Compare(v2, a.v2);
}
public override string ToString() => v1 + " " + v2;
public void Deconstruct(out T a, out U b) { a = v1; b = v2; }
}
static class util {
public static readonly int M = 1000000007;
// public static readonly int M = 998244353;
public static readonly long LM = 1L << 60;
public static readonly double eps = 1e-11;
public static void DBG(string a) => Console.Error.WriteLine(a);
public static void DBG<T>(IEnumerable<T> a) => DBG(string.Join(" ", a));
public static void DBG(params object[] a) => DBG(string.Join(" ", a));
public static void Assert(params bool[] conds) {
if (conds.Any(x => !x)) throw new Exception();
}
public static pair<T, U> make_pair<T, U>(T v1, U v2) => new pair<T, U>(v1, v2);
public static int CompareList<T>(IList<T> a, IList<T> b) where T : IComparable<T> {
for (int i = 0; i < a.Count && i < b.Count; i++)
if (a[i].CompareTo(b[i]) != 0) return a[i].CompareTo(b[i]);
return a.Count.CompareTo(b.Count);
}
public static bool inside(int i, int j, int h, int w) => i >= 0 && i < h && j >= 0 && j < w;
public static readonly int[] dd = { 0, 1, 0, -1 };
// static readonly string dstring = "RDLU";
public static IEnumerable<P> adjacents(int i, int j)
=> Enumerable.Range(0, dd.Length).Select(k => new P(i + dd[k], j + dd[k ^ 1]));
public static IEnumerable<P> adjacents(int i, int j, int h, int w)
=> adjacents(i, j).Where(p => inside(p.v1, p.v2, h, w));
public static IEnumerable<P> adjacents(this P p) => adjacents(p.v1, p.v2);
public static IEnumerable<P> adjacents(this P p, int h, int w) => adjacents(p.v1, p.v2, h, w);
public static IEnumerable<int> all_subset(this int p) {
for (int i = 0; ; i = i - p & p) {
yield return i;
if (i == p) break;
}
}
public static Dictionary<T, int> compress<T>(this IEnumerable<T> a)
=> a.Distinct().OrderBy(v => v).Select((v, i) => new { v, i }).ToDictionary(p => p.v, p => p.i);
public static Dictionary<T, int> compress<T>(params IEnumerable<T>[] a) => compress(a.SelectMany(x => x));
public static T[] inv<T>(this Dictionary<T, int> dic) {
var res = new T[dic.Count];
foreach (var item in dic) res[item.Value] = item.Key;
return res;
}
public static void swap<T>(ref T a, ref T b) where T : struct { var t = a; a = b; b = t; }
public static void swap<T>(this IList<T> a, int i, int j) where T : struct { var t = a[i]; a[i] = a[j]; a[j] = t; }
public static T[] copy<T>(this IList<T> a) {
var ret = new T[a.Count];
for (int i = 0; i < a.Count; i++) ret[i] = a[i];
return ret;
}
}
class Scan {
StreamReader sr;
public Scan() { sr = new StreamReader(Console.OpenStandardInput()); }
public Scan(string path) { sr = new StreamReader(path); }
public int Int => int.Parse(Str);
public long Long => long.Parse(Str);
public double Double => double.Parse(Str);
public string Str => ReadLine.Trim();
public string ReadLine => sr.ReadLine();
public pair<T, U> Pair<T, U>() {
T a; U b;
Multi(out a, out b);
return new pair<T, U>(a, b);
}
public P P => Pair<int, int>();
public int[] IntArr => StrArr.Select(int.Parse).ToArray();
public long[] LongArr => StrArr.Select(long.Parse).ToArray();
public double[] DoubleArr => StrArr.Select(double.Parse).ToArray();
public string[] StrArr => Str.Split(new[]{' '}, StringSplitOptions.RemoveEmptyEntries);
bool eq<T, U>() => typeof(T).Equals(typeof(U));
T ct<T, U>(U a) => (T)Convert.ChangeType(a, typeof(T));
T cv<T>(string s) => eq<T, int>() ? ct<T, int>(int.Parse(s))
: eq<T, long>() ? ct<T, long>(long.Parse(s))
: eq<T, double>() ? ct<T, double>(double.Parse(s))
: eq<T, char>() ? ct<T, char>(s[0])
: ct<T, string>(s);
public void Multi<T>(out T a) => a = cv<T>(Str);
public void Multi<T, U>(out T a, out U b) {
var ar = StrArr; a = cv<T>(ar[0]); b = cv<U>(ar[1]);
}
public void Multi<T, U, V>(out T a, out U b, out V c) {
var ar = StrArr; a = cv<T>(ar[0]); b = cv<U>(ar[1]); c = cv<V>(ar[2]);
}
public void Multi<T, U, V, W>(out T a, out U b, out V c, out W d) {
var ar = StrArr; a = cv<T>(ar[0]); b = cv<U>(ar[1]); c = cv<V>(ar[2]); d = cv<W>(ar[3]);
}
public void Multi<T, U, V, W, X>(out T a, out U b, out V c, out W d, out X e) {
var ar = StrArr; a = cv<T>(ar[0]); b = cv<U>(ar[1]); c = cv<V>(ar[2]); d = cv<W>(ar[3]); e = cv<X>(ar[4]);
}
}
class ReRooting<T>
{
public int NodeCount { get; private set; }
int[][] Adjacents;
int[][] IndexForAdjacent;
T[] Res;
T[][] DP;
T Identity;
Func<T, T, T> Operate;
Func<T, int, T> OperateNode;
public ReRooting(int nodeCount, int[][] edges, T identity, Func<T, T, T> operate, Func<T, int, T> operateNode)
{
NodeCount = nodeCount;
Identity = identity;
Operate = operate;
OperateNode = operateNode;
List<int>[] adjacents = new List<int>[nodeCount];
List<int>[] indexForAdjacents = new List<int>[nodeCount];
for (int i = 0; i < nodeCount; i++)
{
adjacents[i] = new List<int>();
indexForAdjacents[i] = new List<int>();
}
for (int i = 0; i < edges.Length; i++)
{
var edge = edges[i];
indexForAdjacents[edge[0]].Add(adjacents[edge[1]].Count);
indexForAdjacents[edge[1]].Add(adjacents[edge[0]].Count);
adjacents[edge[0]].Add(edge[1]);
adjacents[edge[1]].Add(edge[0]);
}
Adjacents = new int[nodeCount][];
IndexForAdjacent = new int[nodeCount][];
for (int i = 0; i < nodeCount; i++)
{
Adjacents[i] = adjacents[i].ToArray();
IndexForAdjacent[i] = indexForAdjacents[i].ToArray();
}
DP = new T[Adjacents.Length][];
Res = new T[Adjacents.Length];
for (int i = 0; i < Adjacents.Length; i++) DP[i] = new T[Adjacents[i].Length];
if (NodeCount > 1) Initialize();
else if (NodeCount == 1) Res[0] = OperateNode(Identity, 0);
}
public T Query(int node) => Res[node];
private void Initialize()
{
int[] parents = new int[NodeCount];
int[] order = new int[NodeCount];
#region InitOrderedTree
var index = 0;
Stack<int> stack = new Stack<int>();
stack.Push(0);
parents[0] = -1;
while (stack.Count > 0)
{
var node = stack.Pop();
order[index++] = node;
for (int i = 0; i < Adjacents[node].Length; i++)
{
var adjacent = Adjacents[node][i];
if (adjacent == parents[node]) continue;
stack.Push(adjacent);
parents[adjacent] = node;
}
}
#endregion
#region fromLeaf
for (int i = order.Length - 1; i >= 1; i--)
{
var node = order[i];
var parent = parents[node];
T accum = Identity;
int parentIndex = -1;
for (int j = 0; j < Adjacents[node].Length; j++)
{
if (Adjacents[node][j] == parent)
{
parentIndex = j;
continue;
}
accum = Operate(accum, DP[node][j]);
}
DP[parent][IndexForAdjacent[node][parentIndex]] = OperateNode(accum, node);
}
#endregion
#region toLeaf
for (int i = 0; i < order.Length; i++)
{
var node = order[i];
T accum = Identity;
T[] accumsFromTail = new T[Adjacents[node].Length];
accumsFromTail[accumsFromTail.Length - 1] = Identity;
for (int j = accumsFromTail.Length - 1; j >= 1; j--) accumsFromTail[j - 1] = Operate(DP[node][j], accumsFromTail[j]);
for (int j = 0; j < accumsFromTail.Length; j++)
{
DP[Adjacents[node][j]][IndexForAdjacent[node][j]] = OperateNode(Operate(accum, accumsFromTail[j]), node);
accum = Operate(accum, DP[node][j]);
}
Res[node] = OperateNode(accum, node);
}
#endregion
}
}