提交时间:2024-11-28 13:40:39

运行 ID: 35151

#include <bits/stdc++.h> #define int long long using namespace std; int n, k; unordered_map <int, int> mp; signed main() { scanf("%lld %lld", &n, &k); for (int i = 1; i <= n; i++) for (int j = 1; j <= n; j++) mp[i * i + j * j]++; int ans = 0; for (int a = 1; a <= n; a++) { for (int d = 1; d <= n; d++) { ans += mp[d * d - a * a + k]; } } printf("%lld", ans); return 0; }