結果

問題 No.800 四平方定理
コンテスト
ユーザー outline
提出日時 2020-03-28 18:05:27
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
(最新)
AC  
(最初)
実行時間 -
コード長 1,517 bytes
コンパイル時間 1,235 ms
コンパイル使用メモリ 110,092 KB
実行使用メモリ 440,828 KB
最終ジャッジ日時 2025-01-02 11:58:11
合計ジャッジ時間 37,354 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 22 TLE * 8
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <vector>
#include <algorithm>
#include <cmath>
#include <queue>
#include <string>
#include <map>
#include <set>
#include <stack>
#include <tuple>
#include <deque>
#include <numeric>
#include <bitset>
#include <iomanip>
#include <cassert>
#include <chrono>
#include <random>
#include <limits>
#include <iterator>
#include <functional>
#include <sstream>
#include <complex>
using namespace std;

typedef long long ll;
typedef uint64_t ull;
typedef pair<int, int> P;
typedef pair<int, double> Pid;
typedef pair<double, int> Pdi;
typedef pair<ll, int> Pl;
typedef pair<ll, ll> Pll;
typedef pair<int, pair<int, int>> PP;
typedef pair<P, int> PPi;
constexpr double PI = 3.1415926535897932;   // acos(-1)
constexpr double EPS = 1e-9;
constexpr int INF = 1001001001;
constexpr int mod = 1000000007;
// constexpr int mod = 998244353;

#define chmax(x, y) x = max(x, y)
#define chmin(x, y) x = min(x, y)
#define chadd(x, y) x = (x + y) % mod

int main(){
    ios::sync_with_stdio(false);
    cin.tie(nullptr);

    int n, d;
    cin >> n >> d;
    int ub = n * n * 2;
    vector<vector<ll>> cnt(ub + 1, vector<ll>(2, 0));
    for(int x = 1; x <= n; ++x){
        for(int y = 1; y <= n; ++y){
            int foo = x * x + y * y;
            ++cnt[foo][0];
            int bar = x * x - y * y + d;
            if(bar <= 0 || bar > ub)    continue;
            ++cnt[bar][1];
        }
    }
    ll ans = 0;
    for(int i = 1; i <= ub; ++i)    ans += cnt[i][0] * cnt[i][1];
    cout << ans << endl;
}
0