题目描述
函数 fx 定义如下
f(x)={xa1f(x−1)+a2f(x−2)+…+a10f(x−10)(x<10)(x>=10)
其中 ai(1≤i≤10)取值为 0 或者 1。
输入
第一行输入两个数k,m(1≤k,m≤105)
第二行输入10个整数分别表示a数组
输出
输出f(k)对m取模的结果
样例输入1
13 100
0 1 1 0 1 1 1 0 0 0
样例输出1
80
#include <iostream>
using namespace std;
int a[11], num[100005]={0,1,2,3,4,5,6,7,8,9};
int f(int x) {
if (x < 10) return num[x];
else return a[1] * num[x - 1] + a[2] * num[x - 2] + a[3] * num[x - 3]
+ a[4] * num[x - 4] + a[5] * num[x - 5] + a[6] * num[x - 6] +
a[7] * num[x - 7] + a[8] * num[x - 8] + a[9] * num[x - 9] + a[10] * num[x - 10];
}
int main() {
int k, m;
cin >> k >> m;
for (int i = 1; i <= 10; i++) {
cin >> a[i];
}
for (int i = 10; i <= k; i++) {
num[i] = f(i) % m;
}
cout << num[k] << endl;
return 0;
}