解题思路:
首先,后方的灯泡一定不会影响前面的结果,也就是说灯泡放的方案数是递推的,符合无后效性原则,因此我们可以用动态规划的思想来解决。
我们可以先按照每种灯泡的瓦数排序,然后根据题意开f[i][j]数组,表示第i个位置选了j的方案数。
1.考虑边界。f[0][]显然0号点选什么方案都是1,DP是从这里开始的。
2.状态转移。因为灯泡最多只能连续a[].num个,所以状态转移的循环为 for (int k=1;k<=min(a[j].num,i);k++),当前i号点选了j ,那之前连续的起点为i-k,肯定是选了j-1。
3.v数组用来记录可行性,用于判断是否有方案。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int mod=1e9+7;
bool v[105][105];
int f[105][105],b[105],n,m;
struct node{
int w,num,p;
}a[105];
bool cmp(node a,node b){
return a.w<b.w;
}
int main(){
cin>>n>>m;
for (int i=1;i<=n;i++){
cin>>a[i].w>>a[i].num;
a[i].p=i;
}
for (int i=1;i<=m;i++) cin>>b[i];
sort(a+1,a+n+1,cmp);
for (int i=0;i<=n;i++){
f[0][i]=1;
v[0][i]=1;
}
for (int i=1;i<=m;i++)
for (int j=1;j<=n;j++){
v[i][j]=v[i][j-1];
f[i][j]=f[i][j-1];
for (int k=1;k<=min(a[j].num,i);k++)//考虑连续K个的情况
if (!b[i-k+1]||b[i-k+1]==a[j].p){
v[i][j]=v[i][j]||v[i-k][j-1];
f[i][j]=(f[i][j]+f[i-k][j-1])%mod;
}
else
break;
}
if (!v[m][n]) cout<<-1;
else cout<<f[m][n];
return 0;
}