Leetcode LCP 53. 守护太空城 DP 二进制状态压缩

首先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;
    }
};