程式設計之美初賽第一場--樹

描述

有一個N個節點的樹，其中點1是根。初始點權值都是0。

一個節點的深度定義為其父節點的深度+1,。特別的，根節點的深度定義為1。

現在需要支援一系列以下操作：給節點u的子樹中，深度在l和r之間的節點的權值（這裡的深度依然從整個樹的根節點開始計算），都加上一個數delta。

問完成所有操作後，各節點的權值是多少。

為了減少巨大輸出帶來的開銷，假設完成所有操作後，各節點的權值是answer[1..N]，請你按照如下方式計算出一個Hash值（請選擇合適的資料型別，注意避免溢位的情況）。最終只需要輸出這個Hash值即可。

MOD =1000000007; // 10^9 + 7

MAGIC= 12347;

Hash =0;

For i= 1 to N do

   Hash = (Hash * MAGIC + answer[i]) mod MOD;

EndFor

輸入

第一行一個整數T (1 ≤ T ≤ 5)，表示資料組數。

接下來是T組輸入資料，測試資料之間沒有空行。

每組資料格式如下：

第一行一個整數N (1 ≤ N ≤ 105)，表示樹的節點總數。

接下來N - 1行，每行1個數，a (1 ≤ a ≤ N)，依次表示2..N節點的父親節點的編號。

接下來一個整數Q(1 ≤ Q ≤ 105)，表示操作總數。

接下來Q行，每行4個整數，u, l, r, delta (1 ≤ u ≤ N, 1 ≤ l ≤ r ≤ N, -109 ≤ delta ≤ 109)，代表一次操作。

輸出

對每組資料，先輸出一行“Case x: ”，x表示是第幾組資料，然後接這組資料答案的Hash值。

資料範圍

小資料：1 ≤ N, Q ≤ 1000

大資料：1 ≤ N, Q ≤ 105

樣例解釋

點1的子樹中有1,2,3三個節點。其中深度在2-3之間的是點2和點3。

點2的子樹中有2,3兩個節點。其中沒有深度為1的節點。

所以，執行完所有操作之後，只有2,3兩點的權值增加了1。即答案是0 1 1。再計算對應的Hash值即可。

#include <iostream>
#include <stdio.h>
#include <memory.h>
using std::endl;
using std::cin;
using std::cout;
int main()
{
	int Parent[1001];
	int Depth[1001];
	int Answer[1001];
	int T;
	cin >> T;
	int cnt=0;
	while(T--)
	{
		cnt++;
		//初始化
		memset(Parent,0,sizeof(Parent));
		memset(Depth,0,sizeof(Depth));
		memset(Answer,0,sizeof(Answer));
		//初始化根節點編號的深度值
		Depth[1]=1;
		int N;
		cin >> N;
		int n;
		//輸入2..N節點的父親節點的編號
		for(int i=2;i<=N;++i)
		{
			cin >> n;
			//儲存父親節點
			Parent[i]=n;
			//更新深度
			Depth[i]=Depth[n]+1;
		}
		int Q;
		//輸入操作的次數
		cin >> Q;
		int u, l, r, delta;
		for(int i=0;i<Q;++i)
		{
			cin >> u >> l >> r >> delta;
			for(int j=1;j<=N;++j)
			{
				//u的子樹包括u節點本身
				if(Depth[u]>=l&&Depth[u]<=r)
				{
					Answer[u]+=delta;
				}
				//如果節點的深度小於或者等於u，則不可能是u(除了u本身外)的子樹
				if(Depth[j]>Depth[u])
				{
					if(Parent[j]==u)
					{//u的第一代孩子
						if(Depth[j]>=l&&Depth[j]<=r)
						{
							Answer[j]+=delta;
						}
					}else{
						//不是u的直接孩子
						int temp=j;
						while(Depth[temp]!=Depth[u])
						{
							temp=Parent[temp];
						}
						//來判斷是否為u的子樹節點
						if(temp==u&&(Depth[j]>=l&&Depth[j]<=r))
						{
							Answer[j]+=delta;
						}
					}
				}
			}
		}
		//對儲存在Answer中的權值進行最後的輸出處理
		int MOD =1000000007; // 10^9 + 7
		int MAGIC= 12347;
		int Hash =0;
		for(int i=1;i<=N;++i)
		{
			Hash = (Hash * MAGIC + Answer[i]) % MOD;
		}
		//輸出
		cout << "Case " << cnt << ": ";
		cout << Hash << endl;
	}
	return 0;
}