中石油 16304: 数列

题目描述

联欢会上，主持人先在黑板上写了四个数字，分别是：2，0，1，9，然后说：“这串数字从第五个数起，每一个数都是它前四数字之和的个位数字”。请你打印输出这个数列的前N项，并计算前N项的和。

输入

单个整数：表示项数N（1 ≤ N ≤ 100000）

输出

第一行：N个数字，表示所求数列的前N项。 
第二行：单个数字，表示前N项数字的和

样例输入 Copy

10

样例输出 Copy

2 0 1 9 2 2 4 7 5 8 
40

#include<iostream>

using namespace std;

int main()
{
	int n,sum=0,arr[100000]={2,0,1,9};
	cin>>n;
	if(n<=4) {
		for(int i = 0;i<n;i++) {
			cout<<arr[i]<<" ";
			sum += arr[i];
		}
	}else {
		for(int i = 0;i<n;i++) {
			if(i<4) {
				cout<<arr[i]<<" ";
				sum += arr[i];
			}else {
				arr[i] = (arr[i-4]+arr[i-3]+arr[i-2]+arr[i-1])%10;
				sum += arr[i];
				cout<<arr[i]<<" ";
			}
		}
	}
	cout<<endl;
	cout<<sum<<endl;
	return 0;
}