結果

問題 No.1086 桁和の桁和2
ユーザー kriiikriii
提出日時 2020-06-19 22:41:44
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
WA  
実行時間 -
コード長 2,579 bytes
コンパイル時間 425 ms
コンパイル使用メモリ 49,368 KB
実行使用メモリ 6,948 KB
最終ジャッジ日時 2024-07-03 15:15:44
合計ジャッジ時間 49,034 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge3
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 2 ms
6,812 KB
testcase_01 AC 2 ms
6,944 KB
testcase_02 AC 2 ms
6,940 KB
testcase_03 AC 2 ms
6,944 KB
testcase_04 AC 2 ms
6,940 KB
testcase_05 AC 42 ms
6,944 KB
testcase_06 AC 2 ms
6,944 KB
testcase_07 AC 2 ms
6,940 KB
testcase_08 AC 2 ms
6,940 KB
testcase_09 AC 1 ms
6,940 KB
testcase_10 AC 28 ms
6,944 KB
testcase_11 AC 9 ms
6,940 KB
testcase_12 AC 3 ms
6,940 KB
testcase_13 AC 37 ms
6,940 KB
testcase_14 AC 22 ms
6,944 KB
testcase_15 WA -
testcase_16 WA -
testcase_17 WA -
testcase_18 TLE -
testcase_19 WA -
testcase_20 WA -
testcase_21 WA -
testcase_22 TLE -
testcase_23 WA -
testcase_24 WA -
testcase_25 WA -
testcase_26 WA -
testcase_27 WA -
testcase_28 WA -
testcase_29 WA -
testcase_30 TLE -
testcase_31 TLE -
testcase_32 TLE -
testcase_33 TLE -
testcase_34 TLE -
testcase_35 AC 45 ms
6,940 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <stdio.h>
#include <algorithm>
using namespace std;

int N, D[100100];
long long L[100100], R[100100];

const long long mod = 1000000007;

struct arr{
	arr(){
		for (int i = 0; i < 9; i++) a[i] = 0;
	}
	long long a[9];
	arr operator +(arr t){
		arr r;
		for (int i = 0; i < 9; i++){
			r.a[i] = (a[i] + t.a[i]) % mod;
		}
		return r;
	}
};

struct matrix{
	matrix(){
		for (int i = 0; i < 9; i++) for (int j = 0; j < 9; j++) a[i][j] = 0;
	}
	long long a[9][9];

	matrix operator +(matrix t){
		matrix r;
		for (int i = 0; i < 9; i++) for (int j = 0; j < 9; j++){
			r.a[i][j] = (a[i][j] + t.a[i][j]) % mod;
		}
		return r;
	}
	matrix operator *(matrix t){
		matrix r;
		for (int i = 0; i < 9; i++) for (int j = 0; j < 9; j++){
			r.a[i][j] = 0;
			for (int k = 0; k < 9; k++){
				r.a[i][j] = (r.a[i][j] + a[i][k] * t.a[k][j]) % mod;
			}
		}
		return r;
	}
	arr operator *(arr t){
		arr r;
		for (int i = 0; i < 9; i++) for (int j = 0; j < 9; j++){
			r.a[i] = (r.a[i] + a[i][j] * t.a[j]) % mod;
		}
		return r;
	}
}iden, base[70];

arr sum(matrix t)
{
	arr r;
	for (int i = 0; i < 9; i++) for (int j = 0; j < 9; j++){
		r.a[i] = (r.a[i] + t.a[i][j]) % mod;
	}
	return r;
}

pair<arr, arr> fpow(int i, long long n)
{
	if (n <= 0) return { sum(iden), arr() };
	if (n == 1) return { sum(base[i]), sum(iden) };

	auto half = fpow(i + 1, n / 2);
	half.second = (iden + base[i]) * half.second;
	if (n % 2){
		half.second = half.second + half.first;
		half.first = base[i] * half.first;
	}
	return half;
}

int main()
{
	for (int i = 0; i < 9; i++) iden.a[i][i] = 1;
	for (int i = 0; i < 9; i++) for (int j = 0; j < 10; j++) base[0].a[i][(i + j) % 9]++;
	for (int i = 1; i < 70; i++) base[i] = base[i - 1] * base[i - 1];

	scanf ("%d", &N);
	for (int i = 0; i < N; i++) scanf ("%lld", &L[i]);
	for (int i = 0; i < N; i++) scanf ("%lld", &R[i]);
	for (int i = 0; i < N; i++) scanf ("%d", &D[i]);

	int s = 0;
	while (s < N && D[s] == 0) s++;

	for (int i = s; i < N; i++) if (D[i] == 0){
		puts("0");
		return 0;
	}

	for (int i = s; i < N; i++){
		L[i - s] = L[i];
		R[i - s] = R[i];
		D[i - s] = D[i];
	}
	N -= s;

	long long ans = 1; int ld = 0;
	for (int i = 0; i < N; i++){
		int nd = (D[i] + 9 - ld) % 9;

		auto upp = fpow(0, R[i] - L[i]);

		arr p = upp.second;
		long long low = L[i];
		for (int b = 0; low; b++){
			if (low & 1) p = base[b] * p;
			low /= 2;
		}
		ans = ans * (p.a[nd] + (nd == 0)) % mod;
		ld = nd;
	}

	bool g = N > 0;
	for (int i = 0; i < N; i++) if (D[i] != 9) g = 0;
	if (g) ans = (ans + mod - 1) % mod;
	
	printf ("%lld\n", ans);

	return 0;
}
0