GMOJ1.22模拟赛【NOIP2017提高组正式赛】宝藏
Description
参与考古挖掘的小明得到了一份藏宝图,藏宝图上标出了 n 个深埋在地下的宝藏屋,也给出了这 n 个宝藏屋之间可供开发的 m 条道路和它们的长度。
小明决心亲自前往挖掘所有宝藏屋中的宝藏。但是,每个宝藏屋距离地面都很远,也就是说,从地面打通一条到某个宝藏屋的道路是很困难的,而开发宝藏屋之间的道路则相对容易很多。
小明的决心感动了考古挖掘的赞助商,赞助商决定免费赞助他打通一条从地面到某个宝藏屋的通道,通往哪个宝藏屋则由小明来决定。
在此基础上,小明还需要考虑如何开凿宝藏屋之间的道路。已经开凿出的道路可以任意通行不消耗代价。每开凿出一条新道路,小明就会与考古队一起挖掘出由该条道路所能到达的宝藏屋的宝藏。另外,小明不想开发无用道路,即两个已经被挖掘过的宝藏屋之间的道路无需再开发。
新开发一条道路的代价是:
这条道路的长度 × 从赞助商帮你打通的宝藏屋到这条道路起点的宝藏屋所经过的宝藏屋的数量(包括赞助商帮你打通的宝藏屋和这条道路起点的宝藏屋)。
请你编写程序为小明选定由赞助商打通的宝藏屋和之后开凿的道路,使得工程总代价最小,并输出这个最小值。
Input
第一行两个用空格分离的正整数 n 和 m,代表宝藏屋的个数和道路数。
接下来 m 行,每行三个用空格分离的正整数,分别是由一条道路连接的两个宝藏屋的编号(编号为 1~n),和这条道路的长度 v。
Output
输出共一行,一个正整数,表示最小的总代价。
Sample Input
【输入样例 1】
4 5
1 2 1
1 3 3
1 4 1
2 3 4
3 4 1
【样例输入 2】
4 5
1 2 1
1 3 3
1 4 1
2 3 4
3 4 2
Sample Output
【输出样例 1】
4
【输入输出样例 1 说明】
小明选定让赞助商打通了 1 号宝藏屋。小明开发了道路 1->2,挖掘了 2 号宝藏。开发了道路 1->4,挖掘了 4 号宝藏。还开发了道路 4->3,挖掘了 3 号宝藏。工程总代价为:1 × 1 + 1 × 1 + 1 × 2 = 4
(1->2) (1->4) (4->3)
【样例输出 2】
5
【输入输出样例 2 说明】
小明选定让赞助商打通了 1 号宝藏屋。小明开发了道路 1->2,挖掘了 2 号宝藏。开发了道路 1->3,挖掘了 3 号宝藏。还开发了道路 1->4,挖掘了 4 号宝藏。工程总代价为:1 × 1 + 3 × 1 + 1 × 1 = 5
(1->2) (1->3) (1->4)
Data Constraint
对于 20%的数据:
保证输入是一棵树,1≤n≤8,v≤5000 且所有的 v 都相等。
对于 40%的数据:
1≤n≤8,0≤m≤1000,v≤5000 且所有的 v 都相等。
对于 70%的数据:
1≤n≤8,0≤m≤1000,v≤ 5000
对于 100%的数据:
1≤n≤12,0≤m≤1000,v≤ 500000
题解
本题数据非常水导致一大部分打错了的人都AC了。好在最后换了数据。
显然本题需要使用状态压缩。但是怎么压是个问题。我们需要记住每个点的深度。DP的做法可以参考LQX大佬。但是如果求的是最大值LQX大佬的方法是不行的(他自己也承认了)。因为只记录最深深度答案只会多不会少。所以最大值就应使用ZYS大佬的三进制状态压缩,每次只从叶子节点扩展。
总的来说DP方法是非常复杂的。所以我使用DFS!!!
我们枚举起点,记录状态和每个点的深度。然后暴力做法是显然的。
接下来考虑优化。(参考洛谷题解)
首先最优性剪枝是容易想到的
设当前答案为tot,除去已经连的点,剩下的点的最小连边之和为tmp。
那么对于一个点s 如果
d
e
p
s
×
t
m
p
>
=
a
n
s
dep_s \times tmp >=ans
deps×tmp>=ans那么这个点一定不是最优的。
这个还能再优化。
因为我考虑的是每一个点。如果我能保证我枚举的点的深度单调不减。就可以更高效率的筛掉不可能的答案。
所以我们可以在dfs里写一个队列来进一步优化。
然后还有一些小细节。比如预处理把每个节点的边从小到大排个序等
Code
#include <cstdio>
#include <iostream>
#include <algorithm>
#define R register
#define I inline
using namespace std;
const int inf = 9999999;
int n, m, u, v, w, scnt = 1, ans = inf, d[100][100], e[100][100], cnt[100], tot, vis[100], dep[100], pmin, f;
I int read(){
R int res = 0;
R char ch;
for(;ch < '0' || ch > '9';) ch = getchar();
for(;ch >= '0' && ch <= '9';) {res = res * 10 + ch - '0', ch = getchar();}
return res;
}
I bool cmp(int x, int y) {
return d[f][x] < d[f][y];
}
I void dfs(int s, int node) {
if(tot + pmin * dep[vis[s]] >= ans) return ;
if(scnt == n) {
ans = tot;
return ;
}
for(R int i = s; i <= scnt; ++i) {
R int cur = vis[i];
for(R int j = node; j <= cnt[cur]; ++j) {
R int t = e[cur][j];
if(dep[t]) continue;
dep[t] = dep[cur] + 1;
vis[++scnt] = t;
tot += dep[cur] * d[cur][t];
pmin -= d[t][e[t][1]];
dfs(i, j + 1);
pmin += d[t][e[t][1]];
tot -= dep[cur] * d[cur][t];
--scnt;
dep[t] = 0;
}
node = 1;
}
}
int main() {
freopen("treasure.in","r",stdin);
freopen("treasure.out","w",stdout);
n = read(), m = read();
for(R int i = 1; i <= n; ++i)
for(R int j = 1; j <= n; ++j)
d[i][j] = inf;
for(R int i = 1; i <= m; ++i) {
u = read(), v = read(), w = read();
if(d[u][v] == inf) {
d[u][v] = d[v][u] = w,
e[u][++cnt[u]] = v,
e[v][++cnt[v]] = u;
}
else if(w < d[u][v])
d[u][v] = d[v][u] = w;
}
for(R int i = 1; i <= n; ++i) {
f = i;
sort(e[i] + 1, e[i] + cnt[i] + 1, cmp);
pmin += d[i][e[i][1]];
}
for(R int i = 1; i <= n; ++i) {
dep[i] = 1;
vis[1] = i;
pmin -= d[i][e[i][1]];
dfs(1, 1);
pmin += d[i][e[i][1]];
dep[i] = 0;
}
printf("%d\n",ans);
return 0;
}