2-3 Trees
Problem Description
2-3 tree is an elegant data structure invented by John Hopcroft. It is designed to implement the same functionality as the binary search tree. 2-3 tree is an ordered rooted tree with the following properties:
- the root and each internal vertex have either 2 or 3 children;
- the distance from the root to any leaf of the tree is the same.
The only exception is the tree that contains exactly one vertex — in this case the root of the tree is the only vertex, and it is simultaneously a leaf, i.e. has no children. The main idea of the described properties is that the tree with l leaves has the height O(log l).
Given the number of leaves l there can be several valid 2-3 trees that have l leaves. For example, the picture below shows the two possible 2-3 trees with exactly 6 leaves.
Given l find the number of different 2-3 trees that have l leaves. Since this number can be quite large, output it modulo r.
Input
Output
Sample Input
6 1000000000 7 1000000000
Sample Output
2 3
Source
Manager
#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <map>
#include <cstdlib>
#include <sstream>
#include <queue>
#include <cmath>
using namespace std;
int c[2510][2510];
int main()
{
long long n,m;
while(cin>>n>>m){
memset(c,0,sizeof(c));
c[0][0]=1;
c[1][1]=c[1][0]=1;
for(int i=2;i<=n/2;i++){
c[i][0]=1;
for(int j=1;j<=i;j++)
c[i][j]=(c[i-1][j]%m+c[i-1][j-1]%m)%m;
}
long long dp[15010];
memset(dp,0,sizeof(dp));
dp[1]=1;
for(int i=1;i<n/2+1;i++)
{
for(int j=0;j<=i;j++)
{
int q=j*3+(i-j)*2;
dp[q]=(dp[q]+(dp[i]*c[i][j]))%m;
}
}
cout<<dp[n]<<endl;
}
}