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]);
    }
}