yukicoder No.54 - Happy Hallowe'en

Source

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

問題概要

$N$ 個の家があり,家 $k$ に行った時,持っているお菓子の数が $T_k$ より少なければ,$V_k$ 個のお菓子がもらえる(各家につき $1$ 回まで).
最初お菓子を持っていないとして,もらえるお菓子の数の最大値を求める問題.

解法

家 $i$ の直後に家 $j$ に行く場合と,家 $j$ の直後に家 $i$ に行く場合とで,両方の家でお菓子が貰える条件を考える.
この $2$ つの家に行く前に持っているお菓子の数 $k$ は,$k < \min(T_i, T_j)$ でなければいけないというのは,どちらにとっても同じ条件なので考えない.
家 $i$ の後に $j$ に行く場合は,$k < T_j - V_i$ であれば良く,家 $j$ の後に $i$ に行く場合は,$k < T_i - V_j$ であれば良い.
今,$T_j - V_i \geq T_i - V_j$,つまり $T_i + V_i \leq T_j + V_j$ とすると,家 $i$ の後に $j$ に行くための条件は,家 $j$ の後に家 $i$ に行くための条件より緩い.
つまり,家 $j$ に行った後に家 $i$ に行けるなら,家 $i$ に行った後に家 $j$ に行けることになる.
よって,家 $j$ に行った後に家 $i$ に行くことは考えなくて良くて,家 $i$ の後に家 $j$ に行くことだけ考えれば良い.
つまり,$T_k+V_k$ の値でソートして,この値が小さい順に家を訪れることのみを考えれば良い.
後は,答えをDPを用いて計算できる.
状態として,今何個目の家まで訪れた(お菓子をもらわないという選択もできるとする)か,今もらったお菓子の数は幾つか,でそのような状態に移動できるかどうかを調べれば良い.
時間計算量は $O(N \log N + V_k + N T_k)$ 程度になるが,十分に間に合う.
DPをビット並列すれば数十倍速くなる(はず).

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 reader(int *x, int *y){reader(x);reader(y);}
void writer(int x, char c){int s=0,m=0;char f[10];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);}
template<class T1, class T2, class T3> void sort(int N, T1 a[], T2 b[], T3 c[], void *mem){int i;pair<T1,pair<T2,T3> > *r=(pair<T1,pair<T2,T3> >*)mem;rep(i,N)r[i].first=a[i],r[i].second.first=b[i],r[i].second.second=c[i];sort(r,r+N);rep(i,N)a[i]=r[i].first,b[i]=r[i].second.first,c[i]=r[i].second.second;}

int N, V[20000], T[20000], val[20000];
int dp[20001];

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

  reader(&N);
  rep(i,N) reader(V+i, T+i);

  rep(i,N) val[i] = T[i]+V[i];
  sort(N, val, T, V, mem);
  rep(i,20001) dp[i] = 0;
  dp[0] = 1;
  
  rep(k,N){
    for(i=T[k]-1;i>=0;i--) if(dp[i]) dp[i+V[k]] = 1;
  }
  for(i=20000;i>=0;i--) if(dp[i]) { res = i; break; }
  writer(res,'\n');

  return 0;
}

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

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);}
template <class T, class S> void reader(T *x, S *y){reader(x);reader(y);}
void writer(int x, char c){int s=0,m=0;char f[10];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);}
template<class T> void writerLn(T x){writer(x,'\n');}

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;}
template<class T1, class T2, class T3> void sort(int N, T1 a[], T2 b[], T3 c[], void *mem){int i;pair<T1,pair<T2,T3> > *r=(pair<T1,pair<T2,T3> >*)mem;rep(i,N)r[i].first=a[i],r[i].second.first=b[i],r[i].second.second=c[i];sort(r,r+N);rep(i,N)a[i]=r[i].first,b[i]=r[i].second.first,c[i]=r[i].second.second;}

void bitOR(ull a[], ull b[], int stA, int stB, int len){int i,B,E,R,Y,X,G,Z;static ull p[65],q[65];static int f=0;if(!f){f=1;p[0]=0;rep(i,64)p[i+1]=(p[i]<<1)|1ULL;rep(i,65)q[i]=p[64]^p[i];}B=stA/64;E=stA%64;R=(stA+len)/64;Y=(stA+len)%64;X=stA-stB;G=X/64;Z=X%64;if(Z<0)Z+=64,G--;if(stA>stB){if(B==R){if(Z)b[B-G-1]|=(a[B]&p[Y]&q[E])<<(64-Z);b[B-G]|=(a[B]&p[Y]&q[E])>>Z;}else{if(Z)b[B-G-1]|=(a[B]&q[E])<<(64-Z);b[B-G]|=(a[B]&q[E])>>Z;REP(i,B+1,R){if(Z)b[i-G-1]|=a[i]<<(64-Z);b[i-G]|=a[i]>>Z;}if(Z)b[R-G-1]|=(a[R]&p[Y])<<(64-Z);b[R-G]|=(a[R]&p[Y])>>Z;}}else{if(B==R){b[B-G]|=(a[B]&p[Y]&q[E])>>Z;if(Z)b[B-G-1]|=(a[B]&p[Y]&q[E])<<(64-Z);}else{b[R-G]|=(a[R]&p[Y])>>Z;if(Z)b[R-G-1]|=(a[R]&p[Y])<<(64-Z);for(i=R-1;i>B;i--){b[i-G]|=a[i]>>Z;if(Z)b[i-G-1]|=a[i]<<(64-Z);}b[B-G]|=(a[B]&q[E])>>Z;if(Z)b[B-G-1]|=(a[B]&q[E])<<(64-Z);}}}

char memarr[77000000]; void *mem = memarr;

int N, V[20000], T[20000], val[20000];
ull dp[400];

int main(){
  int i, j, k, loop;
  int res=0, tmp;

  reader(&N);
  rep(i,N) reader(V+i, T+i);

  rep(i,N) val[i] = T[i]+V[i];
  sort(N, val, T, V, mem);
  rep(i,400) dp[i] = 0;
  dp[0] = 1;
  
  rep(k,N) bitOR(dp, dp, 0, V[k], T[k]);
  for(i=20000;i>=0;i--) if(dp[i/64]&(1ULL<<(i%64))) { res = i; break; }
  writerLn(res);

  return 0;
}

Current time: 2017年07月24日15時48分55秒
Last modified: 2015年02月02日03時25分34秒 (by laycrs)
Tags: Competitive_Programming yukicoder
トップページに戻る

Logged in as: unknown user (not login)

ログイン: