結果

問題 No.648  お や す み 
ユーザー nobigomu
提出日時 2018-04-27 17:58:01
言語 Lua
(LuaJit 2.1.1734355927)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 1,037 bytes
コンパイル時間 328 ms
コンパイル使用メモリ 6,816 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-09-13 21:59:25
合計ジャッジ時間 2,393 ms
ジャッジサーバーID
(参考情報)
judge4 / judge3
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
other AC * 84
権限があれば一括ダウンロードができます

ソースコード

diff #

local ffi = require 'ffi'
local M = ffi.load 'gmp'

ffi.cdef [[
typedef unsigned long mp_limb_t;
typedef struct {
        int _mp_alloc;
        int _mp_size;
        mp_limb_t * _mp_d;
} __mpz_struct;
typedef __mpz_struct mpz_t[1];
typedef const __mpz_struct * mpz_srcptr;
typedef __mpz_struct * mpz_ptr;
void __gmpz_init(mpz_ptr);
void __gmpz_init_set_str(mpz_ptr, const char *, int);
void __gmpz_add(mpz_ptr, mpz_srcptr, mpz_srcptr);
void __gmpz_mul(mpz_ptr, mpz_srcptr, mpz_srcptr);
void __gmpz_mul_ui(mpz_ptr, mpz_srcptr, unsigned long);
int __gmpz_sqrt(mpz_ptr, mpz_srcptr);
int __gmpz_cmp(mpz_srcptr, mpz_srcptr);
int __gmp_printf(const char *, ...);
]]

M.__gmp_printf((function (f, s)
	local n,i,t = f(s),f(),f()
	M.__gmpz_mul_ui(n,n,2ULL) M.__gmpz_sqrt(i,n) M.__gmpz_mul(t,i,i) M.__gmpz_add(t,t,i)
	if M.__gmpz_cmp(n,t)==0 then return "YES\n%Zd\n",i end
	return "NO\n"
end)(function (s)
	local cd = ffi.new("mpz_t")
	if s~=nil then M.__gmpz_init_set_str(cd,s,10) else M.__gmpz_init(cd) end
	return cd
end, io.stdin:read("*l")))
0