大神与三位小伙伴
描述
L国是一个有着优美景色且物产丰富的国家,很多人都喜欢来这里旅游并且喜欢带走一些纪念品,大神同学也不例外。距离开L国的时间越来越近了,大神同学正在烦恼给她可爱的小伙伴们带什么纪念品好,现在摆在大神同学面前的有三类纪念品A, B, C可以选择,每类纪念品各有N种。其中种类为A_i, B_i, C_i的纪念品价值均为i, 且分别有N+1-i个剩余。现在大神同学希望在三类纪念品中各挑选一件然后赠送给她的三名可爱的小伙伴,但是她又不希望恰好挑出来两件价值相同的纪念品,因为这样拿到相同价值纪念品的两位小伙伴就会认为大神同学偏袒另一位小伙伴而不理睬她超过一星期。现在,大神同学希望你买到的三件纪念品能让三位小伙伴都开心并且不和她闹别扭,她想知道一共有多少种不同挑选的方法?
因为方案数可能非常大,大神同学希望知道挑选纪念品的方案数模10^9+7之后的答案。
输入
第一行包括一个数T,表示数据的组数。
接下来包含T组数据,每组数据一行,包括一个整数N。
输出
对于每组数据,输出一行“Case x: ”,其中x表示每组数据的编号(从1开始),后接一个数,表示模10^9+7后的选择纪念品的方案数。
数据范围
小数据:
1<=T<=10
1<=N<=100
大数据:
1<=T<=1000
1<=N<=10^18
样例解释
对于第二组数据,合法的方案有以下几种,(X,Y,Z)表示选择了A类纪念品中价值为X的,B类纪念品中价值为Y的,C类纪念品中价值为Z的。
(1,1,1): 3*3*3=27种
(1,2,3): 3*2*1=6种
(1,3,2): 3*1*2=6种
(2,1,3): 2*3*1=6种
(2,2,2): 2*2*2=8种
(2,3,1): 2*1*3=6种
(3,1,2): 1*3*2=6种
(3,2,1): 1*2*3=6种
(3,3,3): 1*1*1=1种
一共27+6+6+6+8+6+6+6+1=72种选择纪念品的方案
注意,如(1,1,2), (2,3,3), (3,1,3)都因为恰好选择了两件价值相同的纪念品,所以并不是一种符合要求的纪念品选择方法。
--------------------------------------------------------------------------------------------------------------
我的解法
所有情况减去恰好选择了两件价值相同的纪念品的情况,
结果 res=S^3 - 3*Sum(i^2 * (S-i)) , 其中 i=1,2...N , S=(1+N)*N/2
res=S^3 - 3*Sum(i^2 * S-i^3)
res=S^3 - 3*(Sum(i^2 )* S-Sum(i^3))
是不是感觉很眼熟,对就是平方和和立方和公式,如下:
Sum(i^2 )=N*(N+1)*(2*N+1)/6
Sum(i^3)=[n (n+1) / 2]^2 =S^2
需要注意一点就是:(a-b) mod p =(a%p -b%p+p)%p
代码如下:
#include<iostream>
#include<cstdlib>
const int P=1000000007;
int getSDivP(long long N){//计算S%P
if(0==N%2)
return (N/2%P)*((N+1)%P)%P;
else
return (N%P)*((N+1)/2%P)%P;
}
int getI2DivP(long long N){//计算平方和%P
long long res=1;
long long NN[3]={N,N+1,2*N+1};
bool hasDiv2=false;
bool hasDiv3=false;
for(int i=0;i<=2;++i){
if(!hasDiv2){
if(0==NN[i]%2){
NN[i]/=2;
hasDiv2=true;
}
}
if(!hasDiv3){
if(0==NN[i]%3){
NN[i]/=3;
hasDiv3=true;
}
}
NN[i]%=P;
res*=NN[i];
res%=P;
}
return res;
}
int comp(long long N){
long long S=getSDivP(N);
long long S3=S;
S3*=S;
S3%=P;
long long sum3=S3;//计算立方和%P
S3*=S;
S3%=P;
long long sum2=getI2DivP(N);
sum2*=S;
sum2%=P;
//std::cout<<S<<" "<<S3<<" "<<sum3<<" "<<sum2<<std::endl;
return (S3-(sum2-sum3+P)%P*3%P+P)%P;
}
int main(int argc,char** argv){
int T=0;
long long N=0;
std::cin>>T;
for(int i=1;i<=T;++i){
std::cin>>N;
std::cout<<"Case "<<i<<": "<<comp(N)<<std::endl;;
}
return EXIT_SUCCESS;
}
上面的方法能通过小数据部分,大数据部分NA,群里面说大数据部分是不测试的,感觉大数据部分应该没问题。
大数据部分是资格赛结束后才判的,上面的代码没问题。