Codeforces Round #384 (Div. 2) 734E(二分答案+状态压缩DP)

题目大意

给定一个序列an，序列中只有1~8的8个整数，让你选出一个子序列，满足下列两个要求

1.不同整数出现的次数相差小于等于1

2.子序列中整数分布是连续的，即子序列的整数必须是1,1,1....1，2,2,2.....2,2.......连续分布

做法：

那么我们可以二分不同整数出现的次数，假如说二分出现次数是L，那么可以证明有a个整数出现次数是L，有(8-a)个整数出现次数是L-1（同时不难证明L+1比L更优)

这样每个整数只有两种状态，要么选L个，要么选L-1个。

压缩决策s，s的第i位表示是否对整数i做出了决策

再用容器把每一个不同整数出现的序列记录下来，这样就可以进行dp了

设dp[i][s]表示从原序列扫到前i个数，已经做出了s的决策，出现L次的不同整数个数最大是多少

枚举对第k个整数做决策，dp[i][s]就可以转移到另外两个状态

dp[next(k, L-1)][s|(1<<k)] = max(itself, dp[i][s]);

dp[next(k, L)][s|(1<<k)] = max(itself, dp[i][s]+1);

其中next(k, L)是指从i开始，往后的第L个k的整数位置

然后取dp[1~n][(1<<8)-1]中的最大值，返回答案即可

另外如果最大值小于等于0（注意等于！！），说明不存在解，返回0即可。

需要注意的是，当l = 2都不可行的时候需要特判处理。

#include <iostream>
#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;
const int maxn = ;
const int inf = 1e9;
int dp[maxn][<<], a[maxn], cur[maxn], n;
vector <int> in[];
int can(int L)
{
    for(int i = ; i <= ; i++) cur[i] = ;
    memset(dp, , sizeof(dp));
    dp[][] = ;
    for(int i = ; i <= n; i++)
    {
        for(int s = ; s < (<<); s++)
        {
            if(dp[i][s] < ) continue;
            for(int k = ; k <= ; k++)
            {
                if(s & (<<(k-))) continue;
                if(in[k].size() - cur[k] < L-) continue;
                dp[in[k][cur[k] + L - ]][s|(<<(k-))] = max(dp[in[k][cur[k] + L - ]][s|(<<(k-))], dp[i][s]);
                if(in[k].size() - cur[k] < L)  continue;
                dp[in[k][cur[k] + L - ]][s|(<<(k-))] = max(dp[in[k][cur[k] + L - ]][s|(<<(k-))], dp[i][s]+);
            }
        }
        cur[a[i]]++;
    }
    int ans = -inf;
    for(int i = ; i <= n ; i++) ans = max(ans, dp[i][(<<) - ]);
    if(ans <= ) return ;
    return ans * L + ( - ans)*(L-);
}

int main()
{
    //freopen("a.txt", "r", stdin);
    cin>>n;
    for(int i = ; i <= n; i++) cin>>a[i];
    for(int i = ; i <= n; i++) in[a[i]].push_back(i);
    int l = , r = n, ans = -inf;
    while(l < r)
    {
        int mid = (l+r)>>;
        if(can(mid)) l = mid+;
        else r = mid;
    }
    if(can() == )
    {
        ans = ;
        for(int i = ; i <= ; i++) if(in[i].size() > ) ans++;
        cout<<ans<<endl;
    }
    else
    {
        ans = max(can(l), can(l-));
        cout<<ans<<endl;
    }
}