結果

問題 No.793 うし数列 2
コンテスト
ユーザー 826814741_6
提出日時 2019-03-05 17:09:10
言語 Lua
(LuaJit 2.1.1734355927)
結果
WA  
実行時間 -
コード長 2,109 bytes
コンパイル時間 96 ms
コンパイル使用メモリ 6,944 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-06-23 14:24:28
合計ジャッジ時間 1,019 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 2
other AC * 5 WA * 16
権限があれば一括ダウンロードができます

ソースコード

diff # 

local ffi = require 'ffi'
local M = ffi.load 'ssl'
ffi.cdef [[
typedef struct bignum_st {
	uint64_t *d;
	int top;
	int dmax;
	int neg;
	int flags;
} BN;
typedef BN * BN_ptr;
typedef const BN * BN_srcptr;

BN_ptr BN_new(void);
void BN_free(BN_ptr);

BN_ptr BN_copy(BN_ptr, BN_srcptr);

int BN_set_word(BN_ptr, uint64_t);
int BN_sub_word(BN_ptr, uint64_t);

char * BN_bn2dec(BN_srcptr);
void CRYPTO_free(void *, const char *, int);

typedef struct bignum_pool_item {
	BN vals[16]; // vals[BN_CTX_POOL_SIZE];
	struct bignum_pool_item *prev, *next;
} BN_POOL_ITEM;
typedef struct bignum_pool {
	BN_POOL_ITEM *head, *current, *tail;
	unsigned used, size;
} BN_POOL;
typedef struct bignum_ctx_stack {
	unsigned int *indexes;
	unsigned int depth, size;
} BN_STACK;
typedef struct bignum_ctx {
	BN_POOL pool;
	BN_STACK stack;
	unsigned int used;
	int err_stack;
	int too_many;
	int flags;
} BN_CTX;
typedef BN_CTX * BN_CTX_ptr;

BN_CTX_ptr BN_CTX_new(void);
void BN_CTX_free(BN_CTX_ptr);

int BN_div(BN_ptr, BN_ptr, BN_srcptr, BN_srcptr, BN_CTX_ptr);
int BN_mod_add(BN_ptr, BN_srcptr, BN_srcptr, BN_srcptr, BN_CTX_ptr);
int BN_mod_mul(BN_ptr, BN_srcptr, BN_srcptr, BN_srcptr, BN_CTX_ptr);
int BN_mod_exp(BN_ptr, BN_srcptr, BN_srcptr, BN_srcptr, BN_CTX_ptr);
]]

function bn2str(bn)
	return ffi.string(ffi.gc(M.BN_bn2dec(bn),function (a) M.CRYPTO_free(a,"",0) end))
end

function bn2num(bn)
	return tonumber(bn2str(bn))
end

function ul2bn(ul)
	local bn = ffi.gc(M.BN_new(),M.BN_free)
	if ul~=nil then M.BN_set_word(bn,ul) end
	return bn
end

function getCTX()
	return ffi.gc(M.BN_CTX_new(),M.BN_CTX_free)
end

local readUL = (function ()
	ffi.cdef 'int scanf(const char *, ...);'
	local C,c = ffi.C,ffi.new("long[1]")
	return function () C.scanf("%ld",c) return c[0] end
end)()

do
local ctx,d,ul = getCTX(),ul2bn(1000000007ULL),readUL()
local a,b = ul2bn(),ul2bn()

M.BN_mod_exp(a, ul2bn(10ULL), ul2bn(ul), d, ctx)
M.BN_copy(b, a)

M.BN_sub_word(b, 1ULL)
M.BN_div(b, nil, b, ul2bn(9ULL), ctx)
-- M.BN_div(nil, b, b, d, ctx)
M.BN_mod_mul(b, b, ul2bn(3ULL), d, ctx)

M.BN_mod_add(a, a, b, d, ctx)

print(bn2str(a))
end
0