結果

問題 No.1391 ±1 Abs Sum
ユーザー fgwiebfaoishfgwiebfaoish
提出日時 2021-02-12 22:27:41
言語 C#(csc)
(csc 3.9.0)
結果
AC  
実行時間 194 ms / 2,000 ms
コード長 4,105 bytes
コンパイル時間 2,230 ms
コンパイル使用メモリ 108,788 KB
実行使用メモリ 49,860 KB
最終ジャッジ日時 2023-09-27 05:04:06
合計ジャッジ時間 8,570 ms
ジャッジサーバーID
（参考情報） 		judge15 / judge11
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 53 ms
19,900 KB
testcase_01 AC 54 ms
21,768 KB
testcase_02 AC 53 ms
23,868 KB
testcase_03 AC 54 ms
21,768 KB
testcase_04 AC 56 ms
23,820 KB
testcase_05 AC 56 ms
21,892 KB
testcase_06 AC 54 ms
21,796 KB
testcase_07 AC 55 ms
21,832 KB
testcase_08 AC 61 ms
24,028 KB
testcase_09 AC 60 ms
22,084 KB
testcase_10 AC 54 ms
23,960 KB
testcase_11 AC 194 ms
49,860 KB
testcase_12 AC 152 ms
45,128 KB
testcase_13 AC 177 ms
45,064 KB
testcase_14 AC 137 ms
38,080 KB
testcase_15 AC 141 ms
40,512 KB
testcase_16 AC 163 ms
43,156 KB
testcase_17 AC 151 ms
41,764 KB
testcase_18 AC 186 ms
46,776 KB
testcase_19 AC 143 ms
41,816 KB
testcase_20 AC 194 ms
45,936 KB
testcase_21 AC 142 ms
41,640 KB
testcase_22 AC 143 ms
41,680 KB
testcase_23 AC 165 ms
41,412 KB
testcase_24 AC 147 ms
40,304 KB
testcase_25 AC 152 ms
40,548 KB
testcase_26 AC 153 ms
43,116 KB
testcase_27 AC 125 ms
39,224 KB
testcase_28 AC 126 ms
37,448 KB
testcase_29 AC 151 ms
40,116 KB
testcase_30 AC 135 ms
40,756 KB
testcase_31 AC 193 ms
45,752 KB
testcase_32 AC 148 ms
39,216 KB
testcase_33 AC 152 ms
40,268 KB
testcase_34 AC 166 ms
46,328 KB
testcase_35 AC 62 ms
21,916 KB
testcase_36 AC 178 ms
47,408 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;
using System.Threading;
using System.Runtime.CompilerServices;
using System.Diagnostics;
//using nint=System.Int32;
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 b=Enumerable.Repeat(-1,s[0]).ToArray();
		for(int i = 0;i < s[1];i++) {b[i]=1;}
		if(s[1]==0){
			Console.WriteLine(Min(Fu(a[0]),Fu(a[s[0]-1])));
			return;
		}
		long Fu(int x) {
			var h=new int[s[0]];
			long p=0;
			for(int i = 0;i < s[0];i++) {h[i]=Abs(x-a[i]);}
			Array.Sort(h);
			for(int i = 0;i < s[0];i++) {p+=h[i]*b[i];}
			return p;
		}
		Bit bit=new Bit(s[0]);
		long ans=long.MaxValue>>1;
		for(int i = 0;i < s[0];i++) {bit.Ud(i,a[i]);}
		for(int i = 0,j=-1,k=s[1];i < s[0];i++) {
			while(k!=s[0]&&Abs(a[i]-a[j+1])>Abs(a[i]-a[k])){j++;k++;}
			long p=0;
			if(j!=-1){p+=bit.Get2(0,j)-(j+1L)*a[i];}
			if(k!=s[0]){p+=((long)s[0]-k)*a[i]-bit.Get2(k,s[0]-1);}
			if(j<i){p-=bit.Get2(j+1,i)-((long)i-j)*a[i];}
			if(i<k){p-=((long)k-i)*a[i]-bit.Get2(i,k-1);}
			ans=Min(ans,p);
		}
		Console.WriteLine(ans);
	}
}
public class Bit{
	private int n;
	private long[] bit;
	public Bit(int n){
		this.n=n;
		bit=new long[n+1];
	}
	public void Ud(int a,long p){a++;while(a<=n){bit[a]+=p;a+=a&-a;}}
	public long Get(int n){
		n++;
		long a=0;
		while(n>0){a+=bit[n];n-=n&-n;}
		return a;
	}
	public long Get2(int m,int n){
		long a=0,b=0;
		n++;
		while(n>m){a+=bit[n];n-=n&-n;}
		if(n==m){return a;}
		while(n!=m){b+=bit[m];m-=m&-m;}
		return a-b;
	}
	public void Clear(){bit=new long[n+1];}
}
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(string a,string b){return Array.ConvertAll((a+Console.ReadLine()+b).Split(),int.Parse);}
	public int[] Ia3(int a){return Array.ConvertAll((Console.ReadLine()+" "+a.ToString()).Split(),int.Parse);}
	public long[] La2{get{return Array.ConvertAll(("0 "+Console.ReadLine()+" 0").Split(),long.Parse);}}
	public long[] La3(string a,string b){return Array.ConvertAll((a+Console.ReadLine()+b).Split(),long.Parse);}
	public long[] La3(int a){return Array.ConvertAll((Console.ReadLine()+" "+a.ToString()).Split(),long.Parse);}
	public double[] Da2{get{return Array.ConvertAll(("0 "+Console.ReadLine()+" 0").Split(),double.Parse);}}
	public double[] Da3(string a,string b){return Array.ConvertAll((a+Console.ReadLine()+b).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;}
	public (T,T) Tp2<T>(){var s=Console.ReadLine().Split();return (Ct<T>(s[0]),Ct<T>(s[1]));}
	public (T,T,T) Tp3<T>(){var s=Console.ReadLine().Split();return (Ct<T>(s[0]),Ct<T>(s[1]),Ct<T>(s[2]));}
	public (T1,T1,T2) Tp3<T1,T2>(){var s=Console.ReadLine().Split();return (Ct<T1>(s[0]),Ct<T1>(s[1]),Ct<T2>(s[2]));}
	private T Ct<T>(string s){return (T)Convert.ChangeType(s,typeof(T));}
}
0