NOIP 2018 Simulation Day 3 v题解

题目描述

有$n$个球排成一行，每个球的颜色为黑或白。执行$k$次操作，第$i$次操作形式如下：

1.从$[1,n-i+1]$中，等概率随机选择一个整数$x$

2.移除从左到右的第$x$个球，或从右到左的第$x$个球（也就是从左到右的第$n-i+2-x$个球）。之后，所有右侧球的编号减1.给定每个球的颜色信息，希望最大化移除白球的数量。输出最优策略下，期望移除白球的数量。

数据范围

保证$1\le n\le 30,0\le k\le n$

题解

可以看到$n$的值较小，可以想到状压DP，要倒着做，所以要用记搜。

设$f_{i,j}$表示长度为$i$的状态为$j$的期望值，$j$记录的是每个球是否未被移除，则$f_{i,j}=\sum max(f_{i-1,k_1},f_{i-1,k_2})$

该做法时间复杂度为$O(2^n\ast n)$，会TLE。所以我们考虑进一步优化。

首先，如果设状态为每个球是否未被移除，则会有很多重复的状态。有时候剩下的球颜色序列相同，但状态不同。所以我们可以将状态记成剩下的颜色序列。

还有一个细节。即使时限为3s，但是如果所有状态都用map存，因为map存太多常数会很大，所以最后几个点还是会TLE。因此我们考虑把状态储存小的用数组存，状态储存大的用map，可以提高效率。具体做法见代码。

即使这样，最后一个点仍有可能TLE。我们考虑将状态数组再做一些修改。前文我们设$f_{i,j}$表示长度为$i$的状态为$j$的期望值，第一维数组是否可以省去？

每一个长度为$i$的状态，它的大小都小于$2^i$，所以我们可以试着省去第一维。但如果有状态重复的该如何处理呢？我们可以给每一个长度为$i$的状态的第$i+1$位打上$1$标记，表示这个状态长度为$i$，这样就不会有状态重复了，代码的效率也会有一定的提升。

经过以上的处理，我们的代码可以跑得很快，3s的时限对我们来说就绰绰有余啦~

代码：

#include<bits/stdc++.h>
using namespace std;
int n,k;
double dp[1<<25];
char s[35];
map<int,double>f;
double dfs(int len,int t,int now){
    if(t>k) return 0;
    if(len>24&&f.find(now)!=f.end()) return f[now];
    if(len<=24&&dp[now]!=-1) return dp[now];
    double sum=0;
    int e,vnow;
    for(int i=1;i<=len/2;i++){
        double c1=0,c2=0;
        vnow=((now>>i)<<i-1)|(now&(1<<i-1)-1);
        c1=dfs(len-1,t+1,vnow)+((now>>i-1)&1);
        vnow=((now>>len-i+1)<<len-i)|(now&(1<<len-i)-1);
        c2=dfs(len-1,t+1,vnow)+((now>>len-i)&1);
        sum+=max(c1,c2)*2.0;
    }
    if(len&1){
        int i=len+1>>1;
        vnow=((now>>i)<<i-1)|(now&(1<<i-1)-1);
        sum+=dfs(len-1,t+1,vnow)+((now>>i-1)&1);
    }
    sum=sum/len;
    if(len>24) f[now]=sum;
    else dp[now]=sum;
    return sum;
}
int main()
{
    scanf("%d%d",&n,&k);
    scanf("%s",s+1);
    for(int s=0;s<(1<<25);s++) dp[s]=-1.0;
    int e=0;
    for(int i=1;i<=n;i++)
    if(s[i]=='W') e|=(1<<i-1);
    e|=(1<<n);
    printf("%.8f",dfs(n,1,e));
}