Teen Girl Squad - UVA 11183 - 最小树形图

之前的两题次小生成树的题解一直没补，容我再拖一拖。（手动/doge）

题目

  PDF题目直接附上链接吧：传送门

题意

  很裸的最小树形图题目，直接套模板就可以。

实现

  这道题WA了很多遍因为没有注意到数据范围的问题，数组刚开始给小了。而且OJ一直给我返WA而不是RE之类的，让我很费解。

  值得注意的一点是，在找环的过程中，我原本的代码是：

for(int i=0 ; i<n ; i++) {//标记每个环
ans += in[i];
int v = i;
while(vis[v] != i && id[v] == -1 /*tag*/) {
vis[v] = i;
v = pre[v];
}

  这样的代码交上去之后发现跑了300ms，本来觉得挺短的，看了一下别人跑的时间是30ms，整整十倍啊！令我很不能费解。

  然后就去网上拖代码来看，看来看去没看出来太大的差别。通过代码对比器发现有一个地方不一样。就是在第一段代码标注了

/*tag*/

的地方，别人的代码是这样的：

for(int i=0 ; i<n ; i++) {//标记每个环
ans += in[i];
int v = i;
while(vis[v] != i && id[v] == -1 && v != root) {
vis[v] = i;
v = pre[v];
}

  相比我的代码多了一个

v != root

的判断，感觉加这么一个限制条件并不能快多少，但我加上之后交上去果然也把时间缩短到40ms。有时间去问问学长这个问题吧。

代码

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>

using namespace std;

#define maxn 1007
#define memset(a,b) memset(a,b,sizeof(a))
const int INF = 0x3f3f3f3f;

int pre[maxn],id[maxn],vis[maxn],in[maxn];

struct {
int u,v,w;
}edge[maxn*40];

int directed_MST(int root, int n, int m) {
int ans = 0;
while(true) {
//1.找最小入边
for(int i=0 ; i<n ; i++) in[i] = INF;
for(int i=0 ; i<m ; i++) {
int u = edge[i].u, v = edge[i].v;
if(edge[i].w < in[v] && u!=v) {
pre[v] = u;
in[v] = edge[i].w;
}
}
for(int i=0 ; i<n ; i++) {
if(i == root) continue;
if(in[i] == INF) return -1;
}
//2.找环
int cnt = 0;
memset(id,-1);
memset(vis,-1);
in[root] = 0;
for(int i=0 ; i<n ; i++) {//标记每个环
ans += in[i];
int v = i;
while(vis[v] != i && id[v] == -1 && v != root) {
vis[v] = i;
v = pre[v];
}
if(v != root && id[v] == -1) {
for(int u = pre[v] ; u!=v ; u = pre[u])
id[u] = cnt;
id[v] = cnt++;
}
}
if(cnt == 0) break;
for(int i=0 ; i<n ; i++) if(id[i] == -1) id[i] = cnt++;
//3.缩点，重新标记
for(int i=0 ; i<m ; i++) {
int v = edge[i].v;
edge[i].u = id[edge[i].u];
edge[i].v = id[edge[i].v];
if(edge[i].u != edge[i].v) {
edge[i].w -= in[v];
}
}
n = cnt;
root = id[root];
}
return ans;
}

int main() {
#ifndef ONLINE_JUDGE
freopen("/home/lucienshui/Desktop/in","r",stdin);
#endif
int t,cnt=0,n,m;
scanf("%d",&t);
while(t--) {
printf("Case #%d: ",++cnt);
scanf("%d%d",&n,&m);
for(int i=0 ; i<m ; i++) {
scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w);
}
int ans = directed_MST(0,n,m);
if(ans == -1) printf("Possums!\n");
else printf("%d\n",ans);
}
return 0;
}