qwb与整数对 （思维枚举难）

Problem L: qwb与整数对
Time Limit: 1 Sec Memory Limit: 128 MB
Submit: 214 Solved: 34
[Submit][Status][Web Board]
Description

qwb又遇到了一道数学难题，你能帮助他吗？

给出两个整数n和m，请统计满足0＜a＜b＜n并且使得 (a2+b2+m)/(ab) 的结果是整数的整数对(a,b)的个数。

Input

本题包含多组测试例 。当测试例数据是n=m=0时，表示输入结束。（测试例数量<6000）

每个测试例一行，是两个整数n和m。输入保证0≤n≤1000，-20000

#include <bits/stdc++.h>
using namespace std;

int n[10100],m[10100];
int ret[10101];
vector<int> id[40404];
int main()
{
    int i=0;
    while(scanf("%d%d",&n[i],&m[i])!=EOF&&(n[i]||m[i]))
    {
        id[m[i]+20000].push_back(i);
        i++;
    }
    for(int a=1;a<=1000;a++)
    {
        for(int b=a+1;b<=1000;b++){
            int M=a*b-(a*a+b*b)%(a*b);
            for(int t=0;M+t*(a*b)<=20000;t++)
            {
                for(int i=0;i<id[M+t*(a*b)+20000].size();i++)
                {//利用m[i] 找到n[i]。
                    if(b<n[id[M+t*(a*b)+20000][i]]) ret[id[M+t*(a*b)+20000][i]]++;
                }
                }
            }
            for(int t=1;M-t*(a*b)>=-20000;t++)
            {
                for(int i=0;i<id[M-t*(a*b)+20000].size();i++)
                {
                    if(b<n[id[M-t*(a*b)+20000][i]]) ret[id[M-t*(a*b)+20000][i]]++;
                }
            }
        }
    }
    for(int i=0;n[i]||m[i];i++)
    {
        printf("Case %d: %d\n",i+1,ret[i]);
    }
}