1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87
| const int N = 10001, L = 18, M = 11; const int INF = 1000000000;
class Solution { private: int v[L][N], log[N], mi[L][N]; int f[N];
int get_v(int l, int r) { int z = log[r - l];
return v[z][l] & v[z][r - (1 << z) + 1]; }
int binary_search(int ed, int target) { int l = 1, r = ed + 1;
while (l < r) { int mid = (l + r) >> 1; if (get_v(mid, ed) < target) l = mid + 1; else r = mid; }
return l; }
int get_min(int l, int r) { int z = log[r - l];
return min(mi[z][l], mi[z][r - (1 << z) + 1]); }
public: int minimumValueSum(vector<int>& nums, vector<int>& andValues) { const int n = nums.size(), m = andValues.size();
log[0] = 0; for (int i = 1; i <= n; i++) { v[0][i] = nums[i - 1]; log[i] = (i >> (log[i - 1] + 1)) ? log[i - 1] + 1 : log[i - 1]; }
for (int j = 1; (1 << j) <= n; j++) for (int i = 1; i + (1 << j) - 1 <= n; i++) v[j][i] = v[j - 1][i] & v[j - 1][i + (1 << (j - 1))];
for (int i = 1; i <= n; i++) mi[0][i] = f[i] = INF;
mi[0][0] = f[0] = 0;
for (int k = 1; (1 << k) <= n; k++) for (int i = 0; i + (1 << k) - 1 <= n; i++) mi[k][i] = min(mi[k - 1][i], mi[k - 1][i + (1 << (k - 1))]);
for (int j = 0; j < m; j++) { f[0] = INF; for (int i = 1; i <= n; i++) { int p1 = binary_search(i, andValues[j]); int p2 = binary_search(i, andValues[j] + 1);
if (p1 < p2) f[i] = get_min(p1 - 1, p2 - 2) + nums[i - 1]; else f[i] = INF; }
for (int i = 0; i <= n; i++) { mi[0][i] = f[i]; f[i] = INF; }
for (int k = 1; (1 << k) <= n; k++) for (int i = 0; i + (1 << k) - 1 <= n; i++) mi[k][i] = min(mi[k - 1][i], mi[k - 1][i + (1 << (k - 1))]); }
if (mi[0][n] >= INF) return -1;
return mi[0][n]; } };
|