HackerRank 101 Hack July'14 2問目
problem statement(コンテストページ)
problem statement
バス停に並んでいる $N$ 組のグループが居る.
$k$ 番目のグループは $A_k$ 人からなる.
バスは全て $k$ 人乗りとして,以下の条件1~3.を満たさなければいけない.
考えうる $k$ を全て列挙する問題.
$k$ を決めれば,それが条件を満たすかどうかは,前からシミュレーションすれば $O(N)$ で判定可能.
毎回 $k$ 人乗せるので,$S = A_1+A_2+\cdots+A_N$ とすれば,$k$ は $S$ の約数でなければいけない.
$S$ は高々 $10^9$ で約数はそんなに多くないので,全部試しても間に合う.
(実際には,多くの $k$ に対して,すぐに条件を満たさないことがわかって,判定に $O(N)$ かからないことが多いと思われる)
#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: 2024年03月29日22時19分08秒
Last modified: 2014年11月18日04時12分33秒 (by laycrs)
Tags: Competitive_Programming HackerRank HackerRank_101jul14
トップページに戻る
Logged in as: unknown user (not login)