結果

問題 No.2709 1975 Powers
ユーザー ID 21712
提出日時 2024-11-08 21:51:14
言語 Go
(1.23.4)
結果
AC  
実行時間 8 ms / 2,000 ms
コード長 882 bytes
コンパイル時間 13,631 ms
コンパイル使用メモリ 223,508 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-08 21:51:32
合計ジャッジ時間 14,748 ms
ジャッジサーバーID
(参考情報)
judge2 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 2
other AC * 25
権限があれば一括ダウンロードができます

ソースコード

diff #

package main

import . "fmt"
import . "slices"
import . "math/big"

func main() {
	var n int
	var p,q int64
	Scan(&n,&p,&q)
	x:=make([]int64,n)
	for i:=range x {
		Scan(&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)
}
0