fzu 2216 The Longest Straight 二分

題目鏈接

http://acm.fzu.edu.cn/problem.php?pid=2216

題意

給出n張牌，每張牌有個數值，n張牌中最大的值爲m。值爲0的牌可以代替任意值，問可以湊出最長順子的長度是多少。

思路

用dp[i] 表示從1到j需要填0的個數
dp[j] - d[i] 表示 i+1 到 j需要填0的個數
然後mlogm 枚舉

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>
#include<string>
#include<queue>
#include<stack>
#include<set>
#include<map>
#define ll long long
using namespace std;
const int INF = ( 2e9 ) + 2;
const ll maxn = 1e5+10;
int dp[maxn];
bool vis[maxn];
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        memset(vis,0,sizeof(vis));
        memset(dp,0,sizeof(dp));
        int n,m,temp,z=0;
        scanf("%d%d",&n,&m);

        for(int i=0;i<n;i++)
        {
            scanf("%d",&temp);
            if(temp)
            vis[temp]=1;
            else
            z++;
        }
        // dp[j]  表示從1到j需要填0的個數
        // dp[j] - d[i]  i+1 到 j需要填0的個數 
        for(int i=1;i<=m;i++)
        {
            if(vis[i])
            dp[i]=dp[i-1];
            else
            dp[i]=dp[i-1]+1;
        }
        int ans=-INF;
        for(int i=0;i<m;i++)
        {
            int l=i,r=m;
            while(l<=r)
            {
                int mid=(l+r)>>1;
                if(dp[mid]-dp[i]>z)
                r=mid-1;
                else
                l=mid+1;
            }
            ans=max(ans,r-i);
        }
        printf("%d\n",ans);

    }
}