题目:
给你一维空间的 n 个点,其中第 i 个点(编号从 0 到 n-1)位于 x = i 处,请你找到 恰好 k 个不重叠 线段且每个线段至少覆盖两个点的方案数。线段的两个端点必须都是 整数坐标 。这 k 个线段不需要全部覆盖全部 n 个点,且它们的端点 可以 重合。
请你返回 k 个不重叠线段的方案数。由于答案可能很大,请将结果对 109 + 7 取余 后返回。
示例 1:
输入:n = 4, k = 2
输出:5
解释:
如图所示,两个线段分别用红色和蓝色标出。
上图展示了 5 种不同的方案 {(0,2),(2,3)},{(0,1),(1,3)},{(0,1),(2,3)},{(1,2),(2,3)},{(0,1),(1,2)} 。
示例 2:
输入:n = 3, k = 1
输出:3
解释:总共有 3 种不同的方案 {(0,1)}, {(0,2)}, {(1,2)} 。
示例 3:
输入:n = 30, k = 7
输出:796297179
解释:画 7 条线段的总方案数为 3796297200 种。将这个数对 109 + 7 取余得到 796297179 。
示例 4:
输入:n = 5, k = 3
输出:7
示例 5:
输入:n = 3, k = 2
输出:1
提示:
2 <= n <= 1000
1 <= k <= n-1
代码:
方法一——自己写的,只有最后一个用例超时:
class Solution {
public:
int numberOfSets(int n, int k) {
n=n-1;
int base=1e9+7;
int v[1001][1001]={0};
for(int i=0;i<=n;i++) {
v[i][i] = 1;
v[i][0]=1;
}
for(int i=1;i<=k;i++){
for(int j=i+1;j<=n;j++){
v[j][i]=v[j-1][i];
int f=j-1;
while(f>=i-1){
v[j][i]=(v[j][i]+v[f][i-1])%base;
f--;
}
}
}
return v[n][k];
}
};
思路:
可以按照新增的一个n放的位置长度来确定。
v[j][i]=v[j-1][i]+\sum_{f=i-1}^{f=j-1}v[f][i-1];
从0到最大,累加。
方法二——不超时版本
class Solution {
public:
int numberOfSets(int n, int k) {
long dp[n+1][k+1];
long sum[n+1][k+1];
memset(dp,0,sizeof(dp));
memset(sum,0,sizeof(sum));
for(int i = 1;i<=n;i++)
{
dp[i][0] = 1;
}
for(int i = 1;i<=n;i++)
{
for(int j = 1;j<=k;j++)
{
sum[i][j-1] = (sum[i-1][j-1]+dp[i-1][j-1])%1000000007;
dp[i][j]=(sum[i][j-1] + dp[i-1][j])%1000000007;
}
}
return dp[n][k];
}
};
想法:
理清思路,使用sum记录前n项的和,然后用在后面。