#143500 (C#(csc)) No.551 夏休みの思い出（２）

提出ソース

結果

問題	No.551 夏休みの思い出（２）
ユーザー	kuuso1
提出日時	2017-01-08 03:53:46
言語	C#(csc) (csc 3.9.0)
結果	AC
実行時間	1,414 ms / 4,000 ms
コード長	4,830 bytes
コンパイル時間	1,719 ms
コンパイル使用メモリ	116,412 KB
実行使用メモリ	171,416 KB
最終ジャッジ日時	2024-05-09 23:12:34
合計ジャッジ時間	29,690 ms
ジャッジサーバーID （参考情報）	judge4 / judge3

このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示

入力	結果	実行時間実行使用メモリ
testcase_00	AC	33 ms 28,676 KB
testcase_01	AC	32 ms 26,380 KB
testcase_02	AC	32 ms 26,504 KB
testcase_03	AC	35 ms 26,424 KB
testcase_04	AC	36 ms 26,824 KB
testcase_05	AC	40 ms 26,444 KB
testcase_06	AC	43 ms 28,628 KB
testcase_07	AC	32 ms 26,372 KB
testcase_08	AC	32 ms 28,804 KB
testcase_09	AC	32 ms 28,672 KB
testcase_10	AC	32 ms 30,584 KB
testcase_11	AC	32 ms 26,624 KB
testcase_12	AC	32 ms 26,508 KB
testcase_13	AC	32 ms 26,500 KB
testcase_14	AC	32 ms 26,500 KB
testcase_15	AC	32 ms 26,372 KB
testcase_16	AC	33 ms 28,680 KB
testcase_17	AC	50 ms 33,936 KB
testcase_18	AC	87 ms 45,948 KB
testcase_19	AC	84 ms 47,944 KB
testcase_20	AC	84 ms 45,816 KB
testcase_21	AC	106 ms 54,236 KB
testcase_22	AC	37 ms 33,400 KB
testcase_23	AC	44 ms 34,492 KB
testcase_24	AC	66 ms 38,904 KB
testcase_25	AC	48 ms 34,908 KB
testcase_26	AC	64 ms 39,984 KB
testcase_27	AC	1,230 ms 171,164 KB
testcase_28	AC	1,200 ms 171,168 KB
testcase_29	AC	990 ms 169,380 KB
testcase_30	AC	1,185 ms 171,032 KB
testcase_31	AC	987 ms 170,908 KB
testcase_32	AC	1,199 ms 169,116 KB
testcase_33	AC	1,227 ms 168,864 KB
testcase_34	AC	1,185 ms 167,196 KB
testcase_35	AC	1,046 ms 171,288 KB
testcase_36	AC	1,144 ms 166,936 KB
testcase_37	AC	1,414 ms 171,032 KB
testcase_38	AC	1,363 ms 171,056 KB
testcase_39	AC	1,268 ms 171,416 KB
testcase_40	AC	1,294 ms 168,996 KB
testcase_41	AC	1,035 ms 169,256 KB
testcase_42	AC	1,236 ms 169,280 KB
testcase_43	AC	1,249 ms 169,124 KB
testcase_44	AC	1,258 ms 171,036 KB
testcase_45	AC	1,245 ms 171,300 KB
testcase_46	AC	1,368 ms 171,044 KB
testcase_47	AC	32 ms 26,628 KB
testcase_48	AC	31 ms 26,372 KB

権限があれば一括ダウンロードができます

コンパイルメッセージ

Microsoft (R) Visual C# Compiler version 3.9.0-6.21124.20 (db94f4cc)
Copyright (C) Microsoft Corporation. All rights reserved.

ソースコード

raw source code

using System;
using System.Collections;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.IO;
using System.Diagnostics;
using System.Diagnostics.Contracts;
using CT = System.Diagnostics.Contracts.Contract;
class TEST{
	static void Main(){
		Sol mySol =new Sol();
		mySol.Solve();
	}
}

class Sol{
	public void Solve(){
		
		// O( W + Q*(H+log(mod)) ) where mod / W = H, W:Initialize of BabyStep, H: Giant Step query;
		
		StringBuilder sb = new StringBuilder();
		long W = (int) 5e6;
		long H = mod / W + 1;
		DiscreteLog.Init(mod, primRoot, W);
		
		long Inv2 = ModInv(2,mod);
		
		for(int i=0;i<Q;i++){
			
			long a = 1;
			long b = B[i] * ModInv(A[i],mod) % mod;
			long c = C[i] * ModInv(A[i],mod) % mod;
			long D = ((b * b) % mod +mod - ((4 * a * c) % mod) ) % mod;
			if(D == 0){
				// multiple root case:
				long sol = (mod - b) * Inv2 % mod;
				sb.AppendLine(sol.ToString());
				continue;
			}
			long idx = DiscreteLog.Calc(D);
			if(idx % 2 != 0){
				// no SquareRoot for D;
				sb.AppendLine("-1");
				continue;
			}else{
				long sqrt = ModPow(primRoot,idx/2,mod);
				long[] sols = new long[]{
					( ( mod - b + sqrt ) % mod ) * Inv2 % mod,
					( ( mod - b + mod - sqrt) % mod ) * Inv2 % mod
				};
				Array.Sort(sols);
				sb.AppendLine(String.Format("{0} {1}",sols[0],sols[1]));
			}
		}
		Console.Write(sb.ToString());
	}
	
	int Q;
	long[] A,B,C;
	static long mod;
	long primRoot;
	
	public Sol(){
		
		// input read
		var d = rla();
		mod = d[0];
		primRoot = d[1];
		Q = ri();
		
		A = new long[Q];
		B = new long[Q];
		C = new long[Q];
		
		for(int i=0;i<Q;i++){
			d = rla();
			A[i] = d[0];
			B[i] = d[1];
			C[i] = d[2];
		}
		
	}
	
	
	static bool isPrimitiveRoot(long x, long mod){
		if( x == 0 || x == 1 ) return false;
		long w = (long)1e5;
		long h = mod / w;
		
		Dictionary<long,long> D = new Dictionary<long,long>((int)w+1);
		long xx = 1;
		for(long j=0;j<w;j++){
			if(D.ContainsKey(xx)){
				if(j >= mod - 1){
					return true;
				}else{
					return false;
				}
			}
			D.Add(xx,j);
			xx *= x;
			xx %= mod;
		}
		
		long nxx = ModInv(xx,mod);
		long yy = nxx;
		for(long i=1;i<h+1;i++){
			if(D.ContainsKey(yy)){
				long idx = w * i + D[yy];
				if(idx >= mod - 1){
					return true;
					
				}else{
					return false;
				}
			}
			yy *= nxx;
			yy %= mod;
		}
		return true;
	}
	
	
	
	static long ModInv(long x, long mod){
		long a = 0, b = 0, c = 0;
		if(x == 0) return 0;
		ExtGCD(x, mod, ref a, ref b, ref c);
		if(c != 1)return 0;
		return (a + mod) % mod;
	}
	
	static void ExtGCD(long x,long y,ref long a,ref long b,ref long c){
		long r0 = x; long r1 = y;
		long a0 = 1; long a1 = 0;
		long b0 = 0; long b1 = 1;
		long q1, r2, a2, b2;
		while(r1 > 0){
			q1 = r0 / r1;
			r2 = r0 % r1;
			a2 = a0 - q1 * a1;
			b2 = b0 - q1 * b1;
			r0 = r1; r1 = r2;
			a0 = a1; a1 = a2;
			b0 = b1; b1 = b2;
		}
		c = r0;
		a = a0;
		b = b0;
	}
	
	static long ModPow(long x,long k,long mod){
		if(k == 0)return 1;
		if(x == 0)return 0;
		long ret = 1;
		long xx = x;
		while(k>0){
			if((k&1)==1){ret *= xx; ret %= mod;}
			
			xx *= xx; xx %= mod;
			k>>=1;
		}
		return ret;
	}
	
	static String rs(){return Console.ReadLine();}
	static int ri(){return int.Parse(Console.ReadLine());}
	static long rl(){return long.Parse(Console.ReadLine());}
	static double rd(){return double.Parse(Console.ReadLine());}
	static String[] rsa(char sep=' '){return Console.ReadLine().Split(sep);}
	static int[] ria(char sep=' '){return Array.ConvertAll(Console.ReadLine().Split(sep),e=>int.Parse(e));}
	static long[] rla(char sep=' '){return Array.ConvertAll(Console.ReadLine().Split(sep),e=>long.Parse(e));}
	static double[] rda(char sep=' '){return Array.ConvertAll(Console.ReadLine().Split(sep),e=>double.Parse(e));}
}

class DiscreteLog{

	static long mod;
	static long primRoot;
	static long WInvPrimRoot;	// Pow(primRoot , -W)
	static long W,H;
	static Dictionary<long,long> WTable;
	
	public static void Init(long mod_, long primRoot_, long w){
		mod = mod_;
		primRoot = primRoot_;
		W = w;
		H = mod / w + 1;
		
		WTable = new Dictionary<long,long>((int)W+20);
		long xx = 1;
		for(long j=0;j<w;j++){
			if(WTable.ContainsKey(xx))break;
			WTable.Add(xx,j);
			xx *= primRoot;
			xx %= mod;
		}
		WInvPrimRoot = ModInv(xx,mod);
	}
	
	public static long Calc(long x){
		long nxx = WInvPrimRoot;
		long yy = x;
		for(long i=0;i<H;i++){
			if(WTable.ContainsKey(yy)){
				return W * i + WTable[yy];
			}
			yy *= nxx;
			yy %= mod;
		}
		return 0;
	}
	static long ModInv(long x,long mod){
		return ModPow(x, mod - 2, mod);
	}
	
	static long ModPow(long x,long k,long mod){
		if(k == 0)return 1;
		if(x == 0)return 0;
		long ret = 1;
		long xx = x;
		while(k>0){
			if((k&1)==1){ret *= xx; ret %= mod;}
			
			xx *= xx; xx %= mod;
			k>>=1;
		}
		return ret;
	}
}

yukicoder

結果

テストケース

コンパイルメッセージ

ソースコード