結果
| 問題 |
No.899 γatheree
|
| コンテスト | |
| ユーザー |
fgwiebfaoish
|
| 提出日時 | 2020-03-03 14:35:46 |
| 言語 | C#(csc) (csc 3.9.0) |
| 結果 |
WA
|
| 実行時間 | - |
| コード長 | 5,446 bytes |
| コンパイル時間 | 3,972 ms |
| コンパイル使用メモリ | 119,108 KB |
| 実行使用メモリ | 58,432 KB |
| 最終ジャッジ日時 | 2024-10-13 22:31:59 |
| 合計ジャッジ時間 | 17,969 ms |
|
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 1 |
| other | WA * 23 |
コンパイルメッセージ
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.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.Ia;
for(int i = 0;i<n;i++) {st.Ud3(et.fa[i+1],et.fa[i+1],h[i]);}
int q=sc.I;
StringBuilder sb=new StringBuilder();
for(int i = 0;i<q;i++) {
var e=sc.I;
e++;
long p=0;
int z=et.oa[e];
if(z!=0){
p=st.Get(z,z);
st.Ud3(z,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(z,z);
st.Ud3(z,z,0);
}
}
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;
}
for(int i = n-1;i>1;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;}
}
fgwiebfaoish