标签:back 重要 二维 abs 数加 mat tag false comm
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; } };
标签:back 重要 二维 abs 数加 mat tag false comm
原文地址:https://www.cnblogs.com/jianglangcaijin/p/9463711.html