#433346 (C#(csc)) No.399 動的な領主

提出ソース

結果

問題	No.399 動的な領主
ユーザー	fgwiebfaoish
提出日時	2020-02-22 04:44:12
言語	C#(csc) (csc 3.9.0)
結果	AC
実行時間	685 ms / 2,000 ms
コード長	6,462 bytes
コンパイル時間	1,005 ms
コンパイル使用メモリ	113,752 KB
実行使用メモリ	76,220 KB
最終ジャッジ日時	2024-10-09 13:52:50
合計ジャッジ時間	8,910 ms
ジャッジサーバーID （参考情報）	judge2 / judge4

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

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	29 ms 24,236 KB
testcase_01	AC	29 ms 26,368 KB
testcase_02	AC	29 ms 24,392 KB
testcase_03	AC	29 ms 24,312 KB
testcase_04	AC	32 ms 22,452 KB
testcase_05	AC	71 ms 32,424 KB
testcase_06	AC	680 ms 57,800 KB
testcase_07	AC	679 ms 55,764 KB
testcase_08	AC	684 ms 62,676 KB
testcase_09	AC	682 ms 60,876 KB
testcase_10	AC	34 ms 24,400 KB
testcase_11	AC	71 ms 34,648 KB
testcase_12	AC	579 ms 76,220 KB
testcase_13	AC	559 ms 73,740 KB
testcase_14	AC	338 ms 64,844 KB
testcase_15	AC	374 ms 66,880 KB
testcase_16	AC	411 ms 71,980 KB
testcase_17	AC	685 ms 60,736 KB
testcase_18	AC	671 ms 62,524 KB

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

コンパイルメッセージ

Microsoft (R) Visual C# Compiler version 3.9.0-6.21124.20 (db94f4cc)
Copyright (C) Microsoft Corporation. All rights reserved.

ソースコード

raw source code

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 int[] h;
	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];
		h=new int[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]].Add(e[1]);
			li[e[1]].Add(e[0]);
		}
		int o=0;
		for(int i = 1;i<=n;i++) {
			if(li[i].Count==1){o=i;break;}
		}
		var hl=new Hld(li,n,o);
		var q=sc.I;
		for(int i = 0;i<q;i++) {
			var e=sc.Ia;
			hl.Ud2(e[0],e[1],1);
		}
		var g=new long[q+1];
		for(int i = 1;i<=n;i++) {g[hl.Get(i,i)]++;}
		long ans=0,p=0;
		for(int i = q;i>0;i--) {
			p+=g[i];
			ans+=p*i;
		}
		Console.WriteLine("{0}",ans);
	}
}
public class Hld{
	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,ga;
	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];
		ga=new int[m];
		st=new St[m];
		for(int i = 0;i<m;i++) {st[i]=new St(lia[i].Count);}
		Fu2(r,0);
		lir.Clear();
	}
	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);
			a=ga[da[a]];
			af--;
		}
		while(af<bf){
			p+=st[da[b]].Get(ea[b],lia[da[b]].Count-1);
			b=ga[da[b]];
			bf--;
		}
		while(da[a]!=da[b]){
			p+=st[da[a]].Get(ea[a],lia[da[a]].Count-1);
			p+=st[da[b]].Get(ea[b],lia[da[b]].Count-1);
			a=ga[da[a]];
			b=ga[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;
	}
	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=ga[da[a]];
			af--;
		}
		while(af<bf){
			st[da[b]].Ud2(ea[b],lia[da[b]].Count-1,d);
			b=ga[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=ga[da[a]];
			b=ga[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;
		ga[a]=lir[a];
		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 St{
	private long[] dat,dat2;
	private int m=1,cnt=0;
	public St(int n){
		while(m<n){m<<=1;}
		cnt=m+n-1;
		dat=new long[cnt];
		dat2=new long[cnt+1];
	}
	public void Ud(int q,long b){
		q+=m-1;
		dat[q]+=b;
		while(q>0){dat[(q-1)>>1]=dat[(q-1)>>1]+b;q=(q-1)>>1;}
	}
	private long Fdg(int a,int b,int k,int l,int r){
		if(dat2[k]!=0){
			dat[k]+=dat2[k];
			if(2*k+1<cnt){dat2[2*k+1]+=dat2[k]>>1;}
			if(2*k+2<cnt){dat2[2*k+2]+=dat2[k]>>1;}
			dat2[k]=0;
		}
		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 Ud2(int a,int b,long d){Hn(a,b,0,0,m-1,d*m);}
	private long Hn(int a,int b,int k,int l,int r,long d){
		if(a<=l&&r<=b){dat2[k]+=d;return d;}
		else if(r>=a&&b>=l){
			long q=Hn(a,b,k*2+1,l,(l+r)>>1,d>>1)+Hn(a,b,k*2+2,(l+r+1)>>1,r,d>>1);
			dat[k]+=q;
			return q;
		}
		return 0;
	}
}
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;}
}

yukicoder

結果

テストケース

コンパイルメッセージ

ソースコード