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

提出ソース

結果

問題	No.399 動的な領主
コンテスト
ユーザー	fgwiebfaoish
提出日時	2020-02-22 04:09:10
言語	C#(csc) (csc 3.9.0)
結果	AC
実行時間	620 ms / 2,000 ms
コード長	6,534 bytes
コンパイル時間	1,281 ms
コンパイル使用メモリ	109,056 KB
実行使用メモリ	68,580 KB
最終ジャッジ日時	2024-10-09 13:52:28
合計ジャッジ時間	8,198 ms
ジャッジサーバーID （参考情報）	judge1 / judge5

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

ファイルパターン	結果
other	AC * 19

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

コンパイルメッセージ

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];
		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]);
		}
		h=new int[n+1];
		var hl=new Hld(li,n,Fc1(1,1));
		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);
	}
	static int Fc1(int a,int g){
		h[a]=g;
		int c=a;
		for(int i=0;i<li[a].Count;i++){
			if(h[li[a][i]]==0){
				var d=Fc1(li[a][i],g+1);
				if(h[c]<h[d]){c=d;}
			}
		}
		return c;
	}
}
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;
	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 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=lir[da[a]];
			af--;
		}
		while(af<bf){
			p+=st[da[b]].Get(ea[b],lia[da[b]].Count-1);
			b=lir[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=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;
	}
	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 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

結果

コンパイルメッセージ

ソースコード