关于qsort的比较函数的不同写法对于大数据造成的错误
先看一下这两份代码:
代码1:AC
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cmath>
using namespace std;
#define N 5050
int p[N];
double point[2][110];
struct road
{
int a, b;
double dis;
}node[N];
int cmp(const void * a, const void * b)
{
if ( ((road *)a)->dis - ((road *)b)->dis > 0 ) return 1;
else return -1;
}
int root(int q)
{
if ( p[q] == -1 ) return q;
else return p[q] = root(p[q]);
}
void merge(int a,int b)
{
a = root(a);
b = root(b);
p[a] = b;
}
int main()
{
freopen("in.txt","r",stdin);
int i, j, k, n;
while ( scanf("%d",&n) != -1 )
{
k = 0;
double ans = 0;
int edge = 0;
memset(p, -1, sizeof(p) );
memset(point, 0, sizeof(point) );
memset(node, 0, sizeof(node) );
for ( i = 1; i <= n; i ++ )
{
scanf("%lf%lf",&point[0][i],&point[1][i]);
}
for ( i = 1; i < n; i ++ )
{
for ( j = i+1; j <= n; j ++ )
{
node[k].a = i;
node[k].b = j;
node[k].dis = sqrt( ((point[0][i]-point[0][j]) * (point[0][i]-point[0][j])) + ((point[1][i]-point[1][j]) * (point[1][i]-point[1][j])) );
k ++;
}
}
qsort(node,k,sizeof(road),cmp);
for ( i = 0; i < k; i ++ )
{
if ( root(node[i].a) != root(node[i].b) )
{
merge(node[i].a, node[i].b);
ans += node[i].dis;
edge ++;
}
if ( edge == n - 1 ) break;
}
printf("%.2lf\n",ans);
}
return 0;
}
代码2:WA
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cmath>
using namespace std;
#define N 5050
int p[N];
double point[2][110];
struct road
{
int a, b;
double dis;
}node[N];
int cmp(const void * a, const void * b)
{
return (int)( ((road *)a)->dis - ((road *)b)->dis );
}
int root(int q)
{
if ( p[q] == -1 ) return q;
else return p[q] = root(p[q]);
}
void merge(int a,int b)
{
a = root(a);
b = root(b);
p[a] = b;
}
int main()
{
freopen("in.txt","r",stdin);
int i, j, k, n;
while ( scanf("%d",&n) != -1 )
{
k = 0;
double ans = 0;
int edge = 0;
memset(p, -1, sizeof(p) );
memset(point, 0, sizeof(point) );
memset(node, 0, sizeof(node) );
for ( i = 1; i <= n; i ++ )
{
scanf("%lf%lf",&point[0][i],&point[1][i]);
}
for ( i = 1; i < n; i ++ )
{
for ( j = i+1; j <= n; j ++ )
{
node[k].a = i;
node[k].b = j;
node[k].dis = sqrt( ((point[0][i]-point[0][j]) * (point[0][i]-point[0][j])) + ((point[1][i]-point[1][j]) * (point[1][i]-point[1][j])) );
k ++;
}
}
qsort(node,k,sizeof(road),cmp);
for ( i = 0; i < k; i ++ )
{
if ( root(node[i].a) != root(node[i].b) )
{
merge(node[i].a, node[i].b);
ans += node[i].dis;
edge ++;
}
if ( edge == n - 1 ) break;
}
printf("%.2lf\n",ans);
}
return 0;
}
这两份代码的唯一不同就是qsort的cmp函数的写法不同,第一份是利用比较大小然后返回相应的表示正负的1和-1,第二份是利用减法直接得出正负。但结果却完全不同,代码1 AC了,但代码2 却WA了。看了一下oj的运行时间,代码1是234ms,代码2也是234ms,因此我断定出错的是最后一个数据,由于这道题很明显不存在边界,因此我判断这个出错的数据是大数据。当负的大数减正的大数时,必然会导致越界而使应该得到的负数变成正数,从而使排序错误造成WA。
因此,在写qsort的比较函数时,最好不要使用减法来得到返回值,用比较法得出正负后直接返回正负标志会更好的。