AtCoder Beginner Contest #011 C問題 - 123引き算

Source

AtCoder Beginner Contest #011
問題文

問題概要

最初整数 $N$ が書かれている.
各ステップでは,書かれている整数 $x$ を $x-1$ か $x-2$ か $x-3$ に書き換えることができる.
ただし,$3$ つの整数 $G_1, G_2, G_3$ が与えられ,一瞬でも書かれている整数が $G_1, G_2, G_3$ のどれかになってはいけない.
$100$ ステップ以内に書かれている整数を $0$ にできるかどうかを判定する問題.

解法

$k$ ステップ後に可能な整数を全部列挙し,その情報から,$k+1$ ステップ後に可能な整数を列挙していく.
時間計算量 $O(TN)$,ここで $T=100$ はステップ数.
または,毎回,$G_1,G_2,G_3$ に一致しないようにできるだけ大きな数を引いていく.
時間計算量 $O(T)$.

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(const char c[]){int i;for(i=0;c[i]!='\0';i++)mypc(c[i]);}

int N, NG[310];
int now[310], next[310];

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

  rep(i,310) NG[i] = 0;
  reader(&N);
  rep(i,3) reader(&k), NG[k] = 1;

  rep(i,310) now[i] = 0;
  if(NG[N]==0) now[N] = 1;

  rep(k,100){
    rep(i,310) next[i] = 0;
    rep(i,310) if(now[i]) rep(j,4){
      if(i-j >= 0 && NG[i-j]==0) next[i-j] = 1;
    }
    rep(i,310) now[i] = next[i];
  }

  if(now[0]) writer("YES\n"); else writer("NO\n");

  return 0;
}

Current time: 2017年07月21日13時37分30秒
Last modified: 2014年06月26日21時36分43秒 (by laycrs)
Tags: Competitive_Programming AtCoder AtCoder_Beginner_Contest ABC011 ABC_C
トップページに戻る

Logged in as: unknown user (not login)

ログイン: