Codeforces Round #635 (Div. 1)

传送门

A. Linova and Kingdom

题意：

给定一颗以\(1\)为根的树，现在要选定\(k\)个结点为黑点，一个黑点的贡献为从他出发到根节点经过的白点数量。

问黑点贡献总和最大为多少。

思路：

• 最直接的想法黑点肯定是位于深度越深的点越好，并且有这样一个性质：假设我们选择了一个点，那么该点的所有后代也将会被选择。
• 一个黑点产生的贡献为其深度减去到根路径中黑点的数量。
• 直接按照上述思路贪心不好思考，考虑转化一下贡献的计算方法：我们减去黑点的数量时在其祖先结点再减。也就是说每个黑点会减去其子树\(size\)。
• 那么对于所有点按照\(deep-size\)进行排序，从大到小贪心选即可。

因为一个点选择了所有子树也会被选，并且显然子树的\(deep-size\)更大，所以容易证明这种贪心是正确的。

Code

/*
 * Author:  heyuhhh
 * Created Time:  2020/4/15 22:36:55
 */
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <vector>
#include <cmath>
#include <set>
#include <map>
#include <queue>
#include <iomanip>
#include <assert.h>
#define MP make_pair
#define fi first
#define se second
#define pb push_back
#define sz(x) (int)(x).size()
#define all(x) (x).begin(), (x).end()
#define INF 0x3f3f3f3f
#define Local
#ifdef Local
  #define dbg(args...) do { cout << #args << " -> "; err(args); } while (0)
  void err() { std::cout << std::endl; }
  template<typename T, typename...Args>
  void err(T a, Args...args) { std::cout << a << ' '; err(args...); }
  template <template<typename...> class T, typename t, typename... A>
  void err(const T <t> &arg, const A&... args) {
  for (auto &v : arg) std::cout << v << ' '; err(args...); }
#else
  #define dbg(...)
#endif
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
//head
const int N = 2e5 + 5;

int n, k;
vector <int> G[N];
int d[N], sz[N];

void dfs(int u, int fa) {
    d[u] = d[fa] + 1;
    sz[u] = 1;
    for(auto v : G[u]) if(v != fa) {
        dfs(v, u);
        sz[u] += sz[v];
    }
}

void run() {
    cin >> n >> k;
    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, 0);
    vector <int> v;
    for(int i = 1; i <= n; i++) {
        v.push_back(d[i] - sz[i]);
    }
    sort(all(v));
    reverse(all(v));
    ll ans = 0;
    for(int i = 0; i < k; i++) ans += v[i];
    cout << ans << '\n';
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);
    cout << fixed << setprecision(20);
    run();
    return 0;
}

B. Xenia and Colorful Gems

枚举一维，然后另外两维直接贪心二分即可。

代码写得有点乱。。

Code

/*
 * Author:  heyuhhh
 * Created Time:  2020/4/15 22:47:16
 */
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <vector>
#include <cmath>
#include <set>
#include <map>
#include <queue>
#include <iomanip>
#include <assert.h>
#define MP make_pair
#define fi first
#define se second
#define pb push_back
#define sz(x) (int)(x).size()
#define all(x) (x).begin(), (x).end()
#define INF 0x3f3f3f3f
//#define Local
#ifdef Local
  #define dbg(args...) do { cout << #args << " -> "; err(args); } while (0)
  void err() { std::cout << std::endl; }
  template<typename T, typename...Args>
  void err(T a, Args...args) { std::cout << a << ' '; err(args...); }
  template <template<typename...> class T, typename t, typename... A>
  void err(const T <t> &arg, const A&... args) {
  for (auto &v : arg) std::cout << v << ' '; err(args...); }
#else
  #define dbg(...)
#endif
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
//head
const int N = 1e5 + 5;

int na, nb, nc;

ll a[N], b[N], c[N];

ll calc(int x, int y, int z) {
    if(x && y && z && x <= na && y <= nb && z <= nc)
        return (a[x] - b[y]) * (a[x] - b[y]) + (a[x] - c[z]) * (a[x] - c[z]) + (b[y] - c[z]) * (b[y] - c[z]);
    return 8e18;
}

ll calc_a(int y, int z) {
    int t = lower_bound(a + 1, a + na + 1, b[y]) - a;
    ll res = 9e18;
    res = min(res, calc(t, y, z));
    res = min(res, calc(t - 1, y, z));
    t = lower_bound(a + 1, a + na + 1, c[z]) - a;
    res = min(res, calc(t, y, z));
    res = min(res, calc(t - 1, y, z));
    return res;
}

ll calc_c(int x, int y) {
    int t = lower_bound(c + 1, c + nc + 1, b[y]) - c;
    ll res = 9e18;
    res = min(res, calc(x, y, t));
    dbg(x, y, t, res);
    res = min(res, calc(x, y, t - 1));
    t = lower_bound(c + 1, c + nc + 1, a[x]) - c;
    res = min(res, calc(x, y, t));
    res = min(res, calc(x, y, t - 1));
    return res;
}

ll calc_b(int x, int z) {
    int t = lower_bound(b + 1, b + nb + 1, c[z]) - b;
    ll res = 9e18;
    res = min(res, calc(x, t, z));
    res = min(res, calc(x, t - 1, z));
    t = lower_bound(b + 1, b + nb + 1, a[x]) - b;
    res = min(res, calc(x, t, z));
    res = min(res, calc(x, t - 1, z));
    return res;
}

void run() {
    cin >> na >> nb >> nc;
    for(int i = 1; i <= na; i++) cin >> a[i];
    for(int i = 1; i <= nb; i++) cin >> b[i];
    for(int i = 1; i <= nc; i++) cin >> c[i];
    sort(a + 1, a + na + 1);
    sort(b + 1, b + nb + 1);
    sort(c + 1, c + nc + 1);
    ll ans = 9e18;
    for(int i = 1, t; i <= na; i++) {
        t = lower_bound(b + 1, b + nb + 1, a[i]) - b;
        if(t <= nb) ans = min(ans, calc_c(i, t));
        if(t > 1) ans = min(ans, calc_c(i, t - 1));
        t = lower_bound(c + 1, c + nc + 1, a[i]) - c;
        if(t <= nc) ans = min(ans, calc_b(i, t));
        if(t > 1) ans = min(ans, calc_b(i, t - 1));
    }
    for(int i = 1, t; i <= nb; i++) {
        t = lower_bound(a + 1, a + na + 1, b[i]) - a;
        ans = min(ans, calc_c(t, i));
        ans = min(ans, calc_c(t - 1, i));
        t = lower_bound(c + 1, c + nc + 1, b[i]) - c;
        ans = min(ans, calc_a(i, t));
        ans = min(ans, calc_a(i, t - 1));
    }
    for(int i = 1, t; i <= nc; i++) {
        t = lower_bound(b + 1, b + nb + 1, c[i]) - b;
        ans = min(ans, calc_a(t, i));
        ans = min(ans, calc_c(t - 1, i));
        t = lower_bound(a + 1, a + na + 1, c[i]) - a;
        ans = min(ans, calc_b(t, i));
        ans = min(ans, calc_b(t - 1, i));
    }
    cout << ans << '\n';
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);
    cout << fixed << setprecision(20);
    int T; cin >> T;
    while(T--) run();
    return 0;
}

C. Kaavi and Magic Spell

题意：

给出串\(S,T\)，现在有一个空串\(A\)，然后每次执行以下操作：

• 选择\(S\)的第一个字符并删除，然后将其插入\(A\)的前面或者后面。

现在问有多少种方式使得串\(T\)为串\(A\)的前缀。

思路：

将\(T\)串用\(*\)补齐到与\(S\)长度相同，问题转化为了对于\(T\)的一个前缀，\(S\)有多少种匹配方法。

注意到有个性质：匹配任意一段\(T\)时，我们只会用到\(S\)的一个前缀。

那么我们直接跑个区间\(dp\)，表示匹配\(T[l,r]\)的方案数，转移时直接根据\(s[r-l+1]\)\(O(1)\)转移即可。

最后答案即为\(\sum_{i\geq m}{dp_{1,i}}\)。

细节见代码：

Code

/*
 * Author:  heyuhhh
 * Created Time:  2020/4/16 9:06:01
 */
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cstdio>
#include <vector>
#include <cmath>
#include <set>
#include <map>
#include <queue>
#include <iomanip>
#include <assert.h>
#define MP make_pair
#define fi first
#define se second
#define pb push_back
#define sz(x) (int)(x).size()
#define all(x) (x).begin(), (x).end()
#define INF 0x3f3f3f3f
#define Local
#ifdef Local
  #define dbg(args...) do { cout << #args << " -> "; err(args); } while (0)
  void err() { std::cout << std::endl; }
  template<typename T, typename...Args>
  void err(T a, Args...args) { std::cout << a << ' '; err(args...); }
  template <template<typename...> class T, typename t, typename... A>
  void err(const T <t> &arg, const A&... args) {
  for (auto &v : arg) std::cout << v << ' '; err(args...); }
#else
  #define dbg(...)
#endif
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
//head
const int N = 3000 + 5, MOD = 998244353;

int n, m;
char s[N], t[N];
int dp[N][N];

int solve(int l, int r) {
    if(l > r) return 1;
    if(dp[l][r] != -1) return dp[l][r];
    int k = r - l + 1;
    int res = 0;
    if(l > m || s[k] == t[l]) res += solve(l + 1, r);
    if(r > m || s[k] == t[r]) res = (res + solve(l, r - 1)) % MOD;
    return dp[l][r] = res;
}

void run() {
    cin >> (s + 1) >> (t + 1);
    n = strlen(s + 1), m = strlen(t + 1);
    memset(dp, -1, sizeof(dp));
    int ans = 0;
    for(int i = m; i <= n; i++) {
        ans = (ans + solve(1, i)) % MOD;
    }
    cout << ans << '\n';
}

int main() {
    ios::sync_with_stdio(false);
    cin.tie(0); cout.tie(0);
    cout << fixed << setprecision(20);
    run();
    return 0;
}