提交时间:2024-11-28 13:31:50
运行 ID: 35147
#include<bits/stdc++.h> #define int long long using namespace std; int n,k; int cnt[8000005]; signed main(){ cin>>n>>k; for(int i=1;i<=n;i++){ for(int j=1;j<=n;j++){ cnt[i*i+j*j]++; } } // for(int i=1;i<=n*n+n*n;i++)cout<<cnt[i]<<" "; int ans=0; for(int i=1;i<=n;i++){ for(int j=1;j<=n;j++){ if(k+i*i-j*j>n*n+n*n)continue; if(k+i*i-j*j<0)continue; ans+=cnt[k+i*i-j*j]; // cout<<k+i*i-j*j<<" "<<ans<<endl; } } cout<<ans<<endl; return 0; }