結果

問題 No.235 めぐるはめぐる (5)
ユーザー fgwiebfaoishfgwiebfaoish
提出日時 2020-02-22 15:51:05
言語 C#(csc)
(csc 3.9.0)
結果
AC  
実行時間 4,093 ms / 10,000 ms
コード長 7,405 bytes
コンパイル時間 1,306 ms
コンパイル使用メモリ 118,076 KB
実行使用メモリ 141,920 KB
最終ジャッジ日時 2024-10-09 16:56:28
合計ジャッジ時間 15,541 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge5
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 4,093 ms
141,920 KB
testcase_01 AC 2,297 ms
134,100 KB
testcase_02 AC 3,138 ms
139,636 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];
		var c=sc.Ia;
		var d=sc.Ia;
		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]].Add(e[1]);
			li[e[1]].Add(e[0]);
		}
		StringBuilder sb=new StringBuilder();
		var hl=new Hld(li,n,1);
		for(int i = 0;i<n;i++) {hl.Ud(i+1,c[i],d[i]);}
		var q=sc.I;
		for(int i = 0;i<q;i++) {
			var e=sc.Ia;
			if(e[0]==1){
				Console.WriteLine(hl.Get(e[1],e[2]));
				//sb.Append(hl.Get(e[1],e[2])+"\n");
			}
			else{hl.Ud2(e[1],e[2],e[3]);}
		}
		Console.Write(sb);
	}
}
public class Hld{
	const int mod=(int)1e9+7;
	private List<int>[] li;
	private List<int> lir;
	private List<List<int>> lia=new List<List<int>>();
	private List<List<int>> lib=new List<List<int>>();
	private int[] ba,da,ea,fa;
	private St[] st;
	private int n,m=0,o,r=-1;
	public Hld(List<int>[] li,int n,int o){
		this.li=li;
		this.n=n;
		this.o=o;
		ba=new int[n+1];
		da=new int[n+1];
		ea=new int[n+1];
		lir=new List<int>();
		Fu1(o);
		r=da[o];
		fa=new int[m];
		st=new St[m];
		for(int i = 0;i<m;i++) {st[i]=new St(lia[i].Count);}
		Fu2(r,0);
	}
	public void Ud(int a,long c,long d){
		st[da[a]].Ud(ea[a],c,d);
	}
	public long Get(int a,int b){
		long p=0;
		int af=fa[da[a]],bf=fa[da[b]];
		while(af>bf){
			p+=st[da[a]].Get(ea[a],lia[da[a]].Count-1);
			p%=mod;
			a=lir[da[a]];
			af--;
		}
		while(af<bf){
			p+=st[da[b]].Get(ea[b],lia[da[b]].Count-1);
			p%=mod;
			b=lir[da[b]];
			bf--;
		}
		while(da[a]!=da[b]){
			p+=st[da[a]].Get(ea[a],lia[da[a]].Count-1);
			p%=mod;
			p+=st[da[b]].Get(ea[b],lia[da[b]].Count-1);
			p%=mod;
			a=lir[da[a]];
			b=lir[da[b]];
		}
		if(ea[a]<ea[b]){p+=st[da[a]].Get(ea[a],ea[b]);}
		else{p+=st[da[a]].Get(ea[b],ea[a]);}
		return p%mod;
	}
	public void Ud2(int a,int b,long d){
		int af=fa[da[a]],bf=fa[da[b]];
		while(af>bf){
			st[da[a]].Ud2(ea[a],lia[da[a]].Count-1,d);
			a=lir[da[a]];
			af--;
		}
		while(af<bf){
			st[da[b]].Ud2(ea[b],lia[da[b]].Count-1,d);
			b=lir[da[b]];
			bf--;
		}
		while(da[a]!=da[b]){
			st[da[a]].Ud2(ea[a],lia[da[a]].Count-1,d);
			st[da[b]].Ud2(ea[b],lia[da[b]].Count-1,d);
			a=lir[da[a]];
			b=lir[da[b]];
		}
		if(ea[a]<ea[b]){st[da[a]].Ud2(ea[a],ea[b],d);}
		else{st[da[a]].Ud2(ea[b],ea[a],d);}
	}
	private void Fu2(int a,int g){
		fa[a]=g;
		for(int i=0;i<lib[a].Count;i++){Fu2(lib[a][i],g+1);}
	}
	private void Fu1(int a){
		ba[a]=1;
		da[a]=-1;
		int p=0;
		for(int i=0;i<li[a].Count;i++){
			if(ba[li[a][i]]==0){
				Fu1(li[a][i]);
				if(ba[p]<ba[li[a][i]]){p=li[a][i];}
				ba[a]+=ba[li[a][i]];
			}
		}
		if(ba[a]==1){
			lia.Add(new List<int>());
			lib.Add(new List<int>());
			lir.Add(0);
			lia[m].Add(a);
			da[a]=m;
			m++;
		}
		else{
			ea[a]=lia[da[p]].Count;
			lia[da[p]].Add(a);
			da[a]=da[p];
			if(li[a].Count>2||o==1){
				for(int i=0;i<li[a].Count;i++){
					if(da[li[a][i]]!=-1&&p!=li[a][i]){
						lir[da[li[a][i]]]=a;
						lib[da[a]].Add(da[li[a][i]]);
					}
				}
			}
		}
	}
}
public class Dt{
	const int mod=(int)1e9+7;
	public static int e=0;
	public long d1,t,c1;
	public Dt(bool b){d1=0;c1=e;t=0;}
	public Dt(long d1,long t){
		this.d1=d1;
		this.t=t;
	}
	public Dt(Dt e){d1=e.d1;c1=0;}
	public static Dt Um(Dt a,long n){
		a.d1+=a.t*n;
		a.d1%=mod;
		return a;
	}
	public static Dt Dm(Dt a,long n){
		a.c1+=n;
		a.c1%=mod;
		return a;
	}
	public static Dt Cm(Dt a,Dt b){
		var e=new Dt((a.d1+b.d1)%mod,(a.t+b.t)%mod);
		return e;
	}
}
public class St{
	public Dt[] dat;
	private int m=1,cnt;
	private Dt du;
	public St(int n){
		du=new Dt(true);
		while(m<n){m<<=1;}
		m--;
		cnt=m+n;
		dat=new Dt[cnt+1];
		dat[cnt]=du;
		for(int i=cnt-1;i>=m;i--){dat[i]=new Dt(true);}
		for(int i=m-1;i>=0;i--){dat[i]=(i<<1)+1<m+n?dat[(i<<1)+1]:du;}
	}
	public void Ud(int q,long b1,long t){
		q+=m;
		dat[q]=new Dt(b1,t);
		q=(q-1)>>1;
		while(q>=0){
			Dt o=Dt.Cm(dat[(q<<1)+1],dat[(q<<1)+2]);
			if(dat[q]==o){break;}
			dat[q]=o;
			q=(q-1)>>1;
		}
	}
	private Dt Fdg(int a,int b,int k,int l,int r){
		if(dat[k].c1!=0){Fde(k,Dt.e);}
		if(r<a||b<l){return du;}
		if(a<=l&&r<=b){return dat[k];}
		Dt p=Fdg(a,b,k*2+1,l,(l+r)>>1);
		Dt q=Fdg(a,b,k*2+2,(l+r+1)>>1,r);
		return Dt.Cm(p,q);
	}
	public long Get(int a,int b){return a>b?0:Fdg(a,b,0,0,m).d1;}
	public void Ud2(int a,int b,long d){Hn(a,b,0,0,m,d);}
	private void Hn(int a,int b,int k,int l,int r,long d){
		if(a<=l&&r<=b){Fde(k,d);return;}
		else if(dat[k].c1!=0){Fde(k,Dt.e);}
		if(r>=a&&b>=l){
			Hn(a,b,k*2+1,l,(l+r)>>1,d);
			Hn(a,b,k*2+2,(l+r+1)>>1,r,d);
			dat[k]=Dt.Cm(dat[(k<<1)+1],dat[(k<<1)+2]);
		}
	}
	private void Fde(int k,long d){
		dat[k]=Dt.Dm(dat[k],d);
		dat[k]=Dt.Um(dat[k],dat[k].c1);
		if((k<<1)+2<=cnt){dat[(k<<1)+1]=Dt.Dm(dat[(k<<1)+1],dat[k].c1);}
		if((k<<1)+2<cnt){dat[(k<<1)+2]=Dt.Dm(dat[(k<<1)+2],dat[k].c1);}
		dat[k].c1=Dt.e;
	}
}
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