1382 - The Queue
Time Limit: 2 second(s) | Memory Limit: 32 MB |
On some special occasions Nadia's company provide very special lunch for all employees of the company. Before the food is served all of the employees must stand in a queue in front of the food counter. The company applied a rule for standing in the queue. The rule is nobody can stand anywhere in front of his supervisor in the queue. For example, if Abul is the supervisor of Babul and Abul stands in kth position from the front of the queue, then Babul cannot stand at any position in between 1 and k - 1 from front of the queue.
The company has N employees and each of them has exactly one supervisor except one (CEO) who doesn't have any supervisor.
You have to calculate in how many ways the queue can be created. For this problem, you can safely assume that in at least one way the queue can be created.
Input
Input starts with an integer T (≤ 700), denoting the number of test cases.
Each case starts with a line containing an integer N (1 ≤ N ≤ 1000). Each of the following N - 1 lines will contain two integers a and b (1 ≤ a, b ≤ N, a ≠ b), which denotes that a is the supervisor of b. For the sake of simplicity we are representing each employee by an integer number. Assume that the given input follows the restrictions stated above.
Output
For each case, print the case number and the number of ways to create the queue. The result can be large, print the result modulo 1000 000 007.
Sample Input |
Output for Sample Input |
1 5 2 1 2 3 3 4 3 5 |
Case 1: 8 |
1 #include<stdio.h> 2 #include<algorithm> 3 #include<iostream> 4 #include<string.h> 5 #include<queue> 6 #include<stack> 7 #include<vector> 8 using namespace std; 9 typedef long long LL; 10 const int N=1e9+7; 11 vector<int>vec[1100]; 12 queue<int>que; 13 int ans[2100]; 14 LL dp[1100]; 15 LL cnt[1100]; 16 LL ju[1100][1100]; 17 void dfs(int n); 18 int main(void) 19 { 20 int k,i,j; 21 ju[0][0]=1; 22 ju[1][0]=1; 23 ju[1][1]=1; 24 for(i=2; i<=1099; i++) 25 { 26 for(j=0; j<=i; j++) 27 { 28 if(i==j||j==0) 29 ju[i][j]=1; 30 else ju[i][j]=(ju[i-1][j]+ju[i-1][j-1])%N; 31 } 32 } 33 scanf("%d",&k); 34 int s; 35 int n,m; 36 int x,y; 37 for(s=1; s<=k; s++) 38 { 39 for(i=0; i<1050; i++) 40 { 41 dp[i]=1; 42 vec[i].clear(); 43 } 44 scanf("%d",&n); 45 for(i=1; i<=n; i++) 46 cnt[i]=i; 47 for(i=1; i<n; i++) 48 { 49 scanf("%d %d",&x,&y); 50 vec[x].push_back(y); 51 cnt[y]=x; 52 } 53 int id=1; 54 memset(ans,0,sizeof(ans)); 55 for(i=1; i<=n; i++) 56 { 57 if(cnt[i]==i) 58 id=i; 59 } 60 memset(ans,0,sizeof(ans)); 61 dfs(id); 62 printf("Case %d: ",s); 63 printf("%lld\n",dp[id]); 64 } 65 return 0; 66 } 67 void dfs(int n) 68 { 69 if(!vec[n].size()) 70 { 71 ans[n]=1; 72 dp[n]=1; 73 return ; 74 } 75 int cc=vec[n].size(); 76 int i,j; 77 LL ak=0; 78 ans[n]+=1; 79 for(i=0; i<vec[n].size(); i++) 80 { 81 dfs(vec[n][i]); 82 ans[n]+=ans[vec[n][i]]; 83 dp[n]=(dp[vec[n][i]]*dp[n]%N)*(ju[ans[n]-1][ans[vec[n][i]]])%N ; 84 } 85 }