#1025299 (Go) No.2709 1975 Powers

• 問題: No.2709 1975 Powers
• ユーザー: ID 21712
• 提出日時: 2024-11-08 22:29:36
• 言語: Go (1.22.1)
• 結果: AC
• 実行時間: 8 ms / 2,000 ms
• コード長: 947 bytes
• コンパイル時間: 18,323 ms
• コンパイル使用メモリ: 219,688 KB
• 実行使用メモリ: 5,248 KB
• 最終ジャッジ日時: 2024-11-08 22:30:05
• 合計ジャッジ時間: 17,325 ms
• ジャッジサーバーID （参考情報）: judge1 / judge2

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	1 ms; 5,248 KB
testcase_01	AC	2 ms; 5,248 KB
testcase_02	AC	4 ms; 5,248 KB
testcase_03	AC	5 ms; 5,248 KB
testcase_04	AC	7 ms; 5,248 KB
testcase_05	AC	6 ms; 5,248 KB
testcase_06	AC	4 ms; 5,248 KB
testcase_07	AC	2 ms; 5,248 KB
testcase_08	AC	5 ms; 5,248 KB
testcase_09	AC	6 ms; 5,248 KB
testcase_10	AC	2 ms; 5,248 KB
testcase_11	AC	3 ms; 5,248 KB
testcase_12	AC	5 ms; 5,248 KB
testcase_13	AC	7 ms; 5,248 KB
testcase_14	AC	4 ms; 5,248 KB
testcase_15	AC	5 ms; 5,248 KB
testcase_16	AC	5 ms; 5,248 KB
testcase_17	AC	3 ms; 5,248 KB
testcase_18	AC	5 ms; 5,248 KB
testcase_19	AC	6 ms; 5,248 KB
testcase_20	AC	4 ms; 5,248 KB
testcase_21	AC	5 ms; 5,248 KB
testcase_22	AC	8 ms; 5,248 KB
testcase_23	AC	8 ms; 5,248 KB
testcase_24	AC	6 ms; 5,248 KB
testcase_25	AC	6 ms; 5,248 KB
testcase_26	AC	7 ms; 5,248 KB

ソースコード

package main

import . "fmt"
import . "slices"
import . "math/big"
import . "os"
import bf "bufio"

func main() {
	rd:=bf.NewReader(Stdin)
	var n int
	var p,q int64
	Fscan(rd,&n,&p,&q)
	x:=make([]int64,n)
	for i:=range x {
		Fscan(rd,&x[i])
	}
	Sort(x)
	a:=make([]int64,n)
	b:=make([]int64,n)
	c:=make([]int64,n)
	d:=make([]int64,n)
	z1,z2:=new(Int),new(Int)
	e10:=NewInt(10)
	e9:=NewInt(9)
	e7:=NewInt(7)
	e5:=NewInt(5)
	m:=NewInt(p)
	for i,v:=range x {
		z1.SetInt64(v)
		a[i]=z2.Exp(e10,z1,m).Int64()
		b[i]=z2.Exp(e9,z1,m).Int64()
		c[i]=z2.Exp(e7,z1,m).Int64()
		d[i]=z2.Exp(e5,z1,m).Int64()
	}
	t:=make([][]int,int(p))
	for i,v:=range c {
		for _,w:=range d[i+1:] {
			u:=int((v+w)%p)
			t[u]=append(t[u],i)
		}
	}
	var ans int
	for i,v:=range a {
		for k,w:=range b {
			if k<=i {
				continue
			}
			u:=int((p+q-((v+w)%p))%p)
			if len(t[u])==0 {
				continue
			}
			s,_:=BinarySearch(t[u],k+1)
			ans+=len(t[u])-s
		}
	}
	Println(ans)
}