題目2 : 大神與三位小夥伴
描述
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)都因為恰好選擇了兩件價值相同的紀念品，所以並不是
一種符合要求的紀念品選擇方法。
2 1 3樣例輸出
Case 1: 1 Case 2: 72
//source here
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int maxn=1e9+7;
long long ans;
int main()
{
int i,j,k;
int tes,p;
cin>>tes;
for(int cas=1;cas<=tes;cas++)
{
ans=0;
cin>>p;
printf("Case %d: ",cas);
for(i=1;i<=p;i++)
{
for(j=1;j<=p;j++)
{
for(k=1;k<=p;k++)
{
int tt=0;
if(i==j) tt++;
if(j==k) tt++;
if(i==k) tt++;
if(tt!=1)
{
ans=(ans+i*j*k)%maxn;
}
}
}
}
cout<<ans<<endl;
}
return 0;
}
/*
4 400
*/1.先計算所有情況：
(1,1,1) n*n*n
(1,1,2) n*n*(n-1)
……
(n,n,n) 1*1*1
全部加起來就是(1+2+...+n)^3(注：本文中所有^表示冪，而不是異或)，
同時，1+2+..+n=n*(n+1)/2，
所以，(1+2+...+n)^3=(n*(n+1)/2)^3。
2.排除掉有兩個及以上的同樣價值的：
就會有(x,x,i),(x,i,x)(i,x,x) 三種情況(1 <= x,i <= n)。
所以可以得出一共有(1^2+2^2+3^2+...+n^2)*(1+2+...+n)*3，
又因為1^2+2^2+3^3+...+n^2=n*(n+1)*(2*n+1)/6，
所以(1^2+2^2+3^2+...+n^2)*(1+2+...+n)*3=n*(n+1)*(2*n+1)/6*n*(n+1)/2*3。
3.把三個相同的加回來：
那就是1^3+2^3+...+n^3，
而這個又等於(1+2+...+n)^2，
同時，1+2+..+n=n*(n+1)/2，
就可以再簡化為(n*(n+1)/2)^2。
但是要注意的是，這些在第2中情況裡被算了3次！
所以乘三！
4.得出最後結果
#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int mod=1e9+7;
int main()
{
int tes;
long long ans,n,tmp,t1,t2;
cin>>tes;
for(int cas=1;cas<=tes;cas++)
{
cin>>n;
printf("Case %d: ",cas);
if(n%2==0)
{
t1=n/2%mod;
t2=(n+1)%mod;
}
else
{
t1=(n+1)/2%mod;
t2=n%mod;
}
tmp=t1*t2%mod;
ans=tmp*tmp%mod*tmp%mod;
tmp=tmp*(n%mod)%mod*((n+1)%mod)%mod*((n-1)%mod)%mod;
ans=(ans-tmp+mod)%mod;
cout<<ans<<endl;
}
return 0;
}
/*
4 400
*/