B. 序列删除

题目描述

有一个长度为 \(2N\) 的序列 \(A\)，其中 \(1\) 到 \(N\) 恰好出现两次。你每次可以选择两个相同的数 \(A_l,A_r(l<r)\) 并花费 \(r-l\) 的代价将其删除。求将整个序列删空的最小代价。

思路

有一个很显然的贪心就是：每次取代价最小的两个数删除。所以我们按照代价排序，用树状数组维护区间中未被删除的数的数量即可。

空间复杂度 \(O(N)\)，时间复杂度 \(O(N\log N)\)。

代码

#include<bits/stdc++.h>
using namespace std;
using ll = long long;

const int MAXN = 200001, MAXV = (1 << 20);

struct Node {
  int l, r;
}s[MAXN];

int n, a[MAXN], l[MAXN], tr[MAXN];
ll ans;

void update(int p, int x) {
  for(; p <= 2 * n; tr[p] += x, p += (p & -p)) {
  }
}

int Getsum(int p) {
  int sum = 0;
  for(; p; sum += tr[p], p -= (p & -p)) {
  }
  return sum;
}

int main() {
  ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
  cin >> n;
  for(int i = 1; i <= 2 * n; ++i) {
    cin >> a[i];
    if(!l[a[i]]) {
      l[a[i]] = i;
    }else {
      s[a[i]] = {l[a[i]], i};
    }
  }
  sort(s + 1, s + n + 1, [](const Node &a, const Node &b) -> bool {
    return a.r - a.l < b.r - b.l;
  });
  for(int i = 1; i <= n; ++i) {
    ans += s[i].r - s[i].l - Getsum(s[i].r) + Getsum(s[i].l);
    update(s[i].l, 1), update(s[i].r, 1);
  }
  cout << ans;
  return 0;
}

C. 史莱姆融合

题目描述

在一棵 \(N\) 个结点的树上，有 \(k\) 只史莱姆，一开始在结点 \(id_i\)，体积为 \(v_i\)。它们会如下行动：

• 如果一只史莱姆通过了边权为 \(w\) 的边，则其体积减 \(w\)。若其体积变为非正数，则其消失。
• 如果多只史莱姆待在同一个结点，则它们将会合并为体积为其体积之和的史莱姆。
• 史莱姆最终都要到达一个结点 \(c\)，但它们会按最优方案行动。

有 \(q\) 次查询，每次查询如果史莱姆要到达 \(c\)，且你一开始可以让 \(num(num\in \{0,1\})\) 个史莱姆消失时最终史莱姆的最小体积。

思路

我们先只考虑 \(c=1\) 的情况，那么我们可以令 \(dp_{0/1,u}\) 表示在 \(u\) 子树内删除了 \(0/1\) 个史莱姆，最终子树内的史莱姆到达 \(u\) 的体积。对于 \(dp_{0,u}\)，显然有转移 \(dp_{0,u}=\sum \limits_{(v,w)\in son_u} \max(0,dp_{0,v}-w)\)。而对于 \(dp_{1,u}\)，我们可以选择删掉这个结点本身的，也可以选择一个儿子子树删，也就是 \(dp_{1,u}=\min (\max(0,dp_{0,u}-w),\min\limits_{v_0\in son_u} \{\max(0,dp_{1,v_0})+\sum \limits_{v\in son_u\and v\ne v_0}\max(0,dp_{0,v})\})\)。

接着我们考虑换根，\(dp_{0,u}\) 的换根很简单，但 \(dp_{1,u}\) 比较麻烦。如果 \(v\) 不是 \(dp_{1,u}\) 转移时最小的那个 \(v_0\)，那么可以直接转移。否则必须从次小值转移过来，所以这里还需要记录次小值。

空间复杂度 \(O(N)\)，时间复杂度 \(O(N+K+Q)\)。

代码

#include<bits/stdc++.h>
using namespace std;
using ll = long long;
using pii = pair<int, int>;

const int MAXN = 100001;
const ll INF = (ll)(1e18);

int n, k, q, a[MAXN], son[MAXN];
ll dp[3][MAXN];
vector<pii> e[MAXN];

void dfs(int u, int fa) {
  dp[0][u] = a[u];
  for(pii g : e[u]) {
    int v = g.first, w = g.second;
    if(v != fa) {
      dfs(v, u);
      dp[0][u] += max(0ll, dp[0][v] - w);
    }
  }
  dp[1][u] = dp[0][u] - a[u], dp[2][u] = INF, son[u] = u;
  for(pii g : e[u]) {
    int v = g.first, w = g.second;
    if(v != fa) {
      ll x = dp[0][u] - max(0ll, dp[0][v] - w) + max(0ll, dp[1][v] - w);
      if(x < dp[1][u]) {
        dp[2][u] = dp[1][u], dp[1][u] = x, son[u] = v;
      }else if(x < dp[2][u]) {
        dp[2][u] = x;
      }
    }
  }
}

void DFS(int u, int fa) {
  for(pii g : e[u]) {
    int v = g.first, w = g.second;
    if(v != fa) {
      ll v0 = dp[0][v], u0 = dp[0][u] - max(0ll, v0 - w);
      dp[0][v] += max(0ll, u0 - w);
      dp[1][v] += max(0ll, u0 - w);
      dp[2][v] += max(0ll, u0 - w);
      if(son[u] != v) {
        ll x = dp[0][v] - max(0ll, u0 - w) + max(0ll, dp[1][u] - max(0ll, v0 - w) - w);
        if(x < dp[1][v]) {
          dp[2][v] = dp[1][v], dp[1][v] = x, son[v] = u;
        }else if(x < dp[2][v]) {
          dp[2][v] = x;
        }
      }else {
        ll x = dp[0][v] - max(0ll, u0 - w) + max(0ll, dp[2][u] - max(0ll, v0 - w) - w);
        if(x < dp[1][v]) {
          dp[2][v] = dp[1][v], dp[1][v] = x, son[v] = u;
        }else if(x < dp[2][v]) {
          dp[2][v] = x;
        }
      }
      DFS(v, u);
    }
  }
}

int main() {
  ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
  cin >> n >> k >> q;
  for(int i = 1, u, v, w; i < n; ++i) {
    cin >> u >> v >> w;
    e[u].emplace_back(v, w);
    e[v].emplace_back(u, w);
  }
  for(int i = 1, u; i <= k; ++i) {
    cin >> u >> a[u];
  }
  dfs(1, 0);
  DFS(1, 0);
  for(int i = 1, u, v; i <= q; ++i) {
    cin >> u >> v;
    cout << dp[v][u] << "\n";
  }
  return 0;
}