LeetCode 生成数组

/**
dp[n][k][m] 定义为：n个位置，k个search_cost，以m结尾的数组种类个数
由这个定义，可得初始化
dp[1][1][1...m] = 1
转移方程为：dp[n][k][m] = m*dp[n-1][k][m] + dp[n-1][k-1][m-1] + dp[n-1][k-1][m-2] + ... + dp[n-1][k-1][1]
由此最后结果：n个位置，k个search_cost，最大值m的数组种类为：ans = ans + dp[n][k][1...m]
*/
func numOfArrays(n int, m int, k int) int {
	mod := uint64(1000000007)
	dp := make([][110][110]uint64,110)

	for j:=1; j<=m; j++ {
		dp[1][1][j] = 1
	}

	for i:=2; i<=n; i++ {
		for ki:=1; ki<=k; ki++ {
			for j:=1; j<=m; j++ {
				dp[i][ki][j] = (uint64(j) * dp[i-1][ki][j]) % mod
				for jj:=1; jj<j; jj++ {
					dp[i][ki][j] = (dp[i][ki][j] + dp[i-1][ki-1][jj]) % mod
				}
			}
		}
	}

	res := uint64(0)
	for j := 1; j <= m; j++{
		res = (res + dp[n][k][j]) % mod
	}

	return int(res)
}