【Leetcode】1774. Closest Dessert Cost

题目地址：

https://leetcode.com/problems/closest-dessert-cost/description/

给定两个数组$A$和$B$，分别长$n,m$。$A$代表冰淇淋每个主料的价格，$B$代表每个配料的价格。要做一个冰淇淋，必须恰好含一个主料，配料可以含也可以不含，每个配料最多含$2$个单位。再给定一个目标值$t$，问能做出来的价格离$t$最近的冰淇淋价格是多少。如果答案不唯一，取价格低的那个。

先暴力枚举所有配料的所有组合的价格，可以用二进制枚举。然后对主料和配料的所有可能价格都排个序，接下来就成为双指针问题了。代码如下：

class Solution {
 public:
  int closestCost(vector<int>& a, vector<int>& top, int t) {
    int n = a.size(), m = top.size();
    vector<int> b;
    for (int j = 0; j < 1 << m * 2; j++) {
      int cost = 0;
      bool valid = true;
      for (int k = 0; k < m; k++) {
        int cnt = j >> k * 2 & 3;
        if (cnt > 2) {
          valid = false;
          break;
        }
        cost += cnt * top[k];
      }

      if (!valid) continue;
      b.push_back(cost);
    }

    sort(a.begin(), a.end());
    sort(b.begin(), b.end());
    int res = INT_MAX;
    for (int i = 0, j = b.size() - 1; i < n && j >= 0;) {
      int x = a[i] + b[j];
      if (abs(x - t) < abs(res - t) || abs(x - t) == abs(res - t) && x < res)
        res = x;
      if (a[i] + b[j] > t)
        j--;
      else if (a[i] + b[j] < t)
        i++;
      else
        break;
    }

    return res;
  }
};

时间复杂度$O(n\log n+3^{m}m)$，空间$O(n+3^m)$。