这道题目几天前弄的,测试几个数据也没错,提交总是TLE 吸取以前的经验 把cin改成scanf试了下,还是没发现错误 就此搁置了好几天,在网上找了个kruskal算法的版本提交了下300ms 觉得自己有些地方处理的不好也不至于超时呀(刚开始一直考虑降低并查集树的高度降低时间开销,真实舍本逐末,这些影响不会太大)
于是认真检查,终于发现了玄机,原来以前一直习惯用 queue,它的操作是在探查结束后就会一个个清空整个元素,一直没怎么注意这个问题,而这次用了vector,灾难来临,使用的是逐个遍历 操作结束之后并没有清空元素,看来还是养成好的编程习惯在使用容器前先给容器清零;
其实还有一个比较恶心的地方 是我一直习惯用 同一个测试用例连续输入来检测现场恢复是不是正确的,但这道题目该方法没用,因为一个个边输入多次在vector中有多个副本也不会影响算法得到正确结果,除非输出一下容器当前的大小 汗~~
另外发现了一个问题,oj的测试用例是整个一齐输入 然后比较整个输出的,不然这个程序会在前几个用例时就会报错 而不是TLE 其实也很合理,逐个测试用例交互式对oj开销太大,这样上百个测试用例的数据一齐叠加在vector里 自然会超时
ps:这道题目发现prim算法还是比Kruskal算法高效很多,时间效率高好几倍 46ms VS 300ms
#include <iostream>
#include <iomanip>
#include <math.h>
#define MAX 100000000
using namespace std;
double map[105][105];
double dist[105];
int close[105];
struct node
{
int v1,v2;
};
node point[100];
void getdis(int n)
{
double cost;
for(int i=0;i<n-1;i++)
for(int j=i+1;j<n;j++)
{
cost=sqrt(pow(point[i].v1-point[j].v1,2)+pow(point[i].v2-point[j].v2,2));
//cost=sqrt((point[i].v1-point[j].v1)*(point[i].v1-point[j].v1)+(point[i].v2-point[j].v2)*(point[i].v2-point[j].v2));
if(cost<10||cost>1000)cost=MAX;
map[i][j]=map[j][i]=cost;
}
for(i=0;i<n;i++)map[i][i]=0;
}
void Prim(int n)
{
memset(close,0,sizeof(close));
dist[0]=0;
close[0]=1;
for(int i=0;i<n;i++)
dist[i]=map[0][i];
int minpos;
double minvalue;
double sum=0;
int tag=1;
for(i=1;i<n;i++)
{
minvalue=MAX+1;
for(int j=1;j<n;j++)
if(dist[j]<minvalue&&!close[j])minpos=j,minvalue=dist[j];
close[minpos]=1;
if(dist[minpos]==MAX){tag=0;break;}
else sum+=dist[minpos];
for(j=1;j<n;j++)
if(dist[j]>map[minpos][j]&&!close[j])dist[j]=map[minpos][j];
}
if(tag)cout<<setiosflags(ios::fixed)<<setprecision(1)<<sum*100<<endl;
else cout<<"oh!"<<endl;
}
int main()
{
int t,n;
cin>>t;
while(t--)
{
cin>>n;
for(int i=0;i<n;i++)
cin>>point[i].v1>>point[i].v2;
getdis(n);
Prim(n);
}
return 0;
}