比赛链接:Here
AcWing 3626. 三元一次方程
暴力即可
void solve() {
int n;
cin >> n;
for (int i = 0; i <= n / 3; ++i)
for (int j = 0; j <= n / 5; ++j)
for (int k = 0; k <= n / 7; ++k)
if (i * 3 + j * 5 + k * 7 == n) {
cout << i << " " << j << " " << k << "\n";
return ;
}
cout << -1 << "\n";
}
AcWing 3627. 最大差值
把次k最大值累加到最大值即可
void solve() {
int n, k; cin >> n >> k;
vector<ll>a(n);
for (ll &x : a)cin >> x;
sort(a.begin(), a.end());
int i = n - 2;
while (k and i >= 0) {
if (a[i] == 0) {i--; continue;}
a[n - 1] += a[i];
i--, k--;
}
cout << a[n - 1] << "\n";
}
AcWing 3628. 边的删减
本题给定一个无向连通加权图,从第 1 个点出发到第 i 个点的最短距离记为 di
接着我们需要进行操作,删掉图中的一些边,最多保留原图中的 k 条边
删完边后,如果点 i 到点 1 的距离任然是原图的中最短路长度,则我们称该点为优秀点
我们需要找出一种删边方案,使得最终图中有尽可能多的 优秀点
分析
对于初始图,我们需要得到每个点 i 到起始点 1 的最短路径长度,这毫无疑问会让我们想到最短路算法
因此,一开始我们需要挑选一个最短路算法来求每个点到起点的最短路径
接着看下一个要求,我们要找出一个保留小于等于 k 条边的方案,使得构成最短路的点尽可能多
为了保证我们保留下来的边是有价值的,因此我们保留的边一定是更新出最短路的边
而 \(Dijkstra\)算法 的贪心思想是:每次更新到起点距离最短的点的最短路长度
利用该性质,我们知道:
\(Dijkstra\)算法 对于每轮更新出最短路的点来说,最后一次更新他距离的边一定是 有价值的
这些有价值的边,连成的点就构成了一个最短路树
因此我们直接用 Dijkstra 求一遍最短路的同时,把每个点最后一次被更新最短距离的边都存下来
于是被保留下来边,一定是有价值的边
从前往后(不从前往后保留,就不在一个连通块里,那这样的边也是没价值的),把这些边保留下来即可
注意:用 long long
存路径长度
#define x first
#define y second
using LL = long long;
typedef pair<LL, int> PII;
const int N = 100010, M = 200010;
int n, m, k;
int h[N], e[M], w[M], id[M], ne[M], idx;
LL dist[N];
bool st[N];
vector<int> ans;
void add(int a, int b, int c, int d) { // 添加一条边a->b,边权为c
e[idx] = b, w[idx] = c, id[idx] = d, ne[idx] = h[a], h[a] = idx ++ ;
}
void dijkstra() { // 求1号点到n号点的最短路距离
memset(dist, 0x3f, sizeof dist);
dist[1] = 0;
priority_queue<PII, vector<PII>, greater<PII>> heap;
heap.push({0, 1});
while (heap.size()) {
auto t = heap.top();
heap.pop();
int ver = t.second, distance = t.first;
if (st[ver]) continue;
st[ver] = true;
for (int i = h[ver]; i != -1; i = ne[i]) {
int j = e[i];
if (dist[j] > dist[ver] + w[i]) {
dist[j] = dist[ver] + w[i];
heap.push({dist[j], j});
}
}
}
}
void dfs(int u) {
st[u] = true;
for (int i = h[u]; ~i; i = ne[i]) {
int j = e[i];
if (!st[j] && dist[j] == dist[u] + w[i]) {
if (ans.size() < k) ans.push_back(id[i]);
dfs(j);
}
}
}
int main() {
cin >> n >> m >> k;
memset(h, -1, sizeof h);
for (int i = 1; i <= m; i ++ ) {
int a, b, c;
cin >> a >> b >> c;
add(a, b, c, i), add(b, a, c, i);
}
dijkstra();
memset(st, 0, sizeof st);
dfs(1);
printf("%d\n", ans.size());
for (auto x : ans) printf("%d ", x);
return 0;
}