首先1 <= time[i] <= 5非常的小,我们把时间这个维度可以压缩成5个二进制位,对应位为1表示需要屏障保护,0表示不需要屏障保护
所以定义f[i][j]为在第i个位置屏障状态为j需要的最少能量,其中0<=j<32,正好是5个二进制位,每个位1或0分别对应相应的时间是否有屏障保护
转移:
1.自身开屏障,对应5个二进制位中变成一些单独的1或者连续的1
2.将自身和右边相邻的一起开屏障
不需要再枚举左边开了,因为右边开包括了左边开的情况
具体的细节代码中有注释
class Solution {
public:
int need[103],f[103][32];
const int inf=0x3f3f3f3f;
int defendSpaceCity(vector<int>& time, vector<int>& position) {
int n= time.size(),max_pos=0; //max_pos记录最大的位置
for(int i=0;i<n;i++) {
need[position[i]+1]|=1<<(time[i]-1); //time是从1开始,把他转化为从0开始;positin转化为1开始,方便f[0]初始化
max_pos= max(max_pos,position[i]+1);
}
memset(f,0x3f,sizeof f);
f[0][0]=0;
for(int i=0;i<=max_pos;i++){
for(int s1=0;s1<32;s1++) if(f[i][s1]!=inf){ //当前的状态存在才能往后转移
//1.考虑它自己开屏障,需要的能量记录在cnt1中
for(int s2=0;s2<32;s2++) if((s1&s2)==0){ //当前状态必须是还没开过屏障的才需要开
int cnt1=0;
for(int j=0;j<5;j++) if((1<<j)&s2){
if(j==0||((1<<(j-1))&s2)==0) cnt1+=2; //首次开需要能量2
else cnt1++; //多维持一时刻需要能量1
}
//2.考虑它跟右边的一起开屏障,需要的能量记录在cnt2中
for(int s3=0;s3<32;s3++) if((s3&(s1|s2))==0){ //如果之前有屏障或者本身自己开了屏障就不能再向右边开屏障
int cnt2=0;
for(int j=0;j<5;j++) if((1<<j)&s3){
if(j==0||((1<<(j-1))&s3)==0) cnt2+=3; //首次一起开需要能量3
else cnt2++; //多维持一时刻需要能量1
}
int s=s1|s2|s3; //原有的屏障状态、自己开的屏障状态、向右边开的屏障状态的并集
if((s&need[i])==need[i]) f[i+1][s3]=min(f[i+1][s3],f[i][s1]+cnt1+cnt2); //至少满足need需求才能转移
}
}
}
}
int ans=inf;
for(int i=0;i<32;i++) ans=min(ans,f[max_pos+1][i]);
return ans;
}
};