結果

問題 No.3299 K-th MMA String
コンテスト
ユーザー ID 21712
提出日時 2025-10-19 17:47:49
言語 Go
(1.23.4)
結果
AC  
実行時間 44 ms / 2,000 ms
コード長 1,573 bytes
コンパイル時間 17,206 ms
コンパイル使用メモリ 246,544 KB
実行使用メモリ 10,496 KB
最終ジャッジ日時 2025-10-19 17:48:09
合計ジャッジ時間 15,721 ms
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 20
権限があれば一括ダウンロードができます

ソースコード

diff #

package main

import . "fmt"

// 解説読んだけど分からん…
// 入力条件の上限と問題難易度のメタ読みから全列挙で間に合う可能性を考える方向性?
// 真面目な考察だと
// 先頭A,MA,MMで場合分け数え上げDPでK以上になったところからの復元で間に合うみたいな感じ?
// 合ってるか分からん…
// ↑ 数え上げちゃうと64bit整数に収まらないので無理
// MとAで構成される任意の文字列X
// 文字列Xの適当な位置にMMAを挿入することを考える
// 長さLの文字列XにおけるMとAの組み合わせ総数は(2^L)通り

// そのときMMAを挿入できる位置は(L+1)通り
// 挿入後の文字列総数(2^L)*(L+1)がKの上限値以上になるLを考える
// Kの上限値は10^5なので雑に感覚的にはL=14くらい?全列挙可能ぽい?
// N>Kのときには解説どおり頭にAを並べればよくて
// N<KのときはK番目の文字列の頭(N-K)文字分を削除すればよい

import . "sort"
import . "strings"

func main() {
	var n,k int
	Scan(&n,&k)
	// Aを0、Mを1として二進表現する
	// MMAは110(2)なので6(10)
	h:=[]int{}
	dup:=map[int]bool{}
	for i:=0;i<(1<<14);i++ {
		for j:=0;j<15;j++ {
			m:=(1<<j)-1
			u:=(i>>j)<<(j+3)
			p:=6<<j
			x:=u|p|(i&m)
			if dup[x] {
				continue
			}
			dup[x]=true
			h=append(h,x)
		}
	}
	Ints(h)
	a:=h[k-1]
	s:=""
	for a>0 {
		if (a&1)==0 {
			s="A"+s
		} else {
			s="M"+s
		}
		a=a>>1
	}
	if len(s)<n {
		s=Repeat("A",n)+s
	}
	Println(s[len(s)-n:])
}
0