結果

問題 No.462 6日知らずのコンピュータ
ユーザー yuki2006
提出日時 2016-12-12 22:04:33
言語 Go
(1.23.4)
結果
AC  
実行時間 2 ms / 2,000 ms
コード長 2,913 bytes
コンパイル時間 15,228 ms
コンパイル使用メモリ 228,524 KB
実行使用メモリ 6,820 KB
最終ジャッジ日時 2024-11-29 18:35:12
合計ジャッジ時間 17,211 ms
ジャッジサーバーID
(参考情報)
judge2 / judge4
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 84
権限があれば一括ダウンロードができます

ソースコード

diff #

package main

import (
	"bufio"
	"os"
	"strconv"
	"fmt"
)

func main() {
	mod := int(1e9) + 7
	sc := NewScanner()
	N := sc.NextInt()
	K := sc.NextInt()
	if K == 0 {
		fmt.Println(factorial(N, mod))
		return
	}

	a := sc.NextIntArray()
	// ビットフラグ数が小さい順に並べる
	bitCheck := map[int]int{}

	for i := K - 1; i >= 0; i-- {
		b := bitCount(a[i])
		if _, ok := bitCheck[b]; ok {
			fmt.Println(0)
			return
		}
		bitCheck[b] = a[i]
	}
	prevB := 0
	ans := 1
	for b := 1; b <= N; b++ {
		if _, ok := bitCheck[b]; ok {
			if bitCheck[b] & bitCheck[prevB] != bitCheck[prevB] {
				fmt.Println(0)
				return
			}
			ans *= factorial(b - prevB, mod)
			ans %= mod
			prevB = b
		}
	}
	ans *= factorial(N - prevB, mod)
	ans %= mod

	fmt.Println(ans)
}

func bitCount(n int) int {
	count := 0
	for n > 0 {
		if n % 2 == 1 {
			count++
		}
		n /= 2
	}
	return count
}

func factorial(k int, mod int) int {
	ans := 1
	for i := 1; i <= k; i++ {
		ans *= i
		ans %= mod
	}
	return ans
}

func fillInt(arr []int, v int) {
	for i := 0; i < len(arr); i++ {
		arr[i] = v
	}
}

type Scanner struct {
	r   *bufio.Reader
	buf []byte
	p   int
}

func NewScanner() *Scanner {
	rdr := bufio.NewReaderSize(os.Stdin, 1000)
	return &Scanner{r:rdr}
}
func (s *Scanner) Next() string {
	s.pre()
	start := s.p
	for ; s.p < len(s.buf); s.p++ {
		if s.buf[s.p] == ' ' {
			break
		}
	}
	result := string(s.buf[start:s.p])
	s.p++
	return result
}
func (s *Scanner) NextLine() string {
	s.pre()
	start := s.p
	s.p = len(s.buf)
	return string(s.buf[start:])
}
func (s *Scanner) NextInt() int {
	v, _ := strconv.Atoi(s.Next())
	return v
}
func (s *Scanner) NextInt64() int64 {
	v, _ := strconv.ParseInt(s.Next(), 10, 64)
	return v
}

func (s *Scanner) NextIntArray() []int {
	s.pre()
	start := s.p
	result := []int{}
	for ; s.p < len(s.buf) + 1; s.p++ {
		if s.p == len(s.buf) || s.buf[s.p] == ' ' {
			v, _ := strconv.ParseInt(string(s.buf[start:s.p]), 10, 0)
			result = append(result, int(v))
			start = s.p + 1
		}
	}

	return result
}
func (s *Scanner) NextInt64Array() []int64 {
	s.pre()
	start := s.p
	result := []int64{}
	for ; s.p < len(s.buf) + 1; s.p++ {
		if s.p == len(s.buf) || s.buf[s.p] == ' ' {
			v, _ := strconv.ParseInt(string(s.buf[start:s.p]), 10, 64)
			result = append(result, v)
			start = s.p + 1
		}
	}
	return result
}

func (s *Scanner) NextMap() map[int]bool {
	s.pre()
	start := s.p
	mp := map[int]bool{}
	for ; s.p < len(s.buf); s.p++ {
		if s.buf[s.p] == ' ' {
			v, _ := strconv.Atoi(string(s.buf[start:s.p]))
			mp[v] = true
			start = s.p + 1
		}
	}
	v, _ := strconv.Atoi(string(s.buf[start:s.p]))
	mp[v] = true

	return mp
}

func (s *Scanner) pre() {
	if s.p >= len(s.buf) {
		s.readLine()
		s.p = 0
	}
}
func (s *Scanner) readLine() {
	s.buf = make([]byte, 0)
	for {
		l, p, e := s.r.ReadLine()
		if e != nil {
			panic(e)
		}
		s.buf = append(s.buf, l...)
		if !p {
			break
		}
	}
}

0