问题描述
栋栋正在和同学们玩一个数字游戏。
游戏的规则是这样的:栋栋和同学们一共n个人围坐在一圈。栋栋首先说出数字1。接下来,坐在栋栋左手边的同学要说下一个数字2。再下面的一个同学要从上一个同学说的数字往下数两个数说出来,也就是说4。下一个同学要往下数三个数,说7。依次类推。
为了使数字不至于太大,栋栋和同学们约定,当在心中数到 k-1 时,下一个数字从0开始数。例如,当k=13时,栋栋和同学们报出的前几个数依次为:
1, 2, 4, 7, 11, 3, 9, 3, 11, 7。
游戏进行了一会儿,栋栋想知道,到目前为止,他所有说出的数字的总和是多少。
输入格式
输入的第一行包含三个整数 n,k,T,其中 n 和 k 的意义如上面所述,T 表示到目前为止栋栋一共说出的数字个数。
输出格式
输出一行,包含一个整数,表示栋栋说出所有数的和。
样例输入
3 13 3
样例输出
17
样例说明
栋栋说出的数依次为1, 7, 9,和为17。
数据规模和约定
1 < n,k,T < 1,000,000;
数据量很大 直接模拟肯定不行
分析规律 可以看出每个数字其实是 1+0,1+1,1+1+2,1+1+2+3,···· , 1+(1+2+3+···+m),即1 + m * ( m + 1) / 2
而栋栋每轮说出的数字为(1 + m * ( m + 1) / 2)% k, 其中的m=(i-1)*n , (i = 1,2,3,···T),将T轮的结果相加即可得到结果
为了简化计算量,运用模运算的性质
(a + b) % p = (a % p + b % p) % p
(a * b) % p = (a % p * b % p) % p
当m为奇数时,栋栋说出的数字是(1%k+( (m%k)*( (m+1)/2)%k)%k)%k
当m为偶数时,栋栋说出的数字是(1%k+( ( (m/2)%k) * ( (m+1)%k) )%k)%k
这样计算的数字会比较小
数据类型都要用long long
写好之后提交,还是不能ac,百度了一下,发现一位博主说int
型变量和
int
型变量相乘时,所得数的范围为int
类型。会导致数据丢失
检查一下代码 发现for循环里的 i 是 int 型的,循环中(i-1)和 long long int 型的 N 相乘,我这虽然不是int * int,但是我估计还是产生了溢出,把 i 改为long long之后 , ac
#include<iostream>
using namespace std;
long long N,k,T;
int main()
{
cin>>N>>k>>T;
long long sum=0;
long long cn;
for(long long i=1;i<=T;i++)
{
//(a + b) % p = (a % p + b % p) % p
//(a * b) % p = (a % p * b % p) % p
long long n=(i-1)*N;
//cn=(1+n*(n+1)/2)%k;
if(n%2==0)
{ //(n*(n+1)/2)%k=((n/2)%k*(n+1)%k)%k;
cn=(1%k+( ( (n/2)%k) * ( (n+1)%k) )%k)%k;
}
else
cn=(1%k+((n%k)*((n+1)/2)%k)%k)%k;
sum+=cn;
}
cout<<sum<<endl;
return 0;
}
如有不足之处,欢迎评论指出