結果

問題 No.462 6日知らずのコンピュータ
コンテスト
ユーザー yuki2006
提出日時 2016-12-12 21:19:58
言語 Go
(1.23.4)
結果
WA  
実行時間 -
コード長 2,771 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 15,677 ms
コンパイル使用メモリ 229,280 KB
実行使用メモリ 5,248 KB
最終ジャッジ日時 2024-11-29 17:59:40
合計ジャッジ時間 13,784 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 40 WA * 44
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

package main

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

func main() {
	sc := NewScanner()
	N := sc.NextInt()
	K := sc.NextInt()
	if K == 0 {
		fmt.Println(factorial(N))
		return
	}

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

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

	fmt.Println(factorial(N - K))
}

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

func factorial(k int) int {
	ans := 1
	for i := 1; i <= k; i++ {
		ans *= i
	}
	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