京东2025届秋招 算法开发工程师 第2批笔试

⏰ 时间：2024/08/17
🔄 输入输出：ACM格式
⏳ 时长：2h

本试卷还有选择题部分，但这部分比较简单就不再展示。

1. 第一题

村子里有一些桩子，从左到右高度依次为 $1,1+2,1+2+3,...$，每两颗桩子之间的间隔为 $1$。现在下了一场大雪，但是不知道雪下了多厚，现在给你两个数字，这是雪后某相邻两个桩子在雪面上的高度，请你通过这两个数字计算雪的厚度。

输入描述

在一行中输入两个正整数 $a,b$，$1\le a<b\le 5\cdot 10^5$。

输出描述

在一行中输出一个整数代表雪的厚度。我们可以证明，答案一定存在。

题解

只需注意到 $a_k-a_{k-1}=k$，于是可以用 $b-a$ 定位到右边柱子的下标，计算它的高度即可。

#include <bits/stdc++.h>using namespace std;
using i64 = long long;int main() {i64 a ,b;cin >> a >> b;i64 k = b - a;i64 old_h = k * (k + 1) / 2;cout << old_h - b << endl;return 0;
}

2. 第二题

牛牛有一种锯齿状的积木，这种积木比较长，但是每个单位长度的高度是相等的，高度为 $1$ 或者 $2$。
现在牛牛拿出了两块长度分别为 $n$ 和 $m$ 的积木，她现在想把这两块积木拼接在一起，即使中间有空隙也没有关系。但是拼接后的积木的高度要不超过 $3$，请你帮助牛牛计算在满足这个前提下拼接后的积木的长度最短可以是多少。
例如有两块形状如下的积木:

输入描述

第一行给出两个正整数 $n,m$，代表第一块和第二块积木的长度。
第二行给出 $n$ 个数字代表第一块积木每个单位的高度。
第三行给出 $m$ 个数字代表第二块积木每个单位的高度 $1\le n,m\le 1000$。

输出描述

在一行中输出一个正整数代表拼接后积木的最短长度

题解

本题直接模拟即可。

说白了就是给定 $a$ 和 $b$ 两个数组，初始时先让 $a$ 和 $b$ 的左端对齐。

• 先让 $a$ 在 $b$ 上向右滑动，计算两者重叠部分的按位元素和，只要重叠部分中有一个位置出现了 $>3$ 的情况，说明当前拼接是无效的，继续右移。我们自然是希望 $a$ 右移的距离尽可能地小。
• 再让 $b$ 在 $a$ 上向右滑动，同样是找到正确的拼接位置，并使得 $b$ 右移的距离尽可能小。
• 还要考虑一种特殊情况就是无法上下拼接，此时需要将 $a$ 和 $b$ 直接横向连接，长度为 $m+n$。

#include <bits/stdc++.h>using namespace std;int min_len(const string& a, const string& b) {int n = a.length(), m = b.length();// a在b上滑动for (int i = 0; i < m; i++) {bool valid = true;for (int j = i; j < i + n && j < m; j++) {if (a[j - i] - '0' + b[j] - '0' > 3) {valid = false;break;}}if (valid)return max(m, i + n);}return m + n;
}int main() {int n, m;cin >> n >> m;string a, b;cin >> a >> b;cout << min(min_len(a, b), min_len(b, a)) << endl;return 0;
}

3. 第三题

牛牛也是要回家过年的呢。
牛牛所在的国家有 $n$ 座城市，$m$ 条有向道路，第 $i$ 条道路由城市 $u_i$ 通往城市 $v_i$，通行费为 $w_i$。
作为一头豪气的牛，希望他回家的花费是一个特殊的数字(例如666元)。具体的说，牛牛希望从城市 $1$ 移动到城市 $n$，并恰好花费 $a$ 元。
请你告诉牛牛，他有多少种回家的方案?

输入描述

第一行三个整数 $n,m,a(1\le n\le 100,1\le m\le 1000,1\le a\le 1000)$，含义如题面所示。
接下来 $m$ 行，第 $i$ 行三个整数 $u_i,v_i,w_i(1\le u_i,v_i\le n,1\le w_i\le a)$，描述了一条道路。

输出描述

如果牛牛回家的方案数大于等于 $20220201$ 种，请你在第一行输出 All roads lead to Home!，然后在第二行输出回家的方案数对 $20220201$ 取模的结果。
否则只需要输出一行一个整数，表示牛牛回家的方案数。

题解

本题使用动态规划求解。

设 $dp[i][j]$ 为从城市 $1$ 到城市 $i$，花费恰好为 $j$ 的方案数。由于从 $1$ 到 $1$ 花费为 $0$，也算是一种方案，因此 $dp[1][0]=1$，其他地方为 $0$。

假设 $u\stackrel{w}{\to }v$，且在城市 $u$ 的时候已经花费了 $j$，那么有如下的转移方程

$$dp[v][j+w]=dp[u][j]+dp[v][j+w]$$

注意，在计算 $v$ 的时候，我们必须要保证 $u$ 已经计算完毕，因此要按照拓扑序进行计算。

#include <bits/stdc++.h>using namespace std;const int MOD = 20220201;int main() {ios::sync_with_stdio(false);cin.tie(nullptr);int n, m, a;cin >> n >> m >> a;vector<vector<pair<int, int>>> graph(n + 1);vector<int> in(n + 1, 0);for (int i = 0; i < m; i++) {int u, v, w;cin >> u >> v >> w;graph[u].emplace_back(v, w);in[v]++;}vector<vector<int>> dp(n + 1, vector<int>(a + 1, 0));vector<bool> st(n + 1, 0);dp[1][0] = 1;queue<int> q;for (int i = 1; i <= n; i++) {if (!in[i]) q.push(i);}while (!q.empty()) {auto u = q.front();q.pop();for (auto &[v, w]: graph[u]) {if (!--in[v]) q.push(v);for (int j = 0; j + w <= a; j++) {if (dp[v][j + w] + dp[u][j] >= MOD || st[u]) {st[v] = true;}dp[v][j + w] = (dp[v][j + w] + dp[u][j]) % MOD;}}}if (st[n]) cout << "All roads lead to Home!" << endl;cout << dp[n][a] << endl;return 0;
}