結果

問題 No.800 四平方定理
ユーザー ミドリムシミドリムシ
提出日時 2019-03-17 21:25:23
言語 C++11
(gcc 11.4.0)
結果
AC  
実行時間 1,362 ms / 2,000 ms
コード長 3,134 bytes
コンパイル時間 872 ms
コンパイル使用メモリ 89,304 KB
実行使用メモリ 34,688 KB
最終ジャッジ日時 2024-07-07 20:15:20
合計ジャッジ時間 18,730 ms
ジャッジサーバーID
（参考情報） 		judge4 / judge2
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 9 ms
5,248 KB
testcase_01 AC 16 ms
5,376 KB
testcase_02 AC 12 ms
5,376 KB
testcase_03 AC 14 ms
5,376 KB
testcase_04 AC 12 ms
5,376 KB
testcase_05 AC 13 ms
5,376 KB
testcase_06 AC 18 ms
5,376 KB
testcase_07 AC 16 ms
5,376 KB
testcase_08 AC 10 ms
5,376 KB
testcase_09 AC 16 ms
5,376 KB
testcase_10 AC 590 ms
18,816 KB
testcase_11 AC 644 ms
20,736 KB
testcase_12 AC 686 ms
20,480 KB
testcase_13 AC 523 ms
19,584 KB
testcase_14 AC 566 ms
20,736 KB
testcase_15 AC 680 ms
20,736 KB
testcase_16 AC 570 ms
20,864 KB
testcase_17 AC 570 ms
20,864 KB
testcase_18 AC 761 ms
20,992 KB
testcase_19 AC 751 ms
20,992 KB
testcase_20 AC 1 ms
5,376 KB
testcase_21 AC 1 ms
5,376 KB
testcase_22 AC 759 ms
20,864 KB
testcase_23 AC 1,362 ms
34,560 KB
testcase_24 AC 1,077 ms
34,560 KB
testcase_25 AC 1,318 ms
34,432 KB
testcase_26 AC 1 ms
5,376 KB
testcase_27 AC 2 ms
5,376 KB
testcase_28 AC 1,096 ms
34,560 KB
testcase_29 AC 1,335 ms
34,688 KB
testcase_30 AC 1,161 ms
34,432 KB
testcase_31 AC 993 ms
32,384 KB
testcase_32 AC 1,127 ms
34,560 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <iostream>
#include <algorithm>
#include <vector>
#include <string.h>
#include <random>
#include <iomanip>
using namespace std;
const long mod = 1e9 + 7;
#define all(x) (x).begin(), (x).end()
#define bitcount(n) __builtin_popcountl(long(n))
#define fcout cout << fixed << setprecision(15)
#define highest(x) (63 - __builtin_clzl(x))
template<class T> inline void YES(T condition){ if(condition) cout << "YES" << endl; else cout << "NO" << endl; }
template<class T> inline void Yes(T condition){ if(condition) cout << "Yes" << endl; else cout << "No" << endl; }
template<class T> inline void POSS(T condition){ if(condition) cout << "POSSIBLE" << endl; else cout << "IMPOSSIBLE" << endl; }
template<class T> inline void Poss(T condition){ if(condition) cout << "Possible" << endl; else cout << "Impossible" << endl; }
template<class T> inline void First(T condition){ if(condition) cout << "First" << endl; else cout << "Second" << endl; }
template<class T = string, class U = char>int character_count(T text, U character){ int ans = 0; for(U i: text){ ans += (i == character); } return ans; }
long power(long base, long exponent, long module){ if(exponent % 2){ return power(base, exponent - 1, module) * base % module; }else if(exponent){ long root_ans = power(base, exponent / 2, module); return root_ans * root_ans % module; }else{ return 1; }}
struct position{ int y, x; }; position mv[4] = {{-1, 0}, {0, 1}, {1, 0}, {0, -1}}; // double euclidean(position first, position second){ return sqrt((second.x - first.x) * (second.x - first.x) + (second.y - first.y) * (second.y - first.y)); }
template<class T, class U> string to_string(pair<T, U> x){ return to_string(x.first) + "," + to_string(x.second); } string to_string(string x){ return x; }
template<class itr> void array_output(itr start, itr goal){ string ans; for(auto i = start; i != goal; i++) ans += to_string(*i) + " "; if(!ans.empty()) ans.pop_back(); cout << ans << endl; }
template<class itr> void cins(itr start, itr goal){ for(auto i = start; i != goal; i++){ cin >> (*i); } }
template<class T> T gcd(T a, T b){ if(a && b){ return gcd(min(a, b), max(a, b) % min(a, b)); }else{ return a; }} template<class T> T lcm(T a, T b){ return a / gcd(a, b) * b; }
struct combination{ vector<long> fact, inv; combination(int sz) : fact(sz + 1), inv(sz + 1){ fact[0] = 1; for(int i = 1; i <= sz; i++){ fact[i] = fact[i - 1] * i % mod; } inv[sz] = power(fact[sz], mod - 2, mod); for(int i = sz - 1; i >= 0; i--){ inv[i] = inv[i + 1] * (i + 1) % mod; } } long C(int p, int q) const{ if(q < 0 || p < q) return 0; return (fact[p] * inv[q] % mod * inv[p - q] % mod); } };

int main(){
    int N, D;
    cin >> N >> D;
    int xy[N * N], zw[N * N];
    for(int i = 0; i < N; i++){
        for(int j = 0; j < N; j++){
            xy[i * N + j] = (i + 1) * (i + 1) + (j + 1) * (j + 1);
            zw[i * N + j] = (i + 1) * (i + 1) - (j + 1) * (j + 1);
        }
    }
    sort(xy, xy + N * N);
    long ans = 0;
    for(int i = 0; i < N * N; i++){
        ans += upper_bound(xy, xy + N * N, D - zw[i]) - lower_bound(xy, xy + N * N, D - zw[i]);
    }
    cout << ans << endl;
}
0