HackerRank 20/20 Hack October 2問目 - Angry Children

Source

HackerRank 20/20 Hack October 2問目
problem statement(コンテストページ)
problem statement

問題概要

$N$ 個の整数 $X_1,X_2,\ldots,X_N$ が与えられる.
その中から $K$ 個の整数を選び,選んだ整数の最大値 $-$ 最小値を最小化したい.
選んだ整数の最大値 $-$ 最小値の最小値を求める問題.

解法

$N$ 個の整数をソートしておいて,$X_1 \leq X_2 \leq \cdots \leq X_N$ としておく.
選ぶパターンは連続する $K$ 個を選ぶものだけを考えれば良い.
最大値と最小値は範囲の右端と左端の要素なので,最大値 $-$ 最小値は簡単に計算できる.
合計で時間計算量 $O(N \log N)$.

C++によるスパゲッティなソースコード

#include<bits/stdc++.h>
using namespace std;

#define REP(i,a,b) for(i=a;i<b;i++)
#define rep(i,n) REP(i,0,n)

#define INF 1000000000

#define mygc(c) (c)=getchar_unlocked()
#define mypc(c) putchar_unlocked(c)

void reader(int *x){int k,m=0;*x=0;for(;;){mygc(k);if(k=='-'){m=1;break;}if('0'<=k&&k<='9'){*x=k-'0';break;}}for(;;){mygc(k);if(k<'0'||k>'9')break;*x=(*x)*10+k-'0';}if(m)(*x)=-(*x);}
void reader(int *x, int *y){reader(x);reader(y);}
void writer(int x, char c){int i,sz=0,m=0;char buf[10];if(x<0)m=1,x=-x;while(x)buf[sz++]=x%10,x/=10;if(!sz)buf[sz++]=0;if(m)mypc('-');while(sz--)mypc(buf[sz]+'0');mypc(c);}

int A[100010];

int main(){
  int i, N, K, res = INF;

  reader(&N, &K);
  rep(i,N) reader(A+i);
  sort(A, A+N);
  REP(i,K-1,N) res = min(res, A[i]-A[i-K+1]);
  writer(res, '\n');

  return 0;
}

Current time: 2017年11月22日00時48分12秒
Last modified: 2014年05月11日17時10分53秒 (by laycrs)
Tags: Competitive_Programming HackerRank HackerRank_oct13
トップページに戻る

Logged in as: unknown user (not login)

ログイン: