A.仙男仙女
解题思路:对于每一个 pi,看左右离它最近的两个即可,所以先对 p 排序,枚举每一个 p(提前复制一份数组)并二分它排序后的位置,再看两侧最近的是否满足条件即可。ac代码如下
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using db = double;
#define OKK(x) (x ? "Yes" : "No")
#define LF(x) fixed<<setprecision(x) // c++ 保留小数
const int mod = 1e9 + 7;
const int N = 1e5 + 10;
int a[N], b[N];
void solve() {
int n, ans = 0;
cin >> n;
for (int i = 0; i < n; i++) {
cin >> a[i];
b[i] = a[i];
}
sort(a, a + n);
for (int i = 0, x; i < n; i++) {
cin >> x;
int p = lower_bound(a, a + n, b[i]) - a;
if ((p > 0 && a[p - 1] >= b[i] - x) || (p < n - 1 && a[p + 1] <= b[i] + x)) continue;
ans++;
}
cout << ans;
}
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr), cout.tie(nullptr);
int t = 1;
// cin >> t;
while (t--) solve();
return 0;
}
B.牛郎的微信群
局部稠密图会超时,所以考虑从数学的角度分析,对于一个 u,考虑它的某个邻居 v 中有多少个满足是 u 的朋友的朋友,显然除去 u,还有 degree[v] - 1 个,degree 是每个点的度,建边时同时维护度数即可。
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using db = double;
#define OKK(x) (x ? "Yes" : "No")
#define LF(x) fixed<<setprecision(x) // c++ 保留小数
const int mod = 1e9 + 7;
const int N = 2e5 + 10;
int a[N], deg[N];
void solve() {
int n;
cin >> n;
vector<vector<int>> g(n);
for (int i = 1, u, v; i < n; i++) {
cin >> u >> v;
--u;
--v;
g[u].emplace_back(v);
g[v].emplace_back(u);
deg[u]++;
deg[v]++;
}
for (int i = 0; i < n; i++) {
int res = 0;
for (int j: g[i]) res += g[j].size() - 1;
cout << res << ' ';
}
}
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr), cout.tie(nullptr);
int t = 1;
// cin >> t;
while (t--) solve();
return 0;
}
C.婚礼
首先默认所有人都去当伴郎,为了最优,需选择 a[i] - b[i] 最大的前 n 个去做伴娘,所以需要知道每一人在按照 a[i] - b[i] 排序时排名第几,这可以利用一个索引数组实现,具体计算时,看这个人是不是在前 n 大之中,如果在,那么需要取出它的贡献添加第 n + 1 大的贡献,否则不变。
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using db = double;
#define OKK(x) (x ? "Yes" : "No")
#define LF(x) fixed<<setprecision(x) // c++ 保留小数
const int mod = 1e9 + 7;
const int N = 2e5 + 10;
int a[N], b[N], c[N], d[N];
void solve() {
int n, m;
ll res = 0, nw = 0;
cin >> n >> m;
priority_queue<int, vector<int>, greater<>> q;
for (int i = 0; i < n + m + 1; i++) cin >> a[i];
for (int i = 0; i < n + m + 1; i++) {
cin >> d[i];
res += d[i];
a[i] -= d[i];
}
iota(b + 0, b + n + m + 1, 0);
sort(b, b + n + m + 1, [&](int x, int y) {
return a[x] > a[y];
});
for (int i = 0; i < n; i++) {
nw += a[b[i]];
}
for (int i = 1; i < n + m + 1; i++) {
c[b[i]] = i;
}
for (int i = 0; i < n + m + 1; i++) {
if (c[i] < n) {
cout << res - d[i] + nw - a[i] + a[b[n]] << ' ';
} else {
cout << res - d[i] + nw << ' ';
}
}
}
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr), cout.tie(nullptr);
int t = 1;
// cin >> t;
while (t--) solve();
return 0;
}
D.久别重逢
考虑最终亲密度到达每一个值有几种可能最后统计即可,现在考虑一个值 x,看提升最后一步之前的亲密度的可能情况,最后一步最少加 k,所以范围应该在 [0, x - k],但有可能 x < k,所以和 0 取大去除不合法情况,且发现这些可取值是一个区间所以考虑同时维护前缀和,最后答案就是到达 n 的前缀和,为了处理空前缀和下标统计 +1。
#include <bits/stdc++.h>
using namespace std;
using ll = long long;
using db = double;
#define OKK(x) (x ? "Yes" : "No")
#define LF(x) fixed<<setprecision(x) // c++ 保留小数
const int mod = 1e9 + 7;
const int N = 1e5 + 10;
ll pre[N];
void solve() {
int n, k;
cin >> n >> k;
ll ans = 0;
pre[1] = 1;
for (int i = 1; i <= n; i++) {
pre[i + 1] = pre[max(i - k + 1, 0)] % mod;
pre[i + 1] += pre[i];
pre[i + 1] %= mod;
}
cout << pre[n + 1];
}
int main() {
ios::sync_with_stdio(false);
cin.tie(nullptr), cout.tie(nullptr);
int t = 1;
// cin >> t;
while (t--) solve();
return 0;
}
E.七夕城
自下而上处理每个点的子树内的数,用字典树保存,如果该位置是叶子节点则为 -1,否则从高位到低位匹配当前数,优先走该位相同的分支,如果没有则走不同的