package main import ( "container/heap" . "fmt" "math/bits" "math/rand" "sort" "time" ) // type any = interface{} // for old go version func main() { var input string Scan(&input) if input == "LOCAL" { runLocal() return } s := NewSolver(input) s.Solve() ans := s.Answer() for i, v := range ans { if i > 0 { Print(" ") } Print(v) } Println() } func runLocal() { var n int Scan(&n) src := generate(n) code := encrypto(src) Println("N = ", n, ", |S| =", len(code), ", B = ", len(Sprintf("%b", n))) t := time.Now() s := NewSolver(code) s.Solve() ans := s.Answer() Println(time.Since(t)) m := map[int]bool{} for _, v := range ans { m[v] = true } ok := len(ans) == len(src) for i := range ans { ok = ok && m[i+1] } result := encrypto(ans) if ok && code == result { Println("AC") } else { Println("WA") } if n < 100 { Println(src) if len(code) < 200 { Println(code) } } if n < 100 { Println(ans) if len(code) < 200 { Println(result) } } } func perm0(n int) []int { a := make([]int, n, n) for i := range a { a[i] = i } return a } func encrypto(src []int) string { s := "" for _, v := range src { s = Sprintf("%s%b", s, v) } return s } func generate(n int) []int { src := perm0(n + 1)[1:] rand.Shuffle(len(src), func(i, j int) { src[i], src[j] = src[j], src[i] }) return src } func guessN(code string) (n int, bit uint) { totalbits := 0 for ; ; bit++ { ibitsum := int(bit+1) * (1 << bit) if totalbits+ibitsum > len(code) { rem := (len(code) - totalbits) / int(bit+1) if rem > 0 { n += rem bit++ } return } totalbits += ibitsum n += 1 << bit } } type Solver struct { code string n int bit uint found []bool line []int used []int } func NewSolver(code string) *Solver { n, bit := guessN(code) found := make([]bool, n+1, n+1) line := make([]int, len(code), len(code)) used := make([]int, len(code), len(code)) return &Solver{code, n, bit, found, line, used} } func (s *Solver) Clone() *Solver { code, n, bit := s.code, s.n, s.bit found := make([]bool, n+1, n+1) copy(found, s.found) line := make([]int, len(code), len(code)) copy(line, s.line) used := make([]int, len(code), len(code)) copy(used, s.used) return &Solver{code, n, bit, found, line, used} } func (s *Solver) Answer() []int { ans := make([]int, 0, s.n) for _, v := range s.line { if v > 0 { ans = append(ans, v) } } return ans } func (s *Solver) Solve() { for n, bit := s.n, s.bit; bit > s.bit-2; n, bit = 1<<(bit-1)-1, bit-1 { for ok := true; ok; ok, _ = s.fill(n, bit) { } } t := time.After(7 * time.Second) ch := make(chan *Solver) go s.doRandomFill(ch) go s.doSomethings(ch) select { case result := <-ch: if result != nil { copy(s.line, result.line) } case <-t: } } func (s *Solver) doRandomFill(result chan *Solver) { ch := time.After(6 * time.Second) outer: for { select { case <-ch: return default: } t := s.Clone() for n, bit := t.n, t.bit; bit > 0; n, bit = 1<<(bit-1)-1, bit-1 { for { ok, err := t.fill(n, bit) if err != nil { continue outer } if !ok { break } } if err := t.randomFill(n, bit); err != nil { continue outer } } result <- t break } } func (s *Solver) doSomethings(result chan *Solver) { ch := time.After(6 * time.Second) for { t := s.Clone() if err := t.somethings(); err == nil { result <- t return } select { case <-ch: result <- t return default: } } } type Ints interface { Get(i int) int } type ZeroInts struct{} func (z ZeroInts) Get(i int) int { return 0 } type IntSlice []int func (s IntSlice) Get(i int) int { return s[i] } func (s *Solver) scan(n int, bit uint) (position [][]int) { position = make([][]int, 1<= (1<<(bit-1)) && v <= n { if i+1 == len(s.code) || s.code[i+1] == '1' { position[v] = append(position[v], i-int(bit-1)) } } } return } var ( errFillNotFound = Errorf("ErrFillNotFound") errFillConflict = Errorf("ErrFillConflict") ) func (s *Solver) fill(n int, bit uint) (bool, error) { position := s.scan(n, bit) found := false for v := 1 << (bit - 1); v <= n; v++ { if len(position[v]) != 1 || s.found[v] { if !s.found[v] && len(position[v]) == 0 { return false, errFillNotFound } continue } found = true s.found[v] = true p := position[v][0] s.line[p] = v for i := 0; i < int(bit); i++ { if s.used[p+i] != 0 { return false, errFillConflict } s.used[p+i] = 1 } } return found, nil } func (s *Solver) randomFill(n int, bit uint) error { position := s.scan(n, bit) indexes := perm0(n + 1)[1<<(bit-1):] sort.Slice(indexes, func(i, j int) bool { return len(position[indexes[i]]) < len(position[indexes[j]]) }) for _, v := range indexes { if s.found[v] { continue } if len(position[v]) == 0 { return errFillNotFound } m := rand.Intn(len(position[v])) found := false outer: for i := range position[v] { k := (i + m) % len(position[v]) p := position[v][k] for _, u := range s.used[p : p+int(bit)] { if u != 0 { continue outer } } found = true s.found[v] = true s.line[p] = v for i := 0; i < int(bit); i++ { s.used[p+i] = 1 } break } if !found { return errFillConflict } } return nil } var ( errSomethingsInvalid = Errorf("ErrSomethingsInvalid") errSomethingsTimeout = Errorf("ErrSomethingsTimeout") ) const DEBUG_SOMETHINGS = false func (s *Solver) somethings() error { var index []int if DEBUG_SOMETHINGS { index = make([]int, len(s.code), len(s.code)) for i := range index { index[i] = i % 10 } } counts := make([]int, int(s.bit)+1, int(s.bit)+1) for i := 1; i <= s.n; i++ { if !s.found[i] { counts[bits.Len(uint(i))]++ } } if DEBUG_SOMETHINGS { Println(counts) Println(s.line) } used := make([]int, len(s.code), len(s.code)) // make initial positions position := make([][]int, 1<= (1<<(s.bit-1)) && v <= s.n && !s.found[v] { if i+1 == len(s.code) || s.code[i+1] == '1' { position[v] = append(position[v], i-int(s.bit-1)) } } } } pq := make(PriorityQueue, 0, s.n) for v := 1 << (s.bit - 1); v <= s.n; v++ { if s.found[v] || len(position[v]) == 0 { continue } random := rand.Intn(64) item := &Item{ value: v, random: random, priority: len(position[v])*64 + random, } pq = append(pq, item) } heap.Init(&pq) ch := time.After(3 * time.Second) for { select { case <-ch: return errSomethingsTimeout default: } n, bit := s.n, s.bit for bit > 0 && counts[int(bit)] == 0 { n = 1<<(bit-1) - 1 bit-- } if bit == 0 { break } if bit < s.bit { for i := 1 << (bit - 1); i <= n; i++ { position[i] = nil } var v, u int for i, c := range s.code { v = (v<<1 + int(c-'0')) & (1<= (1<<(bit-1)) && v <= n && !s.found[v] { if i+1 == len(s.code) || s.code[i+1] == '1' { position[v] = append(position[v], i-int(bit-1)) } } } for i := 1 << (bit - 1); i <= n; i++ { if s.found[i] { continue } random := rand.Intn(64) item := &Item{ value: i, random: random, priority: len(position[i])*64 + random, } heap.Push(&pq, item) } } for pq.Len() > 0 { select { case <-ch: return errSomethingsTimeout default: } item := heap.Pop(&pq).(*Item) v := item.value.(int) bit = uint(bits.Len(uint(v))) found, breakable := false, -1 for i := range position[v] { k := (i + item.random) % len(position[v]) p := position[v][k] u := 0 for b := 0; b < int(bit); b++ { u += used[p+b] } if u != 0 { breakable = p continue } found = true counts[int(bit)]-- s.found[v] = true s.line[p] = v for b := 0; b < int(bit); b++ { used[p+b] = 1 } if DEBUG_SOMETHINGS { Println("FOUND", v, bit, counts, p) Println(s.line) Println(used) Println(index) } break } if found { continue } if breakable < 0 { rp := rand.Intn(len(s.code)) vv, uu := 0, 0 for i := rp; i < len(s.code); i++ { vv = (vv<<1 + int(s.code[i]-'0')) & (1<= 0; b++ { if zv := s.line[bp-b]; zv == 0 { continue } else { zbit := bits.Len(uint(zv)) for zb := 0; zb < int(zbit); zb++ { used[bp-b+zb] = 0 } counts[zbit]++ s.found[zv] = false s.line[bp-b] = 0 random := rand.Intn(64) item := &Item{ value: zv, random: random, priority: len(position[zv]) + random, } heap.Push(&pq, item) if zbit < int(s.bit) { position[zv] = nil } if DEBUG_SOMETHINGS { Println("REVERT", zv, zbit, counts, bp-b) Println(s.line) Println(used) Println(index) } break } } } for b := 0; b < int(bit); b++ { if zv := s.line[bp+b]; zv == 0 { continue } else { zbit := bits.Len(uint(zv)) for zb := 0; zb < int(zbit); zb++ { used[bp+b+zb] = 0 } counts[zbit]++ s.found[zv] = false s.line[bp+b] = 0 random := rand.Intn(64) item := &Item{ value: zv, random: random, priority: len(position[zv]) + random, } heap.Push(&pq, item) if zbit < int(s.bit) { position[zv] = nil } if DEBUG_SOMETHINGS { Println("revert", zv, zbit, counts, bp+b) Println(s.line) Println(used) Println(index) } } } counts[int(bit)]-- s.found[v] = true s.line[bp] = v for b := 0; b < int(bit); b++ { used[bp+b] = 1 } if DEBUG_SOMETHINGS { Println("found", v, bit, counts, bp) Println(s.line) Println(used) Println(index) } } } return nil } type Item struct { value any random int priority int } type PriorityQueue []*Item func (pq PriorityQueue) Len() int { return len(pq) } func (pq PriorityQueue) Less(i, j int) bool { return pq[i].priority < pq[j].priority } func (pq PriorityQueue) Swap(i, j int) { pq[i], pq[j] = pq[j], pq[i] } func (pq *PriorityQueue) Push(x any) { item := x.(*Item) *pq = append(*pq, item) } func (pq *PriorityQueue) Pop() any { old := *pq n := len(old) item := old[n-1] old[n-1] = nil *pq = old[:n-1] return item } func (s *Solver) analyze() { }