problem1 link
首先按照type分类,同一类如果都是负数,那么取最大值,否则将所有的正数加起来作为这个type的价值。然后就是二维的背包。
problem2 link
从小到大将每个数字分到A或者B集合。设$f[i][j][m]$表示已经分配完前$i$个数字,A集合中分配了$j$个数字,已经分配的前$i$个数字中的最大的$K$个数字(即$[i-K+1,i]$)在两个集合中的分布情况为$m$的划分方案数。$m$是一个$K$位二进制数字。
problem3 link
设原来的数组为$P$,最终的数组为$S$.
有两个重要的性质:(1)$S$中的数字的前后顺序与其在$P$中的前后顺序是一致的,比如$P=[3,2,1,4,5]$不会使得$S=[2,3,1,4,5]$;(2)$S$中不会出现两段相同的数字,比如$S=[3,2,3,4,5]$
所以只需要挨个确定$S$中的每个数字,并记录$S$中最后确定的数字是$P$中的哪一个即可。
设$f[i][j][k][t]$表示已经确定好了$S$的前$i$个数字,其中$S[i]=P[j]$.并且已经使用了$k$次操作。$t$是一个tag(可以取0或者1两个值),它表示当用$j$去扩展到$i$时是否使用了一次操作。这个的意义在于如果后面还要用$P[j]$去扩展到$S[i+1]$,如果前面已经使用了一次操作,那么后面的操作数就不用再加1了。
code for problem1
#include <limits>
#include <unordered_map>
#include <vector>
class AlienAndHamburgers {
public:
int getNumber(const std::vector<int> &type, const std::vector<int> &taste) {
std::unordered_map<int, int> a;
for (size_t i = 0; i < type.size(); ++i) {
int t = type[i];
int c = taste[i];
if (a.find(t) == a.end()) {
a[t] = c;
continue;
}
if (c < 0) {
if (a[t] < 0) {
a[t] = std::max(a[t], c);
}
} else {
a[t] = std::max(a[t] + c, c);
}
}
int m = static_cast<int>(a.size());
std::vector<std::vector<int>> f(
m, std::vector<int>(m + 1, std::numeric_limits<int>::lowest()));
int idx = 0;
for (const auto &e : a) {
int c = e.second;
if (idx == 0) {
f[0][0] = 0;
f[0][1] = c;
++idx;
continue;
}
for (int j = 0; j <= idx; ++j) {
f[idx][j] = std::max(f[idx][j], f[idx - 1][j]);
f[idx][j + 1] = std::max(f[idx][j + 1], f[idx - 1][j] + c);
}
++idx;
}
int result = 0;
for (int i = 1; i <= idx; ++i) {
result = std::max(result, i * f[idx - 1][i]);
}
return result;
}
};
code for problem2
#include <cmath>
#include <vector>
class AlienAndSetDiv1 {
static constexpr int kMod = 1000000007;
public:
int getNumber(int N, int K) {
std::vector<std::vector<std::vector<int>>> f(
N + N + 1,
std::vector<std::vector<int>>(N + 1, std::vector<int>(1 << K)));
if (K >= N + N) {
return AnySplit(N);
}
for (int i = 0; i < (1 << K); ++i) {
std::vector<int> a, b;
for (int j = 0; j < K; ++j) {
if ((i & (1 << j)) == 0) {
a.push_back(j + 1);
} else {
b.push_back(j + 1);
}
}
bool tag = true;
for (std::size_t j = 0; j < a.size() && j < b.size(); ++j) {
if (std::abs(a[j] - b[j]) < K) {
tag = false;
break;
}
}
if (tag) {
f[K][a.size()][i] += 1;
}
}
std::vector<int> bit_num(1 << K);
for (int i = 1; i < (1 << K); ++i) {
bit_num[i] = bit_num[i >> 1] + (i & 1);
}
std::vector<std::vector<int>> indices0(1 << K, std::vector<int>(K + 1));
std::vector<std::vector<int>> indices1(1 << K, std::vector<int>(K + 1));
for (int i = 0; i < (1 << K); ++i) {
int num0 = 0;
int num1 = 0;
for (int j = 0; j < K; ++j) {
if ((i & (1 << j)) == 0) {
++num0;
indices0[i][num0] = j;
} else {
++num1;
indices1[i][num1] = j;
}
}
}
for (int i = K + 1; i <= N + N; ++i) {
for (int a = 0; a < i && a <= N; ++a) {
int b = i - 1 - a;
for (int k = 0; k < (1 << K); ++k) {
if (f[i - 1][a][k] != 0) {
if (a + 1 <= N) {
if (a >= b) {
(f[i][a + 1][k >> 1] += f[i - 1][a][k]) %= kMod;
} else {
int t = bit_num[k];
if ((t < b - a) || (indices1[k][t - (b - a) + 1] == 0)) {
(f[i][a + 1][k >> 1] += f[i - 1][a][k]) %= kMod;
}
}
}
if (b + 1 <= N) {
if (a <= b) {
(f[i][a][(k >> 1) | (1 << (K - 1))] += f[i - 1][a][k]) %= kMod;
} else {
int t = K - bit_num[k];
if ((t < a - b) || (indices0[k][t - (a - b) + 1] == 0)) {
(f[i][a][(k >> 1) | (1 << (K - 1))] += f[i - 1][a][k]) %=
kMod;
}
}
}
}
}
}
}
int result = 0;
for (int i = 0; i < (1 << K); ++i) {
(result += f[N + N][N][i]) %= kMod;
}
return result;
}
private:
int AnySplit(int n) {
std::vector<std::vector<int>> c(2 * n + 1, std::vector<int>(2 * n + 1, 0));
c[0][0] = c[1][0] = c[1][1] = 1;
for (int i = 2; i <= 2 * n; ++i) {
c[i][0] = c[i][i] = 1;
for (int j = 1; j < i; ++j) {
c[i][j] = (c[i - 1][j - 1] + c[i - 1][j]) % kMod;
}
}
return c[n + n][n];
}
};
code for problem3
#include <vector>
#include <cstring>
int f[200][200][201][2];
class AlienAndPermutation {
public:
int getNumber(const std::vector<int> &P, int K) {
if (K == 0) {
return 1;
}
int n = static_cast<int>(P.size());
std::vector<std::vector<bool>> tag(n, std::vector<bool>(n, true));
for (int i = 0; i < n; ++i) {
for (int j = 0; j < n; ++j) {
for (int k = std::min(i, j), stop = std::max(i, j); k <= stop; ++k) {
if (P[k] > P[i]) {
tag[i][j] = false;
break;
}
}
}
}
memset(f, -1, sizeof(f));
return dfs(0, 0, K, 0, n, tag);
}
private:
int dfs(int i, int j, int k, int t, int n,
const std::vector<std::vector<bool>> &tag) {
constexpr int kMod = 1000000007;
if (k < 0) {
return 0;
}
if (j == n) {
return 1;
}
if (i == n) {
return 0;
}
int &result = f[i][j][k][t];
if (result != -1) {
return result;
}
result = dfs(i + 1, j, k, 0, n, tag);
if (tag[i][j]) {
int new_k = k - ((i != j && t == 0) ? 1 : 0);
int new_t = (t != 0 || i != j) ? 1 : 0;
(result += dfs(i, j + 1, new_k, new_t, n, tag)) %= kMod;
}
return result;
}
};