#253988 (C++11) No.681 Fractal Gravity Glue

• 問題: No.681 Fractal Gravity Glue
• ユーザー: startcpp
• 提出日時: 2018-04-27 23:00:14
• 言語: C++11 (gcc 11.4.0)
• 結果: AC
• 実行時間: 2 ms / 2,000 ms
• コード長: 901 bytes
• コンパイル時間: 451 ms
• コンパイル使用メモリ: 55,896 KB
• 実行使用メモリ: 5,376 KB
• 最終ジャッジ日時: 2024-06-27 22:09:28
• 合計ジャッジ時間: 1,131 ms
• ジャッジサーバーID （参考情報）: judge5 / judge4

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	2 ms; 5,248 KB
testcase_01	AC	2 ms; 5,376 KB
testcase_02	AC	1 ms; 5,376 KB
testcase_03	AC	2 ms; 5,376 KB
testcase_04	AC	1 ms; 5,376 KB
testcase_05	AC	2 ms; 5,376 KB
testcase_06	AC	2 ms; 5,376 KB
testcase_07	AC	2 ms; 5,376 KB
testcase_08	AC	2 ms; 5,376 KB
testcase_09	AC	1 ms; 5,376 KB
testcase_10	AC	2 ms; 5,376 KB
testcase_11	AC	1 ms; 5,376 KB
testcase_12	AC	2 ms; 5,376 KB
testcase_13	AC	1 ms; 5,376 KB
testcase_14	AC	2 ms; 5,376 KB
testcase_15	AC	2 ms; 5,376 KB
testcase_16	AC	2 ms; 5,376 KB
testcase_17	AC	2 ms; 5,376 KB
testcase_18	AC	2 ms; 5,376 KB
testcase_19	AC	2 ms; 5,376 KB

ソースコード

#include <iostream>
#define int long long
using namespace std;

int mod = 1000000007;

int power(int a, int n) {
	if (n == 0) return 1;
	if (n & 1) return (a * power(a, n - 1));
	return power(a * a, n / 2);
}

int powmod(int a, int n, int mod) {
	if (n == 0) return 1;
	if (n & 1) return (a * powmod(a, n - 1, mod)) % mod;
	return powmod((a * a) % mod, n / 2, mod);
}

int fs(int b, int d) {
	int res = (d + 1) * (powmod(d + 1, b, mod) + mod - 1) % mod * powmod(d, mod - 2, mod) % mod;
	return (res + mod - b) % mod;
}

int fn(int b, int d) {
	return power(d + 1, b) - 1;
}

int n, b, d;

signed main() {
	int i;
	
	cin >> n >> b >> d;
	
	int ans = fs(b, d);
	for (i = 0; fn(i, d) <= n; i++); i--;
	
	for (; i >= 0; i--) {
		int res = power(d + 1, i);
		int syo = n / res;
		int hiku = syo * (fs(i, d) + i + 1) % mod;
		ans = (ans + mod - hiku) % mod;
		n %= res;
	}
	cout << ans << endl;
	return 0;
}