T241655 [PFOI Round1] 暴龙的火锅

【LGR-110】洛谷 6 月月赛 I & PFOI Round 1 Div. 2

T241655 [PFOI Round1] 暴龙的火锅

题目背景

暴龙爱吃火锅。

题目描述

定义 S(x)S(x) 表示 xx 的每一位的数字之和，例如：S(14)=1+4=5S(14)=1+4=5，S(114514)=1+1+4+5+1+4=16.S(114514)=1+1+4+5+1+4=16.

另外，定义 fib(x)fib(x) 代表斐波那契数列的第 xx 项，具体地：

fib(1)=fib(2)=1, fib(x)=fib(x−1)+fib(x−2) (x≥3).fib(1)=fib(2)=1, fib(x)=fib(x−1)+fib(x−2) (x≥3).

现在给定 nn，求出下式的值，其中  mod9 表示对 99 取余数：

(S(fib(1))+S(fib(2))+S(fib(3))+...+S(fib(n))) (S(fib(1))+S(fib(2))+S(fib(3))+...+S(fib(n)))mod9.

输入格式

第一行一个整数 TT。

接下来 TT 组问询，每次一个整数 nn。

输出格式

TT 行，每行一个整数代表答案。

输入输出样例

输入 #1复制

3
7
14
114514

输出 #1复制

6
5
8

说明/提示

【样例解释】

对于第一组询问，n=7n=7，答案为：

     (S(fib(1))+S(fib(2))…+S(fib(6))+S(fib(7))) =(1+1+2+3+5+8+(1+3)) =6.​     (S(fib(1))+S(fib(2))…+S(fib(6))+S(fib(7)))mod9=(1+1+2+3+5+8+(1+3))mod9=6.​

---

【数据范围】

「本题采用捆绑测试」

• Subtask 1(10 pts)：T=1, n≤10Subtask 1(10 pts)：T=1, n≤10；
• Subtask 2(30 pts)：T=102, n≤103Subtask 2(30 pts)：T=102, n≤103；
• Subtask 3(60 pts)：Subtask 3(60 pts)：无特殊限制。

对于 100%100% 的数据，满足 1≤T≤105, 1≤n≤1061≤T≤105, 1≤n≤106。

#include <iostream>
#include <vector>
#define ll long long
using namespace std;
int main() {
    ios::sync_with_stdio(false);
    cin.tie(NULL);
    vector<ll> fib(1000001),s(1000001),ans(1000001);
    fib[0]=1;
    fib[1]=1;
    s[0]=1;
    s[1]=1;
    ans[0]=1;
    ans[1]=2;
    for(ll i=2;i<1000000;i++)
    {
        fib[i]=fib[i-1]%9+fib[i-2]%9;
        s[i]=fib[i]%9;
        ans[i]=(ans[i-1]+s[i])%9;
    }
//    cout<<fib[99];
    int T,n;
    cin>>T;
    while(T--)
    {
        cin>>n;
        cout<<ans[n-1]<<endl;
    }
    return 0;
}

注意：求某一整数的各位数字之和再模9等价于直接模9