网上的动态规划代码看不懂,但思路差不多懂了,这里说一下我的想法。
每释放一个囚犯,整个“监狱”就被分为两部分。因此可以使用分治的思路。每次将当前区间分为左右两段并求和,并取各种方法的最小值,最后加上区间长度即可。如果该区间内找不到需释放的囚犯,则直接返回区间长度。
(以样例2为例)
根据上图写出如下伪代码:
solve(l,r):
t=INF,found=false;
for(遍历a,对于所有的a[i]∈[l,r]):
found=true;
t=min(t,solve(l,a[i]-1)+solve(a[i]+1,r);
if(!found) //无法更新
t=0;
return t+(r-l+1);
很显然,每次递归都可以缩小a的遍历区间。为方便起见,设a[0]=0,a[q+1]=p+1,便可写出如下代码:
int solve(int l,int r,const int st=0,const int end=q+1) {
//后两个参数是查找区间
int t=INF;
for(int k=st; k<=end; k++) {
if(a[k]>=l&&a[k]<=r&&!vis[a[k]]) { //释放a[k]号囚犯
vis[a[k]]=1;
int resl=solve(l,a[k]-1,st,k-1),resr=solve(a[k]+1,r,k+1,end);
t=min(t,resl+resr);
vis[a[k]]=0;
}
}
if(t==INF) t=0;
return t+(r-l+1);
}
加上记忆化,应该可以解决问题。
Update: 可以发现vis数组完全没必要,将它删除,可以节省一点时间。