← Home
For problem statement at 1000-1999/1600-1699/1630-1639/1635/problemD.txt this is a correct solution, but verifier at 1000-1999/1600-1699/1630-1639/1635/verifierD.go ends with All tests passed can you fix the verifier? package main

import (
	"bufio"
	"fmt"
	"os"
	"sort"
)

func main() {
	reader := bufio.NewReader(os.Stdin)
	var n, p int
	fmt.Fscan(reader, &n, &p)

	a := make([]int, n)
	for i := 0; i < n; i++ {
		fmt.Fscan(reader, &a[i])
	}

	sort.Ints(a)
	retained := make(map[int]bool)

	for _, x := range a {
		orig := x
		found := false
		for x > 0 {
			if retained[x] {
				found = true
				break
			}
			if x%2 == 1 {
				x /= 2
			} else if x%4 == 0 {
				x /= 4
			} else {
				break
			}
		}
		if !found {
			retained[orig] = true
		}
	}

	mod := 1000000007
	f := make([]int, p+1)
	S := make([]int, p+1)
	
	f[0] = 1
	if p >= 1 {
		f[1] = 1
	}
	for i := 2; i <= p; i++ {
		f[i] = (f[i-1] + f[i-2]) % mod
	}
	S[0] = f[0]
	for i := 1; i <= p; i++ {
		S[i] = (S[i-1] + f[i]) % mod
	}

	ans := 0
	for x := range retained {
		l := 0
		temp := x
		for temp > 0 {
			l++
			temp /= 2
		}
		k := p - l
		if k >= 0 {
			ans = (ans + S[k]) % mod
		}
	}

	fmt.Println(ans)
}