AtCoder AGC035E Develop (DP、图论、计数)

题目链接

https://atcoder.jp/contests/agc035/tasks/agc035_e

题解

没想出来最后一步DP宛如智障……
考虑一个数\(x\notin S\)的条件是\(x\)被删除了且在\(x\)最后一次被删除之后不能再对\(x+2\)和\(x-K\)进行删除操作。也就是说\(x+2\)和\(x-K\)的最晚删除时间要比\(x\)晚。那么我们从\(x\)往\(x+2\)和\(x-K\)连边，形成的图如果有环那么这个方案就不合法，否则合法。
如果\(K\)是偶数，显然很好算；如果\(K\)是奇数，有环的充要条件是存在一个环从某点\(a\)出发先后经过\(1\)条\(+K\)边、若干条\(-2\)边、\(1\)条\(+K\)边、若干条\(-2\)边，且经过的\(-2\)边的总数是\(K\). 这也就意味着我们需要让选出的点中不存在这样的一个环。
如果我们把图换一种方式看，把图排成一个每行\(2\)列的形状，让左边的奇数\(x\)和右边的偶数\(x+K\)在一行，从上往下每一层的两个（或一个）数是它上面一层对应的数\(+2\), 那么刚才提到的环就相当于若干条向上边、\(1\)条向右边、若干条向上边，向上边的总数是\(K\)，最终从一个向左下的边下来。于是我们的目标就是要让“若干条向上边、\(1\)条向右边、若干条向上边”构成的最长的链长度不超过\((K+1)\).
这个可以从上往下DP: 设\(f[i][j][k]\)表示前\(i\)层，“从该层左边开始经过若干条向上边、\(1\)条向右边、若干条向上边的最长链”长度为\(j\)，“从该层右边开始经过若干条向上边的最长链”长度为\(k\). 转移讨论一下两边都不选、选左、选右、左右都选即可。
时间复杂度\(O(n^3)\).

代码

#include<bits/stdc++.h>
#define llong long long
#define pii pair<int,int>
#define riterator reverse_iterator
using namespace std;

inline int read()
{
    int x = 0,f = 1; char ch = getchar();
    for(;!isdigit(ch);ch=getchar()) {if(ch=='-') f = -1;}
    for(; isdigit(ch);ch=getchar()) {x = x*10+ch-48;}
    return x*f;
}

const int N = 150;
llong P;
int n,m;

void updsum(llong &x,llong y) {x = x+y>=P?x+y-P:x+y;}

namespace Solve1
{
    llong f[N+3];
    void solve()
    {
        m>>=1; f[0] = 1ll;
        for(int i=1; i<=n; i++)
        {
            for(int j=max(0,i-m-1); j<i; j++)
            {
                updsum(f[i],f[j]);
            }
        }
        llong ans1 = 0ll,ans2 = 0ll;
        for(int i=max(0,(n>>1)-m); i<=(n>>1); i++) updsum(ans1,f[i]);
        for(int i=max(0,(n+1>>1)-m); i<=(n+1>>1); i++) updsum(ans2,f[i]);
//      printf("ans1=%lld ans2=%lld\n",ans1,ans2);
        printf("%lld\n",ans1*ans2%P);
    }
}

namespace Solve2
{
    llong f[N+3][N+3][N+3];
    void solve()
    {
        f[0][0][0] = 1ll;
        for(int i=1; i+i-m<=n; i++)
        {
            for(int j=0; j<=m+1; j++) for(int k=0; k<=n; k++)
            {
                updsum(f[i][0][0],f[i-1][j][k]);
            }
            if(i+i<=n)
            {
                for(int j=0; j<=m+1; j++) for(int k=0; k<=n; k++)
                {
                    updsum(f[i][0][k+1],f[i-1][j][k]);
                }
            }
            if(i+i-m>=1)
            {
                for(int j=0; j<=m; j++) for(int k=0; k<=n; k++)
                {
                    updsum(f[i][j+(j>0)][0],f[i-1][j][k]);
                }
            }
            if(i+i<=n&&i+i-m>=1)
            {
                for(int j=0; j<=m; j++) for(int k=0; k<=n; k++)
                {
                    int jj = max(j+1,k+2);
                    if(jj<=m+1)
                    {
                        updsum(f[i][jj][k+1],f[i-1][j][k]);
                    }
                }
            }
        }
        llong ans = 0ll;
        for(int j=0; j<=m+1; j++) for(int k=0; k<=n; k++)
        {
            updsum(ans,f[(n+m)>>1][j][k]);
        }
        printf("%lld\n",ans);
    }
}

int main()
{
    scanf("%d%d%lld",&n,&m,&P);
    if(!(m&1)) {Solve1::solve();}
    else {Solve2::solve();}
    return 0;
}