保存されている過去のバージョンの一覧

2014年12月05日00時55分09秒

yukicoder No.33 - アメーバがたくさん

Source

ニコニコミュニティ
問題文

問題概要

一次元上に，最初アメーバが $N$ 匹いて，アメーバ $k$ の座標は $X_k$である．
各アメーバは，毎時間，$3$ つに分裂し，座標 $x$ にいた場合は，座標 $x,x+D,x-D$ の場所にアメーバができる．
ただし，同じ座標に $2$ 匹以上のアメーバが同時に存在する状態になれば，それらのアメーバは合体し $1$ 匹のアメーバとなる．
時間 $T$ 後のアメーバの数を求める問題．

解法

座標を $D$ で割った余り事に独立なので，個別に考える． つまり，各 $0 \leq k < D$ ごとに，座標が $Dx+k$ と表されるアメーバのみを対象にする．
元々の座標で $Dx+k$ というのを，新しく座標 $x$ と思えば，各アメーバは，分裂することで，今いる場所と前後 $\pm 1$ の座標にできる．
つまり，最初（新しい座標で）$x$ にアメーバがいれば，最終的には区間 $[x-T,x+T]$ 内の各座標にアメーバがいることになる．
後は，複数の区間が覆う整数座標の数を調べれば良いのだが，区間の始点でソートして，重なりを排除しつつ数えれば良い．
時間計算量 $O(N \log N)$ だが，以下のコードでは独立に扱う各グループに属すアメーバを愚直にループしているため $O(N^2)$．

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)

#define ll long long
#define ull unsigned ll

template<class T1, class T2> void sort(int N, T1 a[], T2 b[], void *mem){int i;pair<T1,T2> *r=(pair<T1, T2>*)mem;rep(i,N)r[i].first=a[i],r[i].second=b[i];sort(r,r+N);rep(i,N)a[i]=r[i].first,b[i]=r[i].second;}

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(ll x, char c){int s=0,m=0;char f[20];if(x<0)m=1,x=-x;while(x)f[s++]=x%10,x/=10;if(!s)f[s++]=0;if(m)mypc('-');while(s--)mypc(f[s]+'0');mypc(c);}

int N, D, T;
int X[100];

ll res;
int done[100];
ll st[100], ed[100], sz, bef;

int main(){
  int i, j, k;
  void *mem = malloc(10000000);

  reader(&N); reader(&D); reader(&T);
  rep(i,N) reader(X+i), X[i] += 1000000000;

  rep(k,N) if(!done[k]){

    sz = 0;
    rep(i,N) if((X[i]-X[k])%D==0){
      st[sz] = (ll)X[i] / D;
      ed[sz] = (ll)X[i] / D + 2 * T;
      sz++;
      done[i] = 1;
    }
    sort(sz, st, ed, mem);

    bef = -1;
    rep(i,sz){
      bef = max(bef, st[i]);
      res += ed[i] - bef + 1;
      bef = ed[i] + 1;
    }
  }

  writer(res,'\n');

  return 0;
}

Current time: 2024年04月25日19時37分15秒
Last modified: 2014年12月05日00時55分09秒 (by laycrs)
Tags: Competitive_Programming yukicoder
トップページに戻る

Logged in as: unknown user (not login)

ログイン：