結果

問題 No.738 平らな農地
ユーザー fgwiebfaoishfgwiebfaoish
提出日時 2019-10-08 22:34:36
言語 C#(csc)
(csc 3.9.0)
結果
AC  
実行時間 289 ms / 2,000 ms
コード長 7,244 bytes
コンパイル時間 1,004 ms
コンパイル使用メモリ 64,408 KB
実行使用メモリ 40,120 KB
最終ジャッジ日時 2023-08-07 15:22:48
合計ジャッジ時間 19,863 ms
ジャッジサーバーID
（参考情報） 		judge14 / judge13
このコードへのチャレンジ（β）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 60 ms
20,876 KB
testcase_01 AC 61 ms
20,744 KB
testcase_02 AC 60 ms
20,888 KB
testcase_03 AC 60 ms
20,892 KB
testcase_04 AC 60 ms
20,788 KB
testcase_05 AC 71 ms
21,196 KB
testcase_06 AC 71 ms
21,132 KB
testcase_07 AC 71 ms
21,200 KB
testcase_08 AC 69 ms
21,152 KB
testcase_09 AC 68 ms
19,084 KB
testcase_10 AC 68 ms
21,144 KB
testcase_11 AC 69 ms
21,140 KB
testcase_12 AC 69 ms
23,144 KB
testcase_13 AC 71 ms
21,148 KB
testcase_14 AC 69 ms
20,976 KB
testcase_15 AC 266 ms
36,384 KB
testcase_16 AC 273 ms
38,464 KB
testcase_17 AC 280 ms
37,888 KB
testcase_18 AC 280 ms
33,840 KB
testcase_19 AC 233 ms
37,412 KB
testcase_20 AC 273 ms
38,424 KB
testcase_21 AC 289 ms
36,124 KB
testcase_22 AC 283 ms
36,408 KB
testcase_23 AC 281 ms
37,872 KB
testcase_24 AC 264 ms
37,104 KB
testcase_25 AC 69 ms
21,052 KB
testcase_26 AC 69 ms
23,120 KB
testcase_27 AC 68 ms
21,072 KB
testcase_28 AC 68 ms
21,140 KB
testcase_29 AC 69 ms
21,220 KB
testcase_30 AC 70 ms
23,092 KB
testcase_31 AC 69 ms
21,040 KB
testcase_32 AC 69 ms
21,116 KB
testcase_33 AC 69 ms
21,076 KB
testcase_34 AC 69 ms
23,212 KB
testcase_35 AC 69 ms
23,092 KB
testcase_36 AC 69 ms
21,092 KB
testcase_37 AC 69 ms
23,024 KB
testcase_38 AC 69 ms
21,084 KB
testcase_39 AC 68 ms
21,076 KB
testcase_40 AC 68 ms
21,048 KB
testcase_41 AC 69 ms
23,076 KB
testcase_42 AC 70 ms
23,004 KB
testcase_43 AC 68 ms
21,004 KB
testcase_44 AC 69 ms
21,080 KB
testcase_45 AC 265 ms
37,060 KB
testcase_46 AC 267 ms
38,684 KB
testcase_47 AC 267 ms
36,848 KB
testcase_48 AC 246 ms
39,040 KB
testcase_49 AC 244 ms
39,140 KB
testcase_50 AC 243 ms
37,352 KB
testcase_51 AC 272 ms
36,880 KB
testcase_52 AC 257 ms
36,796 KB
testcase_53 AC 260 ms
34,856 KB
testcase_54 AC 272 ms
38,876 KB
testcase_55 AC 281 ms
38,768 KB
testcase_56 AC 274 ms
38,796 KB
testcase_57 AC 257 ms
38,824 KB
testcase_58 AC 260 ms
36,864 KB
testcase_59 AC 267 ms
37,316 KB
testcase_60 AC 253 ms
39,392 KB
testcase_61 AC 260 ms
39,224 KB
testcase_62 AC 252 ms
38,748 KB
testcase_63 AC 276 ms
34,880 KB
testcase_64 AC 266 ms
39,060 KB
testcase_65 AC 113 ms
30,328 KB
testcase_66 AC 115 ms
32,612 KB
testcase_67 AC 228 ms
39,252 KB
testcase_68 AC 227 ms
35,824 KB
testcase_69 AC 250 ms
37,328 KB
testcase_70 AC 238 ms
35,320 KB
testcase_71 AC 166 ms
32,808 KB
testcase_72 AC 179 ms
34,112 KB
testcase_73 AC 175 ms
38,668 KB
testcase_74 AC 181 ms
39,312 KB
testcase_75 AC 245 ms
35,208 KB
testcase_76 AC 233 ms
34,924 KB
testcase_77 AC 236 ms
34,180 KB
testcase_78 AC 249 ms
36,992 KB
testcase_79 AC 287 ms
37,532 KB
testcase_80 AC 267 ms
36,812 KB
testcase_81 AC 240 ms
35,212 KB
testcase_82 AC 232 ms
34,720 KB
testcase_83 AC 191 ms
34,900 KB
testcase_84 AC 227 ms
36,944 KB
testcase_85 AC 133 ms
39,928 KB
testcase_86 AC 127 ms
33,104 KB
testcase_87 AC 163 ms
38,160 KB
testcase_88 AC 160 ms
40,120 KB
testcase_89 AC 61 ms
22,876 KB
testcase_90 AC 60 ms
20,900 KB
testcase_91 AC 60 ms
22,916 KB
権限があれば一括ダウンロードができます

ソースコード

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 void Main(){
		Sc sc=new Sc();
		var s=sc.Ia;
		var a=sc.Ia;
		var avl=new Avl();
		for(int i = 0;i<s[1];i++) {avl.Ud(a[i],a[i]);}
		int c=avl.Ra((avl.cnt+1)>>1).n;
		long ans=(avl.Rd(c+1,mod)-avl.Rg(c+1,mod)*c)+(avl.Rg(0,c-1)*c-avl.Rd(0,c-1));
		for(int i = s[1];i<s[0];i++) {
			avl.Dl(a[i-s[1]]);
			avl.Ud(a[i],a[i]);
			c=avl.Ra((avl.cnt+1)>>1).n;
			ans=Min(ans,(avl.Rd(c+1,mod)-avl.Rg(c+1,mod)*c)+(avl.Rg(0,c-1)*c-avl.Rd(0,c-1)));
		}
		Console.WriteLine("{0}",ans);
	}
}
public class Avl{
	public class Nd{
		public int h=1,c=1;
		public readonly int n;
		public long d,ds;
		public Nd l,r;
		public Nd(int n,int d,Nd l,Nd r){this.n=n;this.d=d;ds=d;this.l=l;this.r=r;}
	}
	public Nd root,du;
	public int cnt=0;
	public Avl(){
		du=new Nd(int.MinValue,0,du,du);du.h=0;du.c=0;
		root=du;
	}
	public void Ud(int n,int d){
		Nd mn=new Nd(n,d,du,du);
		if(root==du){root=mn;}
		else{
			Fu(root,n,mn);
			if(Abs(root.l.h-root.r.h)>1){root=Rotate(root);}
		}
		cnt++;
	}
	private void Fu(Nd t,int n,Nd mn){
		t.c++;
		t.ds+=mn.d;
		if(t.n>n){
			if(t.l!=du){
				Fu(t.l,n,mn);
				if(Abs(t.l.l.h-t.l.r.h)>1){t.l=Rotate(t.l);}
			}
			else{t.l=mn;}
		}
		else{
			if(t.r!=du){
				Fu(t.r,n,mn);
				if(Abs(t.r.l.h-t.r.r.h)>1){t.r=Rotate(t.r);}
			}
			else{t.r=mn;}
		}
		t.h=Max(t.l.h,t.r.h)+1;
	}
	private Nd Rotate(Nd t){
		Nd nd=du;
		if(t.l.h>t.r.h){
			if(t.l.l.h>t.l.r.h){
				nd=t.l;t.l=t.l.r;nd.r=t;
				Ft(t);
			}
			else{
				nd=t.l.r;
				t.l.r=nd.l;nd.l=t.l;
				Ft(t.l);
				t.l=nd.r;nd.r=t;
				Ft(t);
			}
		}
		else{
			if(t.r.l.h>t.r.r.h){
				nd=t.r.l;
				t.r.l=nd.r;nd.r=t.r;
				Ft(t.r);
				t.r=nd.l;nd.l=t;
				Ft(t);
			}
			else{
				nd=t.r;t.r=t.r.l;nd.l=t;
				Ft(t);
			}
		}
		Ft(nd);
		return nd;
	}
	private void Ft(Nd t){
		t.h=Max(t.l.h,t.r.h)+1;
		t.c=t.l.c+t.r.c+1;
		t.ds=t.l.ds+t.r.ds+t.d;
	}
	public bool Dl(int n){
		Nd t=root;
		if(t.n==n){
			root=Fd2(root);
			if(cnt==0){return true;}
			else if(Abs(root.l.h-root.r.h)>1){root=Rotate(root);}
			else{Ft(root);}
			return true;
		}
		bool bo=Fd1(root,n);
		if(Abs(root.l.h-root.r.h)>1){root=Rotate(root);}
		else{Ft(root);}
		return bo;
	}
	private bool Fd1(Nd t,int n){
		if(t.n>n){
			if(t.l==du){return false;}
			else if(t.l.n!=n){
				bool bo=Fd1(t.l,n);
				if(Abs(t.l.l.h-t.l.r.h)>1){t.l=Rotate(t.l);}
				else{Ft(t.l);}
				return bo;
			}
			else{
				t.l=Fd2(t.l);
				Ft(t);
				return true;
			}
		}
		else{
			if(t.r==du){return false;}
			else if(t.r.n!=n){
				bool bo=Fd1(t.r,n);
				if(Abs(t.r.l.h-t.r.r.h)>1){t.r=Rotate(t.r);}
				else{Ft(t.r);}
				return bo;
			}
			else{
				t.r=Fd2(t.r);
				Ft(t);
				return true;
			}
		}
	}
	private Nd Fd2(Nd dn){
		cnt--;
		if(dn.l==du){return dn.r;}
		if(dn.r==du){return dn.l;}
		Nd u=dn.l;
		if(u.r==du){
			u.r=dn.r;
			Ft(u);
			return u;
		}
		u=Fd3(u,dn);
		if(Abs(dn.l.l.h-dn.l.r.h)>1){u.l=Rotate(dn.l);}
		else{Ft(dn.l);}
		Ft(u);
		return u;
	}
	private Nd Fd3(Nd u,Nd dn){
		if(u.r.r!=du){
			Nd v=Fd3(u.r,dn);
			if(Abs(u.r.l.h-u.r.r.h)>1){u.r=Rotate(u.r);}
			else{Ft(u.r);}
			return v;
		}
		else{
			Nd v=u.r;u.r=u.r.l;v.l=dn.l;v.r=dn.r;
			return v;
		}
	}
	public Nd Hs(int n){
		Nd t=root;
		if(cnt==0){return null;}
		if(t.n==n){return t;}
		Nd a=null;
		while(true){
			if(t.n>n){
				if(t.l==du){break;}
				else if(t.l.n!=n){t=t.l;}
				else{a=t.l;break;}
			}
			else{
				if(t.r==du){break;}
				else if(t.r.n!=n){t=t.r;}
				else{a=t.r;break;}
			}
		}
		return a;
	}
	public Nd Ra(int n){return cnt>=n?Fr(root,n):null;}
	private Nd Fr(Nd t,int n){
		if(t.l.c>n-1){return Fr(t.l,n);}
		if(t.l.c<n-1){return Fr(t.r,n-t.l.c-1);}
		return t;
	}
	public Nd Lb(int n){return cnt!=0?Flb(root,n):null;}
	private Nd Flb(Nd t,int n){
		if(t.n<n){
			if(t.r==du){return null;}
			return Flb(t.r,n);
		}
		if(t.n>n){
			if(t.l==du){return t;}
			Nd u=Flb(t.l,n);
			return u==null?t:u;
		}
		return t;
	}
	public Nd Ub(int n){return cnt!=0?Fub(root,n):null;}
	private Nd Fub(Nd t,int n){
		if(t.n>n){
			if(t.l==du){return null;}
			return Fub(t.l,n);
		}
		if(t.n<n){
			if(t.r==du){return t;}
			Nd u=Fub(t.r,n);
			return u==null?t:u;
		}
		return t;
	}
	public long Rg(int a,int b){return cnt!=0?Frg(root,a,b,false,false):0;}
	private int Frg(Nd t,int a,int b,bool l,bool r){
		if(t.h==0){return 0;}
		if(l&&r){return t.c;}
		if(t.n<a){return Frg(t.r,a,b,t.n>=a,r);}
		if(t.n>b){return Frg(t.l,a,b,l,t.n<=b);}
		return Frg(t.l,a,b,l,t.n<=b)+Frg(t.r,a,b,t.n>=a,r)+1;
	}
	public long Rd(int a,int b){return cnt!=0?Frd(root,a,b,false,false):0;}
	private long Frd(Nd t,int a,int b,bool l,bool r){
		if(t.h==0){return 0;}
		if(l&&r){return t.ds;}
		if(t.n<a){return Frd(t.r,a,b,t.n>=a,r);}
		if(t.n>b){return Frd(t.l,a,b,l,t.n<=b);}
		return Frd(t.l,a,b,l,t.n<=b)+Frd(t.r,a,b,t.n>=a,r)+t.d;
	}
	public void Ad(int n,int d){
		Nd t=root;
		if(cnt==0){return;}
		if(t.n==n){t.d+=d;t.ds+=d;return;}
		while(true){
			t.ds+=d;
			if(t.n>n){
				if(t.l==du){break;}
				else if(t.l.n!=n){t=t.l;}
				else{t.l.d+=d;t.l.ds+=d;break;}
			}
			else{
				if(t.r==du){break;}
				else if(t.r.n!=n){t=t.r;}
				else{t.r.d+=d;t.r.ds+=d;break;}
			}
		}
	}
}

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 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