結果
| 問題 |
No.1189 Sum is XOR
|
| コンテスト | |
| ユーザー |
ccppjsrb
|
| 提出日時 | 2020-08-22 17:35:08 |
| 言語 | Go (1.23.4) |
| 結果 |
WA
|
| 実行時間 | - |
| コード長 | 2,528 bytes |
| コンパイル時間 | 14,362 ms |
| コンパイル使用メモリ | 219,168 KB |
| 実行使用メモリ | 13,636 KB |
| 最終ジャッジ日時 | 2024-10-15 11:22:58 |
| 合計ジャッジ時間 | 18,113 ms |
|
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | -- * 2 |
| other | WA * 3 TLE * 1 -- * 17 |
ソースコード
package main
import (
"bufio"
"fmt"
"os"
"strconv"
)
func configure(scanner *bufio.Scanner) {
scanner.Split(bufio.ScanWords)
scanner.Buffer(make([]byte, 1000005), 1000005)
}
func getNextString(scanner *bufio.Scanner) string {
scanned := scanner.Scan()
if !scanned {
panic("scan failed")
}
return scanner.Text()
}
func getNextInt(scanner *bufio.Scanner) int {
i, _ := strconv.Atoi(getNextString(scanner))
return i
}
func getNextInt64(scanner *bufio.Scanner) int64 {
i, _ := strconv.ParseInt(getNextString(scanner), 10, 64)
return i
}
func getNextFloat64(scanner *bufio.Scanner) float64 {
i, _ := strconv.ParseFloat(getNextString(scanner), 64)
return i
}
func main() {
fp := os.Stdin
wfp := os.Stdout
extra := 0
if os.Getenv("I") == "IronMan" {
fp, _ = os.Open(os.Getenv("END_GAME"))
extra = 100
}
scanner := bufio.NewScanner(fp)
configure(scanner)
writer := bufio.NewWriter(wfp)
defer func() {
r := recover()
if r != nil {
fmt.Fprintln(writer, r)
}
writer.Flush()
}()
solve(scanner, writer)
for i := 0; i < extra; i++ {
fmt.Fprintln(writer, "-----------------------------------")
solve(scanner, writer)
}
}
func solve(scanner *bufio.Scanner, writer *bufio.Writer) {
n := getNextInt(scanner)
k := getNextInt(scanner)
aa := make([]int, n)
var ans mint
dp := make([]mint, 1<<10)
for i := 0; i < n; i++ {
aa[i] = getNextInt(scanner)
dp[aa[i]]++
}
for k-1 > 0 {
dp2 := make([]mint, 1<<10)
for i := 0; i < 1<<10; i++ {
for j := i + 1; j < 1<<10; j++ {
if i&j > 0 {
continue
}
dp2[i|j].addAs(dp[i].mul(dp[j]))
}
}
dp = dp2
k--
}
for i := 0; i < 1<<10; i++ {
ans.addAs(dp[i])
}
fmt.Fprintln(writer, ans)
}
type mint int64
func (mt mint) mod() mint {
m := mint(998244353)
mt %= m
if mt < 0 {
return mt + m
}
return mt
}
func (mt mint) inv() mint {
return mt.pow(mint(0).sub(2))
}
func (mt mint) pow(n mint) mint {
p := mint(1)
for n > 0 {
if n&1 == 1 {
p.mulAs(mt)
}
mt.mulAs(mt)
n >>= 1
}
return p
}
func (mt mint) add(x mint) mint {
return (mt + x).mod()
}
func (mt mint) sub(x mint) mint {
return (mt - x).mod()
}
func (mt mint) mul(x mint) mint {
return (mt * x).mod()
}
func (mt mint) div(x mint) mint {
return mt.mul(x.inv())
}
func (mt *mint) addAs(x mint) *mint {
*mt = mt.add(x)
return mt
}
func (mt *mint) subAs(x mint) *mint {
*mt = mt.sub(x)
return mt
}
func (mt *mint) mulAs(x mint) *mint {
*mt = mt.mul(x)
return mt
}
func (mt *mint) divAs(x mint) *mint {
*mt = mt.div(x)
return mt
}
ccppjsrb