package main import ( "bufio" "fmt" "os" ) func main() { yuki2276() } func demo() { fmt.Println(FibonacciSearch(func(x int) int { return x * x }, 0, 100, true)) fmt.Println(FibonacciSearch(func(x int) int { return x * x }, 0, 100, false)) } func yuki2276() { // 替换问号序列后得到AC子序列的最大方案数 // https://yukicoder.me/problems/no/2276 // // A, C, ? からなる長さ N の文字列 S が与えられます。 // あなたは、S に対して次の操作を行います。 // !S に含まれる ? を、それぞれ A または C のいずれかで書き換える。 // !操作後、S から(連続するとは限らない)部分列として AC を取り出す方法の数が得点となります。 // ここで、部分列を取り出す方法が異なるとは、取り出す位置が異なることをいいます。 // 得点の最大値を求めてください。 in := bufio.NewReader(os.Stdin) out := bufio.NewWriter(os.Stdout) defer out.Flush() var n int fmt.Fscan(in, &n) var s string fmt.Fscan(in, &s) sb := []byte(s) replacePos := make([]int, 0) for i := 0; i < n; i++ { if s[i] == '?' { replacePos = append(replacePos, i) } } // 将前n个?替换为A,后n个?替换为C,求AC的个数 fun := func(n int) int { if !(0 <= n && n <= len(replacePos)) { panic("n out of range") } for i := 0; i < n; i++ { sb[replacePos[i]] = 'A' } for i := n; i < len(replacePos); i++ { sb[replacePos[i]] = 'C' } countA := 0 countAC := 0 for _, v := range sb { if v == 'A' { countA++ } if v == 'C' { countAC += countA } } return countAC } _, y := FibonacciSearch(fun, 0, len(replacePos)+1, false) fmt.Fprintln(out, y) } const INF int = 1e18 // 寻找[start,end)中的一个极值点,不要求单峰性质. // // 返回值: (极值点,极值) func FibonacciSearch(f func(x int) int, start, end int, minimize bool) (int, int) { end-- a, b, c, d := start, start+1, start+2, start+3 step := 0 for d < end { b = c c = d d = b + c - a step++ } get := func(i int) int { if end < i { return INF } if minimize { return f(i) } return -f(i) } ya, yb, yc, yd := get(a), get(b), get(c), get(d) for i := 0; i < step; i++ { if yb < yc { d = c c = b b = a + d - c yd = yc yc = yb yb = get(b) } else { a = b b = c c = a + d - b ya = yb yb = yc yc = get(c) } } x := a y := ya if yb < y { x = b y = yb } if yc < y { x = c y = yc } if yd < y { x = d y = yd } if minimize { return x, y } return x, -y } func min(a, b int) int { if a < b { return a } return b } func max(a, b int) int { if a > b { return a } return b }