HackerRank 101 Hack July'14 2問目 - Bus Station

Source

HackerRank 101 Hack July'14 2問目
problem statement(コンテストページ)
problem statement

問題概要

バス停に並んでいる $N$ 組のグループが居る.
$k$ 番目のグループは $A_k$ 人からなる.
バスは全て $k$ 人乗りとして,以下の条件1~3.を満たさなければいけない.

  1. 同じグループは分割してはいけない.
  2. 毎回ちょうど $k$ 人乗らなければいけない.
  3. バス停に並んでいる順番に乗らなければいけない.
  4. 何回か運行し,今バス停に並んでいる人を全員乗せなければいけない.

考えうる $k$ を全て列挙する問題.

解法

$k$ を決めれば,それが条件を満たすかどうかは,前からシミュレーションすれば $O(N)$ で判定可能.
毎回 $k$ 人乗せるので,$S = A_1+A_2+\cdots+A_N$ とすれば,$k$ は $S$ の約数でなければいけない.
$S$ は高々 $10^9$ で約数はそんなに多くないので,全部試しても間に合う.
(実際には,多くの $k$ に対して,すぐに条件を満たさないことがわかって,判定に $O(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 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 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 N, A[110000];

int res[100000], ress;
int sum;

int check(int k){
  int i, j;
  int now;

  now = 0;
  rep(i,N){
    now += A[i];
    if(now > k) return 0;
    if(now==k) now = 0;
  }

  return 1;
}

int main(){
  int i, j, k;

  reader(&N);
  rep(i,N) reader(A+i);

  sum = 0;
  rep(i,N) sum += A[i];

  ress = 0;
  for(i=1;i*i<=sum;i++) if(sum%i==0){
    if(check(i)) res[ress++] = i;
    if(i!=sum/i && check(sum/i)) res[ress++] = sum/i;
  }

  sort(res,res+ress);
  rep(i,ress) writer(res[i], i==ress-1?'\n':' ');

  return 0;
}

Current time: 2017年11月19日12時12分20秒
Last modified: 2014年11月18日04時12分33秒 (by laycrs)
Tags: Competitive_Programming HackerRank HackerRank_101jul14
トップページに戻る

Logged in as: unknown user (not login)

ログイン: