面试题 05.03. 翻转数位一般DP

Posted 2021-12-08 幽殇默

https://leetcode-cn.com/problems/reverse-bits-lcci/
状态表示:f[i][j] 当i==1时为全是1的且以j结尾的全是1的字符串的长度 。 当i==0时，表示有0的时候全是1的且以j结尾的全是1的字符串的长度 。
状态转移如下:

• 如果s[i]=='1' f[1][i]=f[1][i-1]+1; f[0][i]=f[0][i-1]+1;
• 如果s[i]=='0' f[0][i]=f[1][i-1]+1; f[1][i]=0;

class Solution {
public:
    int reverseBits(int num) 
    {
        string s;
        for(int i=0;i<32;i++)
        {
            if(num>>i&1) s+="1";
            else s+="0";
        }
        int f[2][35]={0};
        
        if(s[0]=='0') f[0][0]=1,f[1][0]=0;//初始化
        else f[0][0]=0,f[1][0]=1;
        
        int ans=0;
        for(int i=1;i<32;i++)
        {
            if(s[i]=='1')
            {
                f[1][i]=f[1][i-1]+1;
                f[0][i]=f[0][i-1]+1;
            }
            else
            {
                f[1][i]=0;
                f[0][i]=f[1][i-1]+1;
            }
        }
        for(int i=0;i<32;i++) ans=max({ans,f[0][i],f[1][i]});//所有结果取一个max即可
        return ans;
    }
};