程式設計之美資格賽--題目2 : 大神與三位小夥伴

jsjliuyun發表於2014-04-13
程式設計

題目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
解析：我直接用暴力搞的，大資料應該過不了，小資料過了，大家可以參考一下，取餘的時候不能最後取餘應該數值太大了，要先取餘使數值變小再累加即可。後來想想用減法代替取餘會不會快點，時間上有一點點改進，不知道是不是網速的問題，大資料肯定過不去，得用什麼演算法來減少複雜度哈！下面貼一下程式碼供大家參考哈！
#include <iostream>
using std::endl;
using std::cin;
using std::cout;
int main()
{
	int T;
	cin >> T;
	int num=1000000007;
	for(int cnt=1;cnt<=T;++cnt)
	{
		int N;
		cin >> N;
		int sum=0;
		//先計算三個紀念品價值都相同的
		for(int i=1;i<=N;++i)
		{
			sum+=(N+1-i)*(N+1-i)*(N+1-i);
			//做一個判斷
			if(sum > num)
			{
				sum-=num;
			}
		}
		for(int i=1;i<=N;++i)
			for(int j=1;j<=N;++j)
			{
				//其中兩個紀念品價值相同的不符合條件要排除
				if(j!=i)
				{
					for(int k=1;k<=N;++k)
					{
						//第三個紀念品仍然不能與前兩個相同
						if((k!=i)&&(k!=j))
						{
							//找到符合條件的繼續累加
							sum+=(N+1-i)*(N+1-j)*(N+1-k);
							//做一個判斷
							if(sum > num)
							{
								sum-=num;
							}
						}
					}
				}
			}
			cout << "Case "<< cnt << ": ";
			cout << sum << endl;
	}
	return 0;
}










相關文章