結果

問題 No.738 平らな農地
ユーザー smiken_61smiken_61
提出日時 2019-05-18 18:52:31
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 586 ms / 2,000 ms
コード長 13,392 bytes
コンパイル時間 3,018 ms
コンパイル使用メモリ 184,064 KB
実行使用メモリ 94,108 KB
最終ジャッジ日時 2023-10-17 07:40:23
合計ジャッジ時間 26,256 ms
ジャッジサーバーID
（参考情報） 		judge11 / judge12
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
4,348 KB
testcase_01 AC 2 ms
4,348 KB
testcase_02 AC 2 ms
4,348 KB
testcase_03 AC 2 ms
4,348 KB
testcase_04 AC 2 ms
4,348 KB
testcase_05 AC 9 ms
5,624 KB
testcase_06 AC 11 ms
6,116 KB
testcase_07 AC 9 ms
6,152 KB
testcase_08 AC 5 ms
4,664 KB
testcase_09 AC 3 ms
4,348 KB
testcase_10 AC 3 ms
4,348 KB
testcase_11 AC 5 ms
4,532 KB
testcase_12 AC 4 ms
4,348 KB
testcase_13 AC 9 ms
5,608 KB
testcase_14 AC 4 ms
4,352 KB
testcase_15 AC 266 ms
82,776 KB
testcase_16 AC 266 ms
83,684 KB
testcase_17 AC 313 ms
84,164 KB
testcase_18 AC 305 ms
81,644 KB
testcase_19 AC 360 ms
88,844 KB
testcase_20 AC 274 ms
84,216 KB
testcase_21 AC 349 ms
89,148 KB
testcase_22 AC 283 ms
84,784 KB
testcase_23 AC 338 ms
86,792 KB
testcase_24 AC 347 ms
84,544 KB
testcase_25 AC 4 ms
4,460 KB
testcase_26 AC 4 ms
4,464 KB
testcase_27 AC 4 ms
4,476 KB
testcase_28 AC 4 ms
4,464 KB
testcase_29 AC 4 ms
4,480 KB
testcase_30 AC 4 ms
4,452 KB
testcase_31 AC 4 ms
4,448 KB
testcase_32 AC 4 ms
4,464 KB
testcase_33 AC 4 ms
4,456 KB
testcase_34 AC 4 ms
4,440 KB
testcase_35 AC 4 ms
4,440 KB
testcase_36 AC 4 ms
4,440 KB
testcase_37 AC 4 ms
4,464 KB
testcase_38 AC 4 ms
4,368 KB
testcase_39 AC 5 ms
4,484 KB
testcase_40 AC 5 ms
4,448 KB
testcase_41 AC 4 ms
4,444 KB
testcase_42 AC 4 ms
4,440 KB
testcase_43 AC 4 ms
4,452 KB
testcase_44 AC 4 ms
4,456 KB
testcase_45 AC 252 ms
90,580 KB
testcase_46 AC 256 ms
82,244 KB
testcase_47 AC 252 ms
88,228 KB
testcase_48 AC 225 ms
85,044 KB
testcase_49 AC 221 ms
83,648 KB
testcase_50 AC 225 ms
87,668 KB
testcase_51 AC 268 ms
89,068 KB
testcase_52 AC 245 ms
84,444 KB
testcase_53 AC 250 ms
85,116 KB
testcase_54 AC 265 ms
89,880 KB
testcase_55 AC 274 ms
89,064 KB
testcase_56 AC 262 ms
86,568 KB
testcase_57 AC 242 ms
81,928 KB
testcase_58 AC 243 ms
87,056 KB
testcase_59 AC 242 ms
90,064 KB
testcase_60 AC 238 ms
89,452 KB
testcase_61 AC 244 ms
88,868 KB
testcase_62 AC 235 ms
82,504 KB
testcase_63 AC 277 ms
89,792 KB
testcase_64 AC 261 ms
88,600 KB
testcase_65 AC 497 ms
83,408 KB
testcase_66 AC 523 ms
88,448 KB
testcase_67 AC 347 ms
83,304 KB
testcase_68 AC 336 ms
85,380 KB
testcase_69 AC 524 ms
87,624 KB
testcase_70 AC 413 ms
89,216 KB
testcase_71 AC 346 ms
94,108 KB
testcase_72 AC 281 ms
84,604 KB
testcase_73 AC 244 ms
88,220 KB
testcase_74 AC 307 ms
89,724 KB
testcase_75 AC 475 ms
82,468 KB
testcase_76 AC 354 ms
85,656 KB
testcase_77 AC 542 ms
83,516 KB
testcase_78 AC 586 ms
89,436 KB
testcase_79 AC 563 ms
89,224 KB
testcase_80 AC 419 ms
86,264 KB
testcase_81 AC 491 ms
83,900 KB
testcase_82 AC 457 ms
86,856 KB
testcase_83 AC 339 ms
90,872 KB
testcase_84 AC 296 ms
90,336 KB
testcase_85 AC 565 ms
90,240 KB
testcase_86 AC 483 ms
82,200 KB
testcase_87 AC 135 ms
86,260 KB
testcase_88 AC 129 ms
82,204 KB
testcase_89 AC 2 ms
4,348 KB
testcase_90 AC 1 ms
4,348 KB
testcase_91 AC 2 ms
4,348 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

   #include <bits/stdc++.h>

//    #include <boost/multiprecision/cpp_int.hpp>
 #define int long long
 #define inf  1000000007
 #define pa pair<int,int>
 #define ll long long
 #define pal pair<double,double>
 #define ppap pair<pa,int>
  #define PI 3.14159265358979323846
  #define paa pair<int,char>
  #define  mp make_pair
  #define  pb push_back
  #define EPS (1e-10)
                                          
    int dx[8]={0,1,0,-1,1,1,-1,-1};
    int dy[8]={1,0,-1,0,-1,1,1,-1};
                                            using namespace std;
                                   			class pa3{
                                            	public:
                                            	int x;
                                   				int y,z;
                                            	pa3(int x=0,int y=0,int z=0):x(x),y(y),z(z) {}
                                            	bool operator < (const pa3 &p) const{
                                            		if(x!=p.x) return x<p.x;
                                            		if(y!=p.y) return y<p.y;
                                            		 return z<p.z;
                                            		//return x != p.x ? x<p.x: y<p.y;
                                            	}
                                   				bool operator > (const pa3 &p) const{
                                            		if(x!=p.x) return x>p.x;
                                            		if(y!=p.y) return y>p.y;
                                            		 return z>p.z;
                                            		//return x != p.x ? x<p.x: y<p.y;
                                            	}
                                            	bool operator == (const pa3 &p) const{
                                            		return x==p.x && y==p.y && z==p.z;
                                            	}
                                            		bool operator != (const pa3 &p) const{
                                            			return !( x==p.x && y==p.y && z==p.z);
                                            	}
                                            
                                            };
                                            
                                            class pa4{
                                            	public:
                                            	int x;
                                            	int y,z,w;
                                            	pa4(int x=0,int y=0,int z=0,int w=0):x(x),y(y),z(z),w(w) {}
                                            	bool operator < (const pa4 &p) const{
                                            		if(x!=p.x) return x<p.x;
                                            		if(y!=p.y) return y<p.y;
                                            		if(z!=p.z)return z<p.z;
                                            		return w<p.w;
                                            		//return x != p.x ? x<p.x: y<p.y;
                                            	}
                                            	bool operator > (const pa4 &p) const{
                                            		if(x!=p.x) return x>p.x;
                                            		if(y!=p.y) return y>p.y;
                                            		if(z!=p.z)return z>p.z;
                                            		return w>p.w;
                                            		//return x != p.x ? x<p.x: y<p.y;
                                            	}
                                            	bool operator == (const pa4 &p) const{
                                            		return x==p.x && y==p.y && z==p.z &&w==p.w;
                                            	}
                                            		
                                            
                                            };
                                            class pa2{
                                            	public:
                                            	int x,y;
                                            	pa2(int x=0,int y=0):x(x),y(y) {}
                                            	pa2 operator + (pa2 p) {return pa2(x+p.x,y+p.y);}
                                            	pa2 operator - (pa2 p) {return pa2(x-p.x,y-p.y);}
                                            	bool operator < (const pa2 &p) const{
                                            		return y != p.y ? y<p.y: x<p.x;
                                            	}
                                            	bool operator > (const pa2 &p) const{
                                            		return x != p.x ? x<p.x: y<p.y;
                                            	}
                                            	bool operator == (const pa2 &p) const{
                                            		return abs(x-p.x)==0 && abs(y-p.y)==0;
                                            	}
                                            	bool operator != (const pa2 &p) const{
                                            		return !(abs(x-p.x)==0 && abs(y-p.y)==0);
                                            	}
                                            		
                                            
                                            };
                                            

                      
                                string itos( int i ) {
                                ostringstream s ;
                                s << i ;
                                return s.str() ;
                                }
                                 
                                int gcd(int v,int b){
                                	if(v>b) return gcd(b,v);
                                	if(v==b) return b;
                                	if(b%v==0) return v;
                                	return gcd(v,b%v);
                                }
                 
                            
                                int mod;
int extgcd(int a, int b, int &x, int &y) {
    if (b == 0) {
        x = 1;
        y = 0;
        return a;
    }
    int d = extgcd(b, a%b, y, x);
    y -= a/b * x;
    return d;
}
pa operator+(const pa & l,const pa & r) {   
    return {l.first+r.first,l.second+r.second};                                    
}    
pa operator-(const pa & l,const pa & r) {   
    return {l.first-r.first,l.second-r.second};                                    
}  
                                
                int pr[1200010];
                int inv[1200010];
                
                int beki(int wa,int rr,int warukazu){
                	if(rr==0) return 1%warukazu;
                	if(rr==1) return wa%warukazu;
                	wa%=warukazu;
                	if(rr%2==1) return ((ll)beki(wa,rr-1,warukazu)*(ll)wa)%warukazu;
                	ll zx=beki(wa,rr/2,warukazu);
                	return (zx*zx)%warukazu;
                }

                
    			int comb(int nn,int rr){
    				if(rr<0 || rr>nn || nn<0) return 0;
    				int r=pr[nn]*inv[rr];
    				r%=mod;
    				r*=inv[nn-rr];
    				r%=mod;
    				return r;
    			}
                
                void gya(int ert){
                	pr[0]=1;
                	for(int i=1;i<=ert;i++){
                		pr[i]=(pr[i-1]*i)%mod;
                	}
                		inv[ert]=beki(pr[ert],mod-2,mod);
                	for(int i=ert-1;i>=0;i--){
                		inv[i]=inv[i+1]*(i+1)%mod;
                	}
                }
                
              //   cin.tie(0);
    		//	ios::sync_with_stdio(false);
    			//priority_queue<pa3,vector<pa3>,greater<pa3>> pq;            
                 //sort(ve.begin(),ve.end(),greater<int>());
//    mt19937(clock_per_sec);
  //  mt19937_64 rng(chrono::steady_clock::now().time_since_epoch().count()) ;
      
       
       
                               //----------------kokomade tenpure------------

// verify
// CF 813E
// JAG夏2018 day3D
struct BV{
	
	private:
	public:
	int N;
	vector<int> RAN;
	vector<int> SEL1;
	vector<int> SEL0;
	int k0=0,k1=0;
	vector<bool> VEC;
	void shoki1(int NN){
		
		N=NN;
		
		VEC.resize(N,false);
		RAN.resize(N+1,0);
		
		
	}
	void setbit(int i){
		VEC[i]=1;
	}
	void resetbit(int i){
		VEC[i]=0;
	}
	void shoki2(){
		for(int i=0;i<N;i++){
			if(VEC[i]){
				RAN[i]=k1;
				k1++;
			}
			else{
				RAN[i]=k0;
				k0++;
			}
		}
		SEL1.resize(k1);
		SEL0.resize(k0);
		k0=0,k1=0;
		for(int i=0;i<N;i++){
			if(VEC[i]){
				SEL1[k1]=i;
				k1++;
			}
			else{
				SEL0[k0]=i;
				k0++;
			}
		}
	}
	
	
	int rank(int it){ return RAN[it];}
	int rank0(int it){
		if(it==N) return k0;
		if(VEC[it])return it-RAN[it];
		else return RAN[it];
	}
	int rank1(int it){
		if(it==N) return k1;
		if(VEC[it]==0)return it-RAN[it];
		else return RAN[it];
	}
	int select(int ko,bool b){
		if(b){
			if(ko>=k1)return -1;
			else return SEL1[ko];
		}
		else{
			if(ko>=k0)return -1;
			else return SEL0[ko];
		}
	}
	int access(int it) {return VEC[it];}

};

struct wavlet_matrix{
	private:
	public:
	
	int N;
	
	vector<int> VEC;
	int d;
	
	vector<BV> bv;
	vector<int> zero;
	
	vector<vector<int>> rui;
	
	void shoki1(vector<int> &input,int D){
		VEC=input;
		N=input.size();
		d=D;
		bv.resize(d);
		zero.resize(d);
		rui.resize(d);
	}
	void shoki2(){
		
		for(int i=d-1;i>=0;i--){
			
			bv[i].shoki1(N);
			vector<int> mae,ato;
			for(int j=0;j<N;j++){
			
				if(VEC[j]&(1ll<<i)){
					
					ato.pb(VEC[j]);
					bv[i].setbit(j);
				}
				else{
					mae.pb(VEC[j]);
				}
			}
			
			zero[i]=(int)mae.size();
			bv[i].shoki2();
			
			mae.insert(mae.end(),ato.begin(),ato.end());
			
			rui[i]=mae;
			for(int j=1;j<N;j++)rui[i][j]+=rui[i][j-1];
			
			swap(VEC,mae);
		}
		
	}
	
	
	int access(int r){
		int ans=0;
		for(int i=d-1;i>=0;i--){
			if(bv[i].access(r)){
				ans+=(1ll<<i);
				r=zero[i]+bv[i].rank(r);
			}
			else{
				r=bv[i].rank(r);
			}
		}
		
		return ans;
	}
	
	int range_rank(int val, int l,int r){ // [l,r) にあるvalの個数
		
		
		
		for(int i=d-1;i>=0;i--){
			if(val&(1ll<<i)){
				r=zero[i]+bv[i].rank1(r);
				l=zero[i]+bv[i].rank1(l);
			}
			else{
				r=bv[i].rank0(r);
				l=bv[i].rank0(l);
			}
			
			
			if(l>=r) return 0;
		}
	
	return r-l;
	}
	
	int rank(int val,int pos){// 0,1,...,pos にあるvalの個数
		return range_rank(val,0,pos+1);
	}
	
	int select(int val,int kth){// kth 番目の valの位置 無いなら-1
		
		
		int l=0,r=N;
		for(int i=d-1;i>=0;i--){
			if(val&(1ll<<i)){
				r=zero[i]+bv[i].rank1(r);
				l=zero[i]+bv[i].rank1(l);
			}
			else{
				r=bv[i].rank0(r);
				l=bv[i].rank0(l);
			}
			
			
			if(l+kth>=r) return -1;
		}
		int pos=l+kth;
		//cout<<" "<<pos<<" "<<endl;;
		for(int i=0;i<d;i++){
			if(val&(1ll<<(i))){
			
				pos=bv[i].select(pos-zero[i],1);
			}
			else{
				pos=bv[i].select(pos,0);
			}
	//		cout<<pos<<endl;
		}
		return pos;
		
	}
	
	int x_th_min(int x,int l,int r){
		//区間[l,r)で小さいほうからk番目の値
		// 0 <= x < r-l
		
		int ans=0;
		
		
		for(int i=d-1;i>=0;i--){
			int t1=bv[i].rank0(l);
			int t2=bv[i].rank0(r);
			int z=t2-t1;
		//	cout<<x<<" "<<l<<" "<<r<<" "<<ans<<endl;
			if(z>x){
				l=t1,r=t2;
			}
			else{
				ans+=(1ll<<i);
				x-=z;
				l=zero[i]+l-t1;
				r=zero[i]+r-t2;
			}
		}
		return ans;
	}
	
	int range_min(int l,int r){
		return x_th_min(0,l,r);
	}
	
	int range_max(int l,int r){
		return x_th_min(r-l-1,l,r);
	}
	
	
	int range_count_prefix(int l,int r,int val){
		//区間[l,r)に存在する値が[0,val)な項の個数
		if(val<=0) return 0;
		if(l>=r)return 0;
		val--;
		//[0,val]に変更
		int ans=0;
		
		for(int i=d-1;i>=0;i--){
			int t1=bv[i].rank0(l);
			int t2=bv[i].rank0(r);
			
			if(val&(1ll<<i)){
				ans+=t2-t1;				
				l=zero[i]+l-t1;;
				r=zero[i]+r-t2;
			}
			else{
				l=t1;
				r=t2;
			}
		}
		ans+=r-l;
		
		return ans;
		
	}
	
	int range_count(int l,int r,int vl,int vr){
		//区間[l,r)に存在する値が[vl,vr)な項の個数
		return range_count_prefix(l,r,vr)-range_count_prefix(l,r,vl);
		
	}		
	
	
	
	int range_sum_prefix(int l,int r,int val){
		//区間[l,r)に存在する値が[0,val)な項の和
		if(val<=0) return 0;
		if(l>=r)return 0;
		val--;
		//[0,val]に変更
		int ans=0;
		
		for(int i=d-1;i>=0;i--){
			int t1=bv[i].rank0(l);
			int t2=bv[i].rank0(r);
			
			if(val&(1ll<<i)){
				ans+=(t2>0?rui[i][t2-1]:0)-(t1>0?rui[i][t1-1]:0);
				
				l=zero[i]+l-t1;;
				r=zero[i]+r-t2;
			}
			else{
				l=t1;
				r=t2;
			}
			
		}
		
		ans-=(l>0?rui[0][l-1]:0)-(r>0?rui[0][r-1]:0);
		
		return ans;
		
	}
	
	int range_sum(int l,int r,int vl,int vr){
		//区間[l,r)に存在する値が[vl,vr)な項の和
		return range_sum_prefix(l,r,vr)-range_sum_prefix(l,r,vl);
		
	}
};




 signed main(){
 	

    	       cin.tie(0);
   			ios::sync_with_stdio(false);
int n,k;
 	cin>>n>>k;
 	vector<int> vec(n,0);
 	for(int i=1;i<=n;i++)cin>>vec[i-1];
 	
 		wavlet_matrix wt2;
 	wt2.shoki1(vec,35);
 	wt2.shoki2();
 	int ans=inf*2000000000ll;
 	for(int i=0;i<n-k+1;i++){
 		
 		int med=wt2.x_th_min(k/2,i,i+k);
 	//	cout<<med<<endl;
 		
 		int sk=wt2.range_count(i,i+k,0,med);
 		int uk=wt2.range_count(i,i+k,med+1,inf);
 		
 		int sw=wt2.range_sum(i,i+k,0,med);
 		int uw=wt2.range_sum(i,i+k,med+1,inf);
 		
 		
 		int val=0;
 		
 		val+=sk*med-sw;
 		val+=uw-uk*med;
 		
 		ans=min(ans,val);
 		
 	}
 	cout<<ans<<endl;
 	
 	
 
 	return 0;

  }
0