LeetCode5690. 最接近目标价格的甜点成本

题目

分析

自己想着dp,结果比赛一直卡住。今天看y总直播又学习到了。本题直接暴力搜索就可，用四进制搜。从4^0——4^m减1，并且要排除掉每一位上3的可能。简而言之，就是将一个配料方案看成一个m位的四进制数

代码

class Solution {
public:
    int closestCost(vector<int>& a, vector<int>& b, int T) {
        int res = INT_MAX;
        int n = a.size(), m = b.size();
        for(int i = 0;i < n;i++){
            int s = a[i];  //主料
            for(int j = 0;j < 1<<m*2;j++){  //暴搜4^0~4^m-1
                int r = s;
                bool flag= false;
                for(int k = 0;k < m;k++){ //找到不合法的
                    int t = j >> k * 2 & 3;  //四进制扣数
                    if(t == 3){   //每种配料选择为0、1、2，3不合法
                        flag = true;
                        break;
                    }
                    r += b[k] * t;
                }
                if(flag == true) continue;
                if(abs(r - T) < abs(res - T) || (abs(r-T) == abs(res - T) && r < res))
                    res = r;
            }
        }
        return res;
    }
};

1.  j < 1<<m*2 本操作就是四进制最大值,*2表示两位一组

2.  t = j >> k * 2 & 3; //四进制扣数