ch03 部分题目思路

G. 收集

由于稀有度相同的物品需要一起处理,我们先把他们聚集到一起。

类似这样:

cpp 复制代码
vector<int> g[maxn];
...
{
  cin >> x >> c;
  g[c].push_back(x);
}

那么我们需要一个贪心的思路:

  • 肯定是按 c c c 从小往大收集的;
  • 对于相同的 c c c 收集完最后一个肯定是要么停留最左边 要么停留在最右边

故设 d p ( i , 0 ) dp(i,0) dp(i,0) 表示收集完稀有度为 i i i 的物品后停留在最左边, d p ( i , 1 ) dp(i,1) dp(i,1) 则表示停留在最右边。

对于转移,则是讨论一下:

  • 收集完第 i − 1 i-1 i−1 层的物品后,在最左边还是最右边,将来要停留在这一层的最左边还是最右边,即 d p ( i − 1 , 0 / 1 ) dp(i-1,0/1) dp(i−1,0/1) 转移到 d p ( i , 0 / 1 ) dp(i,0/1) dp(i,0/1)。

注意的是,可能存在某一层是没有物品的,而下一层是有物品的,需要存储上一层的 c,设其为 pre。则状态转移使用 d p [ p r e ] [ ] dp[pre][] dp[pre][] 而不是 d p [ i − 1 ] [ ] dp[i-1][] dp[i−1][]

cpp 复制代码
#include <bits/stdc++.h>
using namespace std;
#define ll long long

ll a[200010][2], x;
ll dp[200010][2]; // dp[i][0]: 从小到大走;dp[i][1]:从大往小走
bool vis[200010];
int main() {
    int n, c;
    cin >> n;
    for (int i = 1; i <= n; ++i) {
        a[i][0] = INT_MAX; // c == i 的位置最小值
        a[i][1] = INT_MIN; // c == i 的位置最大值
    }
    for (int i = 1; i <= n; ++i) {
        cin >> x >> c;
        a[c][0] = min(a[c][0], x);
        a[c][1] = max(a[c][1], x);
        vis[c] = 1;
    }
    int p = 0; // 前一个 c
    vis[n + 1] = 1; // 最后一层回到位置 0, a[n+1][0] = a[n+1][1] = 0
    for (int i = 1; i <= n + 1; ++i) {
        if (!vis[i]) continue;
        for (int j = 0; j < 2; ++j) {
            dp[i][j] = min(abs(a[p][0] - a[i][j]) + dp[p][1], 
                abs(a[p][1] - a[i][j]) + dp[p][0]) + a[i][1] - a[i][0]; 
        }
        p = i;
    }
    cout << dp[n + 1][0] << endl;
    return 0;
}

H. 选择

对于每一个 i i i,我们考虑让 a i a_i ai 和 b i b_i bi 之间建一条边,则这些边之间形成了若干个环

则原问题等价于对于每个环,任意两条相邻的边至少选一条 ,不同的环之间没有限制

dp 算出每个环选点的方案数,然后再乘起来,就是总的方案数

  • 相信大家都会做一排 物品,相邻两件至少选一件,求方案数记作 f[i]

  • 现在处理环记作 g[i],将下面两个方案相加

    • 最后一个不选:f[i-3]
    • 最后一个选:f[i-1]

可以先递推计算 f,再递推计算 'g'

也可以整理得到: g[i] = f[i-3] + f[i-1] = (f[i-4]+f[i-5]) + (f[i-2]+f[i-3]) = (f[i-2]+f[i-4]) + (f[i-3]+f[i-5]) = g[i-1] + g[i-2]

cpp 复制代码
#include <bits/stdc++.h>
using namespace std;

using ll = long long;
const ll M = 998244353;
const int maxn = 2e5 + 10;

// 相邻的数至少选一个
// 线性:f[i] = f[i-2] + f[i-1]
// 环形:g[i] = f[i-3] + f[i-1] --> g[i] = g[i-1] + g[i-2]
ll g[maxn];
int a[maxn], bi, nxt[maxn];
bool vis[maxn];
int main() {
    g[1] = 1, g[2] = 3;
    for (int i = 3; i < maxn; ++i) {
        g[i] = (g[i - 1] + g[i - 2]) % M;
    }

    int n, x, b;
    cin >> n;
    for (int i = 1; i <= n; ++i) cin >> a[i];
    for (int i = 1; i <= n; ++i) {
        cin >> bi;
        nxt[a[i]] = bi;
    }
    ll res = 1;
    for (int i = 1; i <= n; ++i) {
        if (!vis[i]) {
            int p = i, cnt = 0;
            while (!vis[p]) {
                vis[p] = 1, ++cnt;
                p = nxt[p];
            }
            res = res * g[cnt] % M;
        }
    }
    cout << res << endl;
    return 0;
}
相关推荐
SirLancelot140 分钟前
数据结构-Set集合(一)Set集合介绍、优缺点
java·开发语言·数据结构·后端·算法·哈希算法·set
YouQian7721 小时前
label 拓扑排序
数据结构·算法
YouQian7721 小时前
(补题)小塔的饭
算法
歌者長門1 小时前
做题笔记:某大讯飞真题28道
java·数据结构·算法
是店小二呀1 小时前
【动态规划 | 多状态问题】动态规划求解多状态问题
算法·动态规划
竹子_232 小时前
《零基础入门AI:传统机器学习核心算法解析(KNN、模型调优与朴素贝叶斯)》
人工智能·算法·机器学习
boyedu2 小时前
哈希指针与数据结构:构建可信数字世界的基石
数据结构·算法·区块链·哈希算法·加密货币
✿ ༺ ོIT技术༻2 小时前
剑指offer第2版:双指针+排序+分治+滑动窗口
算法·排序算法·剑指offer·双指针·滑动窗口·分治
细嗅蔷薇@2 小时前
C语言在键盘上输入一个3行3列矩阵的各个元素的值(值为整数),然后输出主对角线元素的积,并在fun()函数中输出。
c语言·算法·矩阵
新时代苦力工3 小时前
桶排序-Java实现
数据结构·算法·排序算法