結果

問題 No.1054 Union add query
コンテスト
ユーザー fgwiebfaoish
提出日時 2020-05-15 23:22:23
言語 C#(csc)
(csc 3.9.0)
結果
AC  
実行時間 1,404 ms / 2,000 ms
コード長 4,640 bytes
コンパイル時間 1,305 ms
コンパイル使用メモリ 110,720 KB
実行使用メモリ 92,500 KB
最終ジャッジ日時 2024-09-19 13:34:09
合計ジャッジ時間 12,950 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 8
権限があれば一括ダウンロードができます
コンパイルメッセージ
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;
//using nint=System.Int32;
static class Program{
	const int mod=(int)1e9+7;
	const double eps=1e-11;
	static void Main(){
		Sc sc=new Sc();
		var s=sc.Ia;
		var h=new int[s[1]][];
		Uf uf=new Uf(s[0]+1);
		for(int i = 0;i<s[1];i++) {
			h[i]=sc.Ia3(false,0,true,0);
			if(h[i][0]==1){
				uf.Union(h[i][1],h[i][2]);
			}
			else if(h[i][0]==2){
				int r=uf.Root(h[i][1]);
				h[i][3]=r;
				h[i][4]=uf.l[r].n;
			}
		}
		var g=new int[s[0]+1];
		for(int i = 2;i<=s[0];i++) {uf.Union(1,i);}
		var bit=new St(s[0]+1);
		int k=0;
		StringBuilder sb=new StringBuilder();
		uf.Scan(uf.Root(1),uf.Size(1),x=>{
			g[x]=k;
			k++;
		});
		for(int i = 0;i<s[1];i++) {
			if(h[i][0]==2){
				bit.Ud2(g[h[i][3]],g[h[i][4]],h[i][2]);
			}
			else if(h[i][0]==3){
				sb.Append(bit.Get(g[h[i][1]])+"\n");
			}
		}
		Console.Write(sb);
	}
}
public class St{
	private long[] dat;
	private int m=1,cnt=0;
	public St(int n){
		while(m<n){m<<=1;}
		cnt=m+n-1;
		dat=new long[cnt];
	}
	public void Ud(int q,long b){dat[q+m-1]+=b;}
	public long Get(int q){
		long p=0;
		q+=m-1;
		while(q>=0){p+=dat[q];q=(q-1)>>1;}
		return p;
	}
	public void Ud2(int a,int b,long d){Hn(a,b,0,0,m-1,d);}
	private void Hn(int a,int b,int k,int l,int r,long d){
		if(a<=l&&r<=b){dat[k]+=d;}
		else if(l<=b&&a<=r){
			Hn(a,b,k*2+1,l,(l+r)>>1,d);
			Hn(a,b,k*2+2,(l+r+1)>>1,r,d);
		}
	}
}
public class Uf{
	public class Nd{
		public int n;
		public Nd ne;
		public Nd(int n){this.n=n;}
	}
	private int[] arr,hi;
	public Nd[] f,l;
	public int all;
	public Uf(int n){
		arr=new int[n];
		hi=new int[n];
		f=new Nd[n];
		l=new Nd[n];
		all=n;
		for(int i=0;i<n;i++){
			arr[i]=-1;
			f[i]=l[i]=new Nd(i);
		}
	}
	public bool Union(int a,int b){
		a=Root(a);
		b=Root(b);
		if(a!=b){
			if(hi[a]<hi[b]){a^=b;b^=a;a^=b;}
			arr[a]+=arr[b];
			hi[a]=Max(hi[a],hi[b]+1);
			arr[b]=a;
			all--;
			l[a].ne=f[b];
			l[a]=l[b];
			return true;
		}
		return false;
	}
	public bool Same(int a,int b){return Root(a)==Root(b);}
	public int Size(int a){return -arr[Root(a)];}
	public int Root(int a){return arr[a]<0?a:arr[a]=Root(arr[a]);}
	public void Scan(int a,int m,Action<int> fu){
		var e=f[a];
		for(int j = 0;j<m;j++) {
			fu(e.n);
			e=e.ne;
		}
	}
}
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?" 0 0":"")).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?" 0 0":"")).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