蓝桥杯1443数字游戏

题目内容

栋栋正在和同学们玩一个数字游戏。

游戏的规则是这样的：栋栋和同学们一共n个人围坐在一圈。栋栋首先说出数字1。接下来，坐在栋栋左手边的同学要说下一个数字2。再下面的一个同学要从上一个同学说的数字往下数两个数说出来，也就是说4。下一个同学要往下数三个数，说7。依次类推。

为了使数字不至于太大，栋栋和同学们约定，当在心中数到 k-1 时，下一个数字从0开始数。例如，当k=13时，栋栋和同学们报出的前几个数依次为：
1, 2, 4, 7, 11, 3, 9, 3, 11, 7。

游戏进行了一会儿，栋栋想知道，到目前为止，他所有说出的数字的总和是多少。

样例说明
栋栋说出的数依次为1, 7, 9，和为17。
数据规模和约定
1 < n,k,T < 1,000,000；

输入

输入的第一行包含三个整数 n,k,T，其中 n 和 k 的意义如上面所述，T 表示到目前为止栋栋一共说出的数字个数。

输出

输出一行，包含一个整数，表示栋栋说出所有数的和。

样例输入

3 13 3

样例输出

17

题目分析

做之前没注意看T的取值范围，就用常规的方法依次求解，果然就出现了TLE。
后来找了一下规律：
a1 = 1;
a2 = a1 + (1+2+3+4+…+n);
a3 = a2 + (n+1 + n+2 + n+3 +…+2n);
由此会发现其中的规律，令累加的数记为add，则初始值add1（a1 + add1 = a2）相当于是在a1的基础上加了add变为a2，所以add1 = (1+n)n/2，（等差数列求和），之后会发现a2 + add = a3的过程中，此时的add2 = (n+1+2n)n/2，记为add2，而add2-add1 = n*n，所以可以得出每次的add都是在前一次的基础上加了n的平方，这样一来题目就会迎刃而解了。

代码部分

先是附上超时代码（用数组做的，当时没多想）

#include<iostream>
using namespace std;
#include<cstring>

#define maxn 1000005
long long int a[maxn], sum, n, k, T, add = 0;

int main()
{
    cin>>n>>k>>T;
    int temp = n;
    memset(a,0,sizeof(a));
    a[0] = 1;
    sum += a[0];
    int t = 1;
    while(T != 1)
    {
        T--;
        n = temp;
        a[t] = a[t-1];
        while(n--)
        {
            add++;
            a[t] = (a[t] + add) % k;
        }
        sum += a[t];
        t++;
    }
    cout<<sum<<endl;
    return 0;
}

AC代码：

#include <iostream>
using namespace std;

int main()
{
    long long int n, k, T, sum, number, add;
    cin >> n >> k >> T;
    sum = 1;
    number = 1;
    add = n*(n+1)/2;
    for (int i = 1; i < T; i++)
    {
        number += add;
        add += n*n;
        if(number > (k-1))
            number = number%k;
        sum += number;
    }
    cout << sum << endl;
    return 0;
}