2018年第九届蓝桥杯 第七题：螺旋折线（满分19分）

标题：螺旋折线

如图p1.png所示的螺旋折线经过平面上所有整点恰好一次。 

对于整点(X, Y)，我们定义它到原点的距离dis(X, Y)是从原点到(X, Y)的螺旋折线段的长度。 

例如dis(0, 1)=3, dis(-2, -1)=9 

给出整点坐标(X, Y)，你能计算出dis(X, Y)吗？

【输入格式】
X和Y 

对于40%的数据，-1000 <= X, Y <= 1000 
对于70%的数据，-100000 <= X， Y <= 100000 
对于100%的数据, -1000000000 <= X, Y <= 1000000000 

【输出格式】
输出dis(X, Y) 

【样例输入】
0 1

【样例输出】
3

资源约定：
峰值内存消耗（含虚拟机） < 256M
CPU消耗  < 1000ms

 

#include <iostream>
#include <cmath>
using namespace std;
// x,y在对角线上 sat. -x = y 
long long  dialog_step(int x,int y)
{
    long long  n = 0;
    if (y>0)
        n = 2*y - 1;
    else
        n = 2*x;
    return n*(n+1);    
}
// step of any point
long long  xy_step(int x, int y)
{
    // split the space to 4 subspace
    if (y>0 && (-y<=x && x<=y)) 
    {
        // (-y,y)
        long long  step = dialog_step(-y,y)+(x+y);
        return step;
    }
    if (x>0 && (-x<y && y<x))
    {
        // (x,-x)
        long long  step = dialog_step(x,-x) - abs(-x-y);
        return step;
    }
    if (y<0 && (-abs(y)<=x && x<= abs(y)))
    {
        // (-y,y)
        long long  step = dialog_step(-y,y) + abs(x-(-y));
        return step;
    }
    if (x<0 && (-abs(x)<y && y<abs(x)))
    {
        //(x,-x)
        long long  step = dialog_step(x,-x) - abs(-x-y);
        return step;
    }
}
int main()
{
//    cout<<dialog_step(-1,1)<<endl; // 2
//    cout<<dialog_step(1,-1)<<endl; // 6    
//    cout<<dialog_step(-2,2)<<endl; // 12
//    cout<<dialog_step(2,-2)<<endl; // 20
    cout<<xy_step(-1,0)<<endl; // 1
    cout<<xy_step(-1,1)<<endl; // 2
    cout<<xy_step(0,1)<<endl; // 3    
    cout<<xy_step(1,1)<<endl; // 4    
    cout<<xy_step(1,0)<<endl;  // 5 
    cout<<xy_step(1,-1)<<endl;  // 6 
    cout<<xy_step(0,-1)<<endl; // 7
    cout<<xy_step(-1,-1)<<endl; // 8
    cout<<xy_step(-2,-1)<<endl; // 9
    cout<<xy_step(-2,0)<<endl; // 10
    cout<<xy_step(1,-2)<<endl; // 21
    cout<<xy_step(20,2)<<endl; // 21
    return 0;        
}