AtCoder Beginner Contest #006 B問題 - トリボナッチ数列

Source

AtCoder Beginner Contest #006
問題文

問題概要

トリボナッチ数列 $\{a_k\}$ を
 $a_1 = 0,\; a_2 = 0,\; a_3 = 1, \; a_n = a_{n-1} + a_{n-2} + a_{n-3}, \;\; n \geq 4$
で定義する.
正整数 $N$ が与えられるので,$a_N\ {\rm mod}\ 10007$ を求める問題.

解法

$a_k$ を $k$ が小さい方から漸化式に従って計算していく.

C++によるスパゲッティなソースコード

#include<cstdio>
#include<vector>
#include<set>
#include<map>
#include<queue>
#include<algorithm>
#include<string>
#include<utility>
#include<iostream>
#include<cmath>
using namespace std;

#define REP(i,a,b) for(i=a;i<b;i++)
#define rep(i,n) REP(i,0,n)

#define M 10007

void reader(int *x){int k,m=0;*x=0;for(;;){k=getchar_unlocked();if(k=='-'){m=1;break;}if('0'<=k&&k<='9'){*x=k-'0';break;}}for(;;){k=getchar_unlocked();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)putchar_unlocked('-');while(sz--)putchar_unlocked(buf[sz]+'0');putchar_unlocked(c);}

int tri[1000001];

int main(){
  int i, n;

  reader(&n);
  tri[1]=tri[2]=0;
  tri[3]=1;
  REP(i,4,n+1) tri[i]=(tri[i-1]+tri[i-2]+tri[i-3])%M;

  writer(tri[n], '\n');

  return 0;
}

Current time: 2017年07月23日17時37分22秒
Last modified: 2014年05月24日04時09分00秒 (by laycrs)
Tags: Competitive_Programming AtCoder AtCoder_Beginner_Contest ABC006 ABC_B
トップページに戻る

Logged in as: unknown user (not login)

ログイン: