原题:qwb与整数对
原题大意
给出两个整数n和m,请统计满足0<a<b<n并且使得 (a2+b2+m)/(ab) 的结果是整数的整数对(a,b)的个数。
本题包含多组测试例 。当测试例数据是n=m=0时,表示输入结束。(测试例数量<6000)
每个测试例一行,是两个整数n和m。输入保证0≤n≤1000,-20000 < m < 20000。
算法分析
case=1的时候很容易可以想到一个n^2的暴力
那么在case比较大的时候,把所有case根据n的数值进行排序。再离线处理,枚举每一组(a,b),穷举整数t∈[-20000,20000],当符合条件(a^2+b^2+t)%ab==0时,ans[t]++,最后根据case的顺序从离线处理中获取答案,复杂度O(∑(i=1..n,j=2..n) (40000/ab)) + O(n^1.5)
程序代码
#include <bits/stdc++.h>
using namespace std;
int n[10100], m[10100];
int ret[10100];
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);//M的功能类似矫正参数
for(int t = 0; 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 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]);
return 0;
}