結果

問題 No.899 γatheree
ユーザー fgwiebfaoishfgwiebfaoish
提出日時 2020-03-03 15:08:08
言語 C#(csc)
(csc 3.9.0)
結果
AC  
実行時間 899 ms / 2,000 ms
コード長 5,577 bytes
コンパイル時間 2,403 ms
コンパイル使用メモリ 104,828 KB
実行使用メモリ 56,752 KB
最終ジャッジ日時 2023-08-04 01:30:49
合計ジャッジ時間 21,251 ms
ジャッジサーバーID
(参考情報)
judge11 / judge12
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 61 ms
22,844 KB
testcase_01 AC 62 ms
20,828 KB
testcase_02 AC 61 ms
20,776 KB
testcase_03 AC 61 ms
20,852 KB
testcase_04 AC 61 ms
20,932 KB
testcase_05 AC 61 ms
18,780 KB
testcase_06 AC 894 ms
56,328 KB
testcase_07 AC 891 ms
54,168 KB
testcase_08 AC 898 ms
56,140 KB
testcase_09 AC 893 ms
56,188 KB
testcase_10 AC 894 ms
54,148 KB
testcase_11 AC 891 ms
54,112 KB
testcase_12 AC 888 ms
56,244 KB
testcase_13 AC 884 ms
54,088 KB
testcase_14 AC 881 ms
56,156 KB
testcase_15 AC 883 ms
54,084 KB
testcase_16 AC 878 ms
56,156 KB
testcase_17 AC 889 ms
56,148 KB
testcase_18 AC 899 ms
56,160 KB
testcase_19 AC 882 ms
56,224 KB
testcase_20 AC 899 ms
54,320 KB
testcase_21 AC 840 ms
54,700 KB
testcase_22 AC 848 ms
56,660 KB
testcase_23 AC 852 ms
56,752 KB
権限があれば一括ダウンロードができます
コンパイルメッセージ
Microsoft (R) Visual C# Compiler version 3.9.0-6.21124.20 (db94f4cc)
Copyright (C) Microsoft Corporation. All rights reserved.

ソースコード

diff #

using System;
using System.Collections.Generic;
using System.Collections;
using System.Collections.Specialized;
using System.Linq;
using System.Text;
using System.IO;
using System.Reflection;
using static System.Math;
using System.Numerics;
static class Program{
	const int mod=(int)1e9+7;
	static List<int>[] li;
	static bool[] b;
	static void Main(){
		Sc sc=new Sc();
		var s=sc.Ia;
		int n=s[0],m=n-1;
		li=new List<int>[n+1];
		b=new bool[n+1];
		for(int i=1;i<=n;i++){li[i]=new List<int>();}
		for(int i=0;i<m;i++){
			var e=sc.Ia;
			li[e[0]+1].Add(e[1]+1);
			li[e[1]+1].Add(e[0]+1);
		}
		Et et=new Et(li,1);
		St st=new St(n+1);
		var h=sc.Ia2;
		for(int i = 1;i<=n;i++) {st.Ud3(et.fa[i],et.fa[i],h[i]);}
		int q=sc.I;
		StringBuilder sb=new StringBuilder();
		for(int i = 0;i<q;i++) {
			var e=sc.I+1;
			long p=0;
			int z=et.oa[e];
			if(z!=0){
				p=st.Get(et.fa[z],et.fa[z]);
				st.Ud3(et.fa[z],et.fa[z],0);
				p+=st.Get(et.ixl[z],et.ixr[z]);
				st.Ud3(et.ixl[z],et.ixr[z],0);
				z=et.oa[z];
				if(z!=0){
					p+=st.Get(et.fa[z],et.fa[z]);
					st.Ud3(et.fa[z],et.fa[z],0);
				}
			}
			else{p=st.Get(et.fa[e],et.fa[e]);}
			if(et.ixl[e]<=et.ixr[e]){
				p+=st.Get(et.ixl[e],et.ixr[e]);
				st.Ud3(et.ixl[e],et.ixr[e],0);
			}
			if(et.ixl2[e]<=et.ixr2[e]){
				p+=st.Get(et.ixl2[e],et.ixr2[e]);
				st.Ud3(et.ixl2[e],et.ixr2[e],0);
			}
			st.Ud3(et.fa[e],et.fa[e],p);
			sb.Append(p+"\n");
		}
		Console.Write(sb);
	}
}
public class Et{
	private List<int>[] li;
	public int[] at,ixl,ixr,fa,oa,ixl2,ixr2;
	public Et(List<int>[] li,int o){
		int n=li.Length;
		this.li=li;
		at=new int[n];
		ixl=new int[n];
		ixr=new int[n];
		fa=new int[n];
		oa=new int[n];
		ixl2=new int[n];
		ixr2=new int[n];
		var qu=new Queue<int>();
		qu.Enqueue(o);
		fa[o]=1;
		at[1]=o;
		int f=2;
		while(qu.Count>0){
			var e=qu.Dequeue();
			ixl[e]=f;
			for(int i = 0;i<li[e].Count;i++) {
				if(fa[li[e][i]]==0){
					qu.Enqueue(li[e][i]);
					fa[li[e][i]]=f;
					oa[li[e][i]]=e;
					at[f]=li[e][i];
					f++;
				}
			}
			ixr[e]=f-1;
			ixl2[e]=n+1;
			ixr2[e]=-1;
		}
		for(int i = n-1;i>1;i--) {
			if(ixl[i]<=ixr[i]){
				ixl2[oa[i]]=Min(ixl2[oa[i]],ixl[i]);
				ixr2[oa[i]]=Max(ixr2[oa[i]],ixr[i]);
			}
		}
	}
}
public class St{
	public long[] dat,dat3;
	private bool[] bo3;
	private int m=1,cnt=0;
	public St(int n){
		while(m<n){m<<=1;}
		cnt=m+n-1;
		dat=new long[cnt];
		dat3=new long[cnt+1];
		bo3=new bool[cnt];
	}
	private long Fdg(int a,int b,int k,int l,int r){
		if(k>=cnt){return 0;}
		if(bo3[k]){
			dat[k]=dat3[k];
			if(2*k+1<cnt){dat3[2*k+1]=dat3[k]>>1;bo3[2*k+1]=true;}
			if(2*k+2<cnt){dat3[2*k+2]=dat3[k]>>1;bo3[2*k+2]=true;}
			bo3[k]=false;
		}
		if(r<a||b<l){return 0;}
		if(a<=l&&r<=b){return dat[k];}
		else{return Fdg(a,b,k*2+1,l,(l+r)>>1)+Fdg(a,b,k*2+2,(l+r+1)>>1,r);}
	}
	public long Get(int a,int b){return Fdg(a,b,0,0,m-1);}
	public void Ud3(int a,int b,long d){Hn3(a,b,0,0,m-1,d*m);}
	private long Hn3(int a,int b,int k,int l,int r,long d){
		if(k>=cnt){return 0;}
		if(a<=l&&r<=b){dat3[k]=d;bo3[k]=true;return d;}
		if(bo3[k]){
			dat[k]=dat3[k];
			if(2*k+1<cnt){dat3[2*k+1]=dat3[k]>>1;bo3[2*k+1]=true;}
			if(2*k+2<cnt){dat3[2*k+2]=dat3[k]>>1;bo3[2*k+2]=true;}
			bo3[k]=false;
		}
		if(r>=a&&b>=l){
			dat[k]=Hn3(a,b,k*2+1,l,(l+r)>>1,d>>1)+Hn3(a,b,k*2+2,(l+r+1)>>1,r,d>>1);
			return dat[k];
		}
		return dat[k];
	}
}
public class Sc{
	public int I{get{return int.Parse(Console.ReadLine());}}
	public long L{get{return long.Parse(Console.ReadLine());}}
	public double D{get{return double.Parse(Console.ReadLine());}}
	public string S{get{return Console.ReadLine();}}
	public int[] Ia{get{return Array.ConvertAll(Console.ReadLine().Split(),int.Parse);}}
	public long[] La{get{return Array.ConvertAll(Console.ReadLine().Split(),long.Parse);}}
	public double[] Da{get{return Array.ConvertAll(Console.ReadLine().Split(),double.Parse);}}
	public string[] Sa{get{return Console.ReadLine().Split();}}
	public object[] Oa{get{return Console.ReadLine().Split();}}
	public int[] Ia2{get{return Array.ConvertAll(("0 "+Console.ReadLine()+" 0").Split(),int.Parse);}}
	public int[] Ia3(int a){return Array.ConvertAll((a.ToString()+" "+Console.ReadLine()).Split(),int.Parse);}
	public int[] Ia3(bool a,int b,bool c,int d){return Array.ConvertAll(((a?b.ToString()+" ":"")+Console.ReadLine()+(c?" "+d.ToString():"")).Split(),int.Parse);}
	public long[] La2{get{return Array.ConvertAll(("0 "+Console.ReadLine()+" 0").Split(),long.Parse);}}
	public long[] La3(int a){return Array.ConvertAll((a.ToString()+" "+Console.ReadLine()).Split(),long.Parse);}
	public long[] La3(bool a,int b,bool c,int d){return Array.ConvertAll(((a?b.ToString()+" ":"")+Console.ReadLine()+(c?" "+d.ToString():"")).Split(),long.Parse);}
	public double[] Da2{get{return Array.ConvertAll(("0 "+Console.ReadLine()+" 0").Split(),double.Parse);}}
	public double[] Da3(int a){return Array.ConvertAll((a.ToString()+" "+Console.ReadLine()).Split(),double.Parse);}
	public double[] Da3(bool a,int b,bool c,int d){return Array.ConvertAll(((a?b.ToString()+" ":"")+Console.ReadLine()+(c?" "+d.ToString():"")).Split(),double.Parse);}
	public T[] Arr<T>(int n,Func<T> f){var a=new T[n];for(int i=0;i<n;i++){a[i]=f();}return a;}
	public T[] Arr<T>(int n,Func<int,T> f){var a=new T[n];for(int i=0;i<n;i++){a[i]=f(i);}return a;}
	public T[] Arr<T>(int n,Func<string[],T> f){var a=new T[n];for(int i=0;i<n;i++){a[i]=f(Console.ReadLine().Split());}return a;}
	public T[] Arr<T>(int n,Func<int,string[],T> f){var a=new T[n];for(int i=0;i<n;i++){a[i]=f(i,Console.ReadLine().Split());}return a;}
}
0