2D N体问题C代码出现NaN值求助：引力计算异常排查

我一眼就看到几个会触发NaN的关键问题，咱们一个个拆解：

1. 最致命的复制粘贴错误：dy计算写错了 #

在updateBodyForces函数里，你错误地把y方向的差值写成了x方向的差值：

dy = bodyB->rx - bodyA->rx; // 这里错了！

正确的写法应该是：

dy = bodyB->ry - bodyA->ry;

这个错误会导致dx和dy完全相同，力的方向彻底混乱，还会间接导致距离计算异常，是引发NaN的潜在诱因之一。

2. 浮点数相等判断不可靠，直接导致除以0 #

你用==判断两个物体是否位置重合：

if (bodyB->rx == bodyA->rx && bodyB->ry == bodyA->ry) {

但浮点数存在精度限制，经过几次迭代计算后，两个物体的位置可能非常接近但不完全相等，此时==判断会失效。如果这时候dx和dy的实际值为0（或因为精度误差导致距离dist为0），后续的dx/dist和dy/dist就会变成0/0，直接产生NaN。

修复方案：用epsilon阈值判断近似重合 #

把相等判断改成基于极小阈值的近似判断，比如：

const double EPS = 1e-9; // 可根据你的精度需求调整
if (fabs(bodyB->rx - bodyA->rx) < EPS && fabs(bodyB->ry - bodyA->ry) < EPS) {

3. 优化距离计算，避免冗余操作（顺带提升性能） #

你当前计算距离的方式是sqrt(pow(dx,2) + pow(dy,2))，可以先计算距离的平方dist_sq = dx*dx + dy*dy，这样既可以用来判断近似重合（dist_sq < EPS*EPS），又能避免重复计算平方，提升代码效率：

void updateBodyForces (Body* bodyA, Body* bodyB) {
 double dx, dy, dist_sq, dist, force;
 dx = bodyB->rx - bodyA->rx;
 dy = bodyB->ry - bodyA->ry; // 已修正dy的计算
 dist_sq = dx*dx + dy*dy;
 const double EPS = 1e-9;
 
 if (dist_sq < EPS*EPS) {
 dist = 1.0;
 } else {
 dist = sqrt(dist_sq);
 }
 // 用dist_sq代替pow(dist,2)，更高效
 force = (G * bodyA->mass * bodyB->mass) / (dist_sq + 100);
 bodyA->fx += force * dx / dist;
 bodyA->fy += force * dy / dist;
}

4. 额外防护：避免质量为0的情况 #

虽然你当前的getRandomBody把质量固定为20，但如果后续修改了质量赋值逻辑，最好在updateBody里加个防护，避免除以0：

void updateBody (Body* bodyInstance, int timestamp) {
 const double EPS = 1e-9;
 if (fabs(bodyInstance->mass) < EPS) { // 质量接近0时跳过计算
 return;
 }
 bodyInstance->vx += timestamp * bodyInstance->fx / bodyInstance->mass;
 bodyInstance->vy += timestamp * bodyInstance->fy / bodyInstance->mass;
 bodyInstance->rx += timestamp * bodyInstance->vx;
 bodyInstance->ry += timestamp * bodyInstance->vy;
};

总结一下：你遇到的NaN问题主要由dy的计算错误和浮点数相等判断失效导致的除以0引起，修复这两个核心问题后，NaN应该就不会再出现了。

内容的提问来源于stack exchange，提问作者Moonjsit