編程之美資格題目2 : 大神與三位小夥伴

時間限制:2000ms
單點時限:1000ms
內存限制:256MB

描述

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>
using namespace std;
#define M 1000000007
typedef unsigned long long mylong64;
mylong64 computcount(mylong64 N)
{
	bool flag[101][101][101]={0};
	mylong64 count1=0;
	for (mylong64 x=1;x<=N;++x)
	{
		for (mylong64 y=1;y<=N;++y)
		{
			for (mylong64 z=1;z<=N;++z)
			{

				if (x==y||y==z||x==z)
				{
					if (x==y&&y==z)
					{
						if (!flag[x][y][z])
						{
							count1+=((N+1-x)*(N+1-y)*(N+1-z)%M);
							flag[x][y][z]=1;
							if (count1>M)
							{
								count1=count1%M;
							}
						}
					}
				}
				else
				{
					if (!flag[x][y][z])
					{
						count1+=((N+1-x)*(N+1-y)*(N+1-z)%M)*6;
						flag[x][y][z]=flag[x][z][y]=flag[y][x][z]=flag[y][z][x]=flag[z][x][y]=flag[z][y][x]=1;
						if (count1>M)
						{
							count1=count1%M;
						}
					}
				}
			}
		}
	}
	return count1;
}

int main()
{
	int n;
	cin>>n;
	mylong64 *p=new mylong64[n];
	for (int i=0;i<n;i++)
	{
		cin>>p[i];
	}

	for (int i=0;i<n;i++)
	{
		p[i]=computcount(p[i]);
		cout<<"Case "<<i+1<<": "<<p[i]<<endl;
	}
	return 0;
}

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章