HDOJ1030 明明测试数据都对了的，为什么老是WA啊 …… 大牛请进 …

http://acm.hdu.edu.cn/showproblem.php?pid=1030
#include <iostream>
#include <math.h>using namespace std;
int fun(double d)//求M、N所在的行
{
if(sqrt(d)>(int)sqrt(d)) return (int)sqrt(d)+1;
else return (int)sqrt(d);
}
void flag(int M,int m,int & flag1,int & flag1_) //flag1和flag1_分别为数字左右两边空的行数flag1是右边，flag1_是左边
{
if(M%2==m%2) { flag1=(m*m-M)/2; flag1_=(M-(m-1)*(m-1)-1)/2; }
else   { flag1=(m*m-M-1)/2; flag1_=(M-1-(m-1)*(m-1)-1)/2; }
}
int Rought(int M,int N,int m,int n)//用于求路径
{ int sum=0;
if (m==n) { return (N-M); }
int flag1,flag1_,flag2,flag2_;
flag(M,m,flag1,flag1_);
flag(N,n,flag2,flag2_);
if (flag2>=flag1&&flag2_>=flag1_)
{
sum+=(n-m-1)*2;
if (N%2==M%2) { if ((n-m)%2)sum++; else sum+=2; }
else if (N%2!=M%2) { if ((n-m)%2)sum+=2; else sum++; } }
else if ((flag1-flag2)>0)
{
sum+=(n-m)*2;
int M_=n*n-flag1*2;
sum+=(N-M_); }
else
{ sum+=(n-m)*2;
   int M_=(n-1)*(n-1)+flag1_*2+1;
   sum+=(M_-N);
}
return sum;
}
int main()
{
int M,N;
int m,n;
cin>>M>>N;
if (M>N){int tmp =N;N=M;M=tmp;}//就算时保证M值小于N值
m=fun((double)M);
n=fun((double)N);
if (m%2==M%2) cout<<Rought(M,N,m,n)<<endl;//如果是正的三角形直接求解如果是倒置的三角形就分别求M加一和减一的路径，然后加一就是路程
else { int a=Rought(M+1,N,m,n),b=Rought(M-1,N,m,n); cout<<(a<b?a:b)+1<<endl; } return 0;}