线性dp、区间dp、树形dp

线性dp:(从头开始到一个节点)

https://www.starrycoding.com/problem/80

// Problem: ${json.name}
// Contest: ${json.group}
// URL: ${json.url}
// Memory Limit: ${json.memoryLimit} MB
// Time Limit: ${json.timeLimit} ms
// More thought is needed

#include <bits/stdc++.h>
#define endl '\n'
#define int long long
using namespace std;
const int N = 1e6 + 7;
int a[105];
int dp[105][105];
void solve() {
  int n, m;
  cin >> n >> m;
  for (int i = 1; i <= n; i++) cin >> a[i];
  dp[0][0] = 1;
  for (int i = 1; i <= n; i++) {
    for (int j = 0; j <= m; j++) {
      for (int k = 0; k <= a[i] && k <= j; k++) {
        dp[i][j] = (dp[i][j] + dp[i - 1][j - k]) % N;
      }
    }
  }
  cout << dp[n][m];
}

signed main() {
  ios::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);
  solve();
  return 0;
}

区间dp:(从某一个节点开始到某一个节点)

https://www.starrycoding.com/problem/114

// Problem: ${json.name}
// Contest: ${json.group}
// URL: ${json.url}
// Memory Limit: ${json.memoryLimit} MB
// Time Limit: ${json.timeLimit} ms
// More thought is needed

#include <bits/stdc++.h>
#define endl '\n'
#define int long long
using namespace std;
const int N = 3e2 + 5, inf = 4e18;
int a[N], prefix[N];
int dp[N][N];
void solve() {
  int n;
  cin >> n;
  for (int i = 1; i <= n; i++) {
    cin >> a[i];
  }
  for (int i = 1; i <= n; i++) {
    prefix[i] = prefix[i - 1] + a[i];
  }
  for (int len = 2; len <= n; len++) {
    for (int i = 1, j = i + len - 1; j <= n; i++, j++) {
      dp[i][j] = inf;
      for (int k = i; k < j; k++) {
        dp[i][j] =
            min(dp[i][k] + dp[k + 1][j] + prefix[j] - prefix[i - 1], dp[i][j]);
      }
    }
  }
  cout << dp[1][n];
}

signed main() {
  ios::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);
  solve();
  return 0;
}

树形dp:

https://www.starrycoding.com/submission/35278

// Problem: ${json.name}
// Contest: ${json.group}
// URL: ${json.url}
// Memory Limit: ${json.memoryLimit} MB
// Time Limit: ${json.timeLimit} ms
// More thought is needed

#include <bits/stdc++.h>
#define endl '\n'
#define int long long
using namespace std;
const int N = 1e5 + 9;
const int inf = 4e18, p = 1e6 + 7;
int w[N], dp[N];
vector<int> g[N];
void dfs(int x, int pre) {
  dp[x] = w[x];
  for (auto &y : g[x]) {
    if (y == pre) continue;
    dfs(y, x);
    dp[x] = max(dp[x], dp[x] + dp[y]);
  }
}
void solve() {
  int n;
  cin >> n;
  for (int i = 1; i <= n; i++) cin >> w[i];
  for (int i = 1; i <= n; i++) {
    dp[i] = 0;
    g[i].clear();
  }
  for (int i = 1; i < n; i++) {
    int u, v;
    cin >> u >> v;
    g[u].push_back(v), g[v].push_back(u);
  }
  dfs(1, -1);
  cout << *max_element(dp + 1, dp + 1 + n);
}

signed main() {
  ios::sync_with_stdio(false), cin.tie(nullptr), cout.tie(nullptr);
  int _;
  cin >> _;
  while (_--) solve();
  return 0;
}

发表评论

您的邮箱地址不会被公开。 必填项已用 * 标注

滚动至顶部