CodeChef - MEXDIV Mex division

CodeChef - MEXDIV Mex division

思维，递推计数

题意

给你一个序列，让你对他进行划分。使划分后的每个集合的mex值都不大于k。
mex函数是定义在自然数集合上的，表示不再这个集合里面的最小自然数。

思路

不会递推计数啊。。。

f[i]表示前i个数的划分方案。现在我已经算好前i-1，对于i，把它加进去后，如果不影响，那么方案数乘2，表示i-1和i之间可以加分割也可以不加。如果影响，那一定是因为0到k的数都出现过了。我们用一个map记录0到k每个数字出现的次数，并用一个指针维护当前考虑的集合的左端点，cnt维护0到k出现的数字个数，sum维护当前方案数。如果加入a[i]导致cnt超了，那么右移左指针。并减去f[l-1]，因为去掉l意味着前l-1个数构成的划分方案不能考虑了。

代码

#include<cstdio>
#include<cstring>
#include<queue>
#include<iostream>
#include<algorithm>
#include<set>
#include<map>
#define M(a,b) memset(a,b,sizeof(a))
using namespace std;
const int MAXN=500007;
const int oo=0x3f3f3f3f;
typedef long long LL;
const LL loo=4223372036854775807ll;
typedef long double LB;
const LL mod=1e9+7;

int num[MAXN];
LL f[MAXN];
map<int, int> mp;
int main()
{
    int n, k;
    while(scanf("%d%d", &n, &k)==2)
    {
        M(f, 0);mp.clear();
        for(int i=1;i<=n;i++)
            scanf("%d", &num[i]);

        f[0]=1;LL cnt=0, sum=1;

        int l=1;
        for(int i=1;i<=n;i++)
        {
            mp[num[i]]++;if(num[i]<=k&&mp[num[i]]==1) cnt++;
            while(cnt>k)
            {
                mp[num[l]]--;
                if(num[l]<=k&&mp[num[l]]==0) cnt--;
                sum=(sum-f[l-1]+mod)%mod;
                l++;
            }
            f[i]=sum;
            sum+=f[i];sum%=mod;//sum*=2;
        }
        printf("%lld\n", f[n]);
    }
    return 0;
}