結果

問題 No.800 四平方定理
ユーザー noshi91noshi91
提出日時 2019-03-17 21:30:20
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 106 ms / 2,000 ms
コード長 6,064 bytes
コンパイル時間 587 ms
コンパイル使用メモリ 70,204 KB
実行使用メモリ 66,108 KB
最終ジャッジ日時 2023-09-22 03:53:17
合計ジャッジ時間 3,885 ms
ジャッジサーバーID
(参考情報)
judge13 / judge14
このコードへのチャレンジ(β)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 20 ms
65,884 KB
testcase_01 AC 19 ms
65,840 KB
testcase_02 AC 19 ms
65,884 KB
testcase_03 AC 20 ms
65,800 KB
testcase_04 AC 20 ms
65,832 KB
testcase_05 AC 19 ms
65,948 KB
testcase_06 AC 20 ms
65,828 KB
testcase_07 AC 19 ms
65,884 KB
testcase_08 AC 19 ms
65,900 KB
testcase_09 AC 19 ms
65,968 KB
testcase_10 AC 53 ms
65,908 KB
testcase_11 AC 55 ms
65,888 KB
testcase_12 AC 60 ms
65,896 KB
testcase_13 AC 52 ms
65,824 KB
testcase_14 AC 58 ms
65,836 KB
testcase_15 AC 56 ms
65,832 KB
testcase_16 AC 57 ms
65,836 KB
testcase_17 AC 55 ms
65,796 KB
testcase_18 AC 61 ms
65,896 KB
testcase_19 AC 59 ms
65,900 KB
testcase_20 AC 19 ms
65,832 KB
testcase_21 AC 18 ms
65,824 KB
testcase_22 AC 60 ms
65,928 KB
testcase_23 AC 104 ms
65,912 KB
testcase_24 AC 96 ms
65,968 KB
testcase_25 AC 106 ms
65,856 KB
testcase_26 AC 18 ms
65,944 KB
testcase_27 AC 20 ms
65,800 KB
testcase_28 AC 99 ms
65,900 KB
testcase_29 AC 105 ms
65,960 KB
testcase_30 AC 97 ms
65,908 KB
testcase_31 AC 90 ms
65,996 KB
testcase_32 AC 101 ms
66,108 KB
権限があれば一括ダウンロードができます

ソースコード

diff #

//#define NDEBUG
#include <cstddef>
#include <cstdint>
#include <vector>

namespace n91 {

	using i8 = std::int_least8_t;
	using i32 = std::int_least32_t;
	using i64 = std::int_least64_t;
	using u8 = std::uint_least8_t;
	using u32 = std::uint_least32_t;
	using u64 = std::uint_least64_t;
	using isize = std::ptrdiff_t;
	using usize = std::size_t;

	class rep {
		const usize f, l;

	public:
		class itr {
			friend rep;
			usize i;
			constexpr itr(const usize x) noexcept : i(x) {}

		public:
			void operator++() noexcept { ++i; }
			constexpr usize operator*() const noexcept { return i; }
			constexpr bool operator!=(const itr x) const noexcept { return i != x.i; }
		};
		constexpr rep(const usize first, const usize last) noexcept
			: f(first), l(last) {}
		constexpr itr begin() const noexcept { return itr(f); }
		constexpr itr end() const noexcept { return itr(l); }
	};
	class revrep {
		const usize f, l;

	public:
		class itr {
			friend revrep;
			usize i;
			constexpr itr(usize x) noexcept : i(x) {}

		public:
			void operator++() noexcept { --i; }
			constexpr usize operator*() const noexcept { return i; }
			constexpr bool operator!=(const itr x) const noexcept { return i != x.i; }
		};
		constexpr revrep(usize first, usize last) noexcept : f(--first), l(--last) {}
		constexpr itr begin() const noexcept { return itr(l); }
		constexpr itr end() const noexcept { return itr(f); }
	};
	template <class T> using vec_alias = std::vector<T>;
	template <class T> auto md_vec(const usize n, const T &value) {
		return std::vector<T>(n, value);
	}
	template <class... Args> auto md_vec(const usize n, Args... args) {
		return std::vector<decltype(md_vec(args...))>(n, md_vec(args...));
	}
	template <class T> constexpr T difference(const T &a, const T &b) {
		return a < b ? b - a : a - b;
	}

} // namespace n91

#include <cassert>
#include <cstdint>

namespace n91 {

	template <std::uint_least32_t Modulus> class modint {
		using u32 = std::uint_least32_t;
		using u64 = std::uint_least64_t;
		using i64 = std::int_fast64_t;
		class optimize_tag_t {};
		static constexpr optimize_tag_t optimize_tag{};

	public:
		using value_type = u32;
		value_type a;

	private:
		static constexpr value_type cst(i64 x) noexcept {
			x %= static_cast<i64>(Modulus);
			if (x < static_cast<i64>(0)) {
				x += static_cast<i64>(Modulus);
			}
			return static_cast<value_type>(x);
		}

		constexpr modint(optimize_tag_t, const value_type &x) noexcept : a(x) {}

	public:
		constexpr modint() noexcept : a(static_cast<value_type>(0)) {}
		constexpr modint(const i64 &x) noexcept : a(cst(x)) {}
		constexpr value_type value() const noexcept { return a; }
		constexpr modint operator+(const modint &o) const noexcept {
			return modint(optimize_tag,
				a + o.a < Modulus ? a + o.a : a + o.a - Modulus);
		}
		constexpr modint operator-(const modint &o) const noexcept {
			return modint(optimize_tag, a < o.a ? a + Modulus - o.a : a - o.a);
		}
		constexpr modint operator*(const modint &o) const noexcept {
			return modint(
				optimize_tag,
				static_cast<value_type>(static_cast<u64>(a) * static_cast<u64>(o.a) %
					static_cast<u64>(Modulus)));
		}
		constexpr modint operator/(const modint &o) const {
			return modint(
				optimize_tag,
				static_cast<value_type>(static_cast<u64>(a) * static_cast<u64>((~o).a) %
					static_cast<u64>(Modulus)));
		}
		modint &operator+=(const modint &o) noexcept {
			if ((a += o.a) >= Modulus)
				a -= Modulus;
			return *this;
		}
		modint &operator-=(const modint &o) noexcept {
			if (a < o.a)
				a += Modulus;
			a -= o.a;
			return *this;
		}
		modint &operator*=(const modint &o) noexcept {
			a = static_cast<value_type>(static_cast<u64>(a) * static_cast<u64>(o.a) %
				static_cast<u64>(Modulus));
			return *this;
		}
		modint &operator/=(const modint &o) {
			a = static_cast<u64>(a) * (~o).a % Modulus;
			return *this;
		}
		constexpr modint inverse() const noexcept {
			assert(a != static_cast<value_type>(0) && "0 does not have inverse");
			return pow(static_cast<u64>(Modulus - static_cast<value_type>(2)));
		}
		constexpr modint operator~() const noexcept { return inverse(); }
		constexpr modint operator-() const noexcept {
			if (a == static_cast<value_type>(0)) {
				return modint(optimize_tag, static_cast<value_type>(0));
			}
			else {
				return modint(optimize_tag, Modulus - a);
			}
		}
		modint &operator++() noexcept {
			if (++a == Modulus) {
				a = static_cast<value_type>(0);
			}
			return *this;
		}
		modint &operator--() noexcept {
			if (a == static_cast<value_type>(0)) {
				a = Modulus;
			}
			--a;
			return *this;
		}
		constexpr bool operator==(const modint &o) const noexcept { return a == o.a; }
		constexpr bool operator!=(const modint &o) const noexcept { return a != o.a; }
		constexpr bool operator<(const modint &o) const noexcept { return a < o.a; }
		constexpr bool operator<=(const modint &o) const noexcept { return a <= o.a; }
		constexpr bool operator>(const modint &o) const noexcept { return a > o.a; }
		constexpr bool operator>=(const modint &o) const noexcept { return a >= o.a; }
		constexpr explicit operator bool() const noexcept { return a; }
		constexpr explicit operator value_type() const noexcept { return a; }
		modint pow(u64 x) const noexcept {
			u64 t = a, u = 1;
			while (x) {
				if (x & 1)
					u = u * t % Modulus;
				t = (t * t) % Modulus;
				x >>= 1;
			}
			return modint(optimize_tag, static_cast<value_type>(u));
		}
	};

} // namespace n91


#include <algorithm>
#include <iostream>
#include <utility>
#include <array>

namespace n91 {

	void main_() {
		static constexpr usize MaxN = 2000;
		u32 n, d;
		std::cin >> n >> d;
		std::array<u64, MaxN*MaxN * 2 + 1> count = {};
		u64 ans = 0;
		for (u32 x = 1;x <= n;++x) {
			for (u32 y = 1;y <= n;++y) {
				++count[x*x + y*y];
			}
		}
		for (u32 z = 1;z <= n;++z) {
			for (u32 w = 1;w <= n;++w) {
				if (z*z <= w*w + d) {
					ans += count[w*w + d - z*z];
				}
			}
		}
		std::cout << ans << std::endl;
	}

} // namespace n91

int main() {
	n91::main_();
	return 0;
}
0