結果

問題 No.16 累乗の加算
コンテスト
ユーザー nobigomu
提出日時 2018-04-26 18:30:19
言語 Lua
(LuaJit 2.1.1734355927)
結果
AC  
実行時間 2 ms / 5,000 ms
コード長 1,030 bytes
コンパイル時間 97 ms
コンパイル使用メモリ 5,248 KB
実行使用メモリ 5,376 KB
最終ジャッジ日時 2024-06-27 21:13:11
合計ジャッジ時間 689 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge1
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
other AC * 14
権限があれば一括ダウンロードができます

ソースコード

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_d(mpz_ptr, double);
void __gmpz_set_str(mpz_ptr, const char *, int);
void __gmpz_add(mpz_ptr, mpz_srcptr, mpz_srcptr);
void __gmpz_mod(mpz_ptr, mpz_srcptr, mpz_srcptr);
void __gmpz_powm(mpz_ptr, mpz_srcptr, mpz_srcptr, mpz_srcptr);
int __gmp_printf(const char *, ...);
]]

M.__gmp_printf("%Zd\n", (function (f, x, _, s)
	local b,m,e,r,t = f(x),f(1000003),f(),f(),f()
	for d in s:gmatch("%d+") do
		M.__gmpz_set_str(e,d,10) M.__gmpz_powm(r,b,e,m) M.__gmpz_add(t,t,r)
	end
	M.__gmpz_mod(t,t,m)
	return t
end)(function (n)
	local cd = ffi.new("mpz_t")
	if n~=nil then M.__gmpz_init_set_d(cd,n) else M.__gmpz_init(cd) end
	return cd
end, io.stdin:read("*n"), io.stdin:read("*l"), io.stdin:read("*l")))
0