結果

問題 No.434 占い
ユーザー 37zigen
提出日時 2020-04-04 05:33:05
言語 Java
(openjdk 23)
結果
AC  
実行時間 960 ms / 2,000 ms
コード長 1,399 bytes
コンパイル時間 2,257 ms
コンパイル使用メモリ 81,960 KB
実行使用メモリ 68,080 KB
最終ジャッジ日時 2024-07-03 06:59:10
合計ジャッジ時間 11,533 ms
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 27
権限があれば一括ダウンロードができます

ソースコード

diff #

import java.util.Arrays;
import java.util.Scanner;

public class Main {

	public static void main(String[] args) {
		new Main().run();
	}
	
	int[] toint(char[] cs) {
		int[] ret=new int[cs.length];
		for(int i=0;i<ret.length;++i)ret[i]=(int)(cs[i]-'0');
		return ret;
	}
	
	int pow(int a,int n) {
		int ret=1;
		for(;n>0;n>>=1,a=a*a%9)if(n%2==1)ret=ret*a%9;
		return ret;
	}
	
	int inv(int a) {
		if(a%3==0)throw new AssertionError();
		return pow(a%9,5);
	}
	
	int count3(int a) {
		int ret=0;
		while(a%3==0) {
			a/=3;
			++ret;
		}
		return ret;
	}
	
	int mul(int...a) {
		int r=1;
		for(int v:a)r=(int)((long)r*v%9);
		return r;
	}

	void run() {
		Scanner sc=new Scanner(System.in);
		int T=sc.nextInt();
		while(T-->0) {
			int[] a=toint(sc.next().toCharArray());
			int n=a.length;
			int binom=1;
			int c3=0;
			int ans=0;
			boolean zero=Arrays.stream(a).filter(v->v==0).count()==n;
			for(int i=0;i<n;++i) {
				if(c3<0)throw new AssertionError();
				ans+=mul(pow(3,c3),binom,a[i]);
				ans%=9;
				int numerator=n-1-i;
				int denominator=i+1;
				if(i!=n-1) {
					while(numerator%3==0) {
						numerator/=3;++c3;
					}
					while(denominator%3==0) {
						denominator/=3;--c3;
					}
					binom=mul(binom,numerator,inv(denominator));
				}
			}
			System.out.println(zero?0:(ans==0?9:ans));
		}
	}
	
	void tr(Object...objects) {System.out.println(Arrays.deepToString(objects));}
}

0