模板题——数位DP、状态压缩、记忆化搜索

DP啊！难的真难，简单的也真简单，正让人哭笑不得，喜忧参半 😉

1.整数的划分

完全背包解法
状态表示：f[i][j]表示只从1~i中选，且总和等于j的方案数
状态转移方程:f[i][j] = f[i - 1][j] + f[i][j - i]

#include <bits/stdc++.h>

using namespace std;
const int N=1010,mod=1e9+7;
int n;
int f[N];
int main()
{
    cin>>n;
    f[0]=1;
    for(int i=1;i<=n;i++)
        for(int j=i;j<=n;j++)
        f[j]=(f[j]+f[j-i])%mod;
    cout<<f[n]<<endl;
    return 0;
}

另解：
状态表示：f[i][j]表示总和为i，总个数为j的方案数
状态转移方程：f[i][j] = f[i - 1][j - 1] + f[i - j][j];

#include <bits/stdc++.h>

using namespace std;
const int N=1010,mod=1e9+7;
int n;
int f[N][N];
int main()
{
    cin>>n;
    f[1][1]=1;
    for(int i=2;i<=n;i++)
        for(int j=1;j<=i;j++)
        f[i][j]=(f[i-1][j-1]+f[i-j][j])%mod;
    int res=0;
    for(int i=1;i<=n;i++)
        res=(res+f[n][i])%mod;
    cout<<res<<endl;
    return 0;
}

2.计数问题：分情况讨论【0的边界很模糊】

#include <bits/stdc++.h>

using namespace std;
const int N=10;
/*
0的第一种情况：001~abc-1, 999
其余数字：000~abc-1[1也特判]
以某数字出现在第4位为例
abc
    1. num[i] < x, 0
    2. num[i] == x, 0~efg——>efg+1
    3. num[i] > x, 0~999 ——>1000

*/
int get(vector<int> num, int l,int r)//把按位数存储的数字转化成一个整数
{
    int res=0;
    for(int i=l;i>=r;i--)
        res=res*10+num[i];
    return res;
}
int power10(int x)//10^x
{
    int res=1;
    while(x--) res*=10;
    return res;
}
int countt(int n,int x)//1~n中x出现的次数
{
    if(!n) return 0;
    vector<int > num;
    while(n)//获取n的每一位
    {
        num.push_back(n%10);
        n/=10;
    }
    n=num.size();
    int res=0;
    for(int i=n-1-!x;i>=0;i--)//枚举数字i可以出现在哪一位上，0时要特判
    {
        if(i<n-1)
        {
            res+=get(num,n-1,i+1)*power10(i);
            if(!x) res-=power10(i);
        }
        if(num[i]==x) res+=get(num,i-1,0)+1;
        else if(num[i]>x) res+=power10(i);
    }
    return res;
}
int main()
{
    int a,b;
    while(cin>>a>>b,a)
    {
        if(a>b) swap(a,b);
        for(int i=0;i<=9;i++)
            cout<<countt(b,i)-countt(a-1,i)<<' ';
        cout<<endl;
    }
    return 0;
}

3.状态压缩：蒙德里安的梦想【卡了2小时，已经不想说话了/(ㄒoㄒ)/~~】

#include <bits/stdc++.h>

using namespace std;
const int N=12,M=1<<N;//2^n种状态
int n,m;
long long f[N][M];
bool st[M];
int main()
{
    while(cin>>n>>m,n||m)
    {
        for(int i=0;i<1<<n;i++)//当前行
        {
            int cnt=0;//上一行0的个数
            st[i]=true;//当前这一行0的个数是否合法——>是否为偶数
            for(int j=0;j<n;j++)//枚举这一行的每一列
                if(i>>j&1)//定位到当前这一方格如果为1
                {
                    if(cnt&1) st[i]=false;//判断一下前面方格0的个数是否为偶数
                    cnt=0;
                }
                else cnt++;
            if(cnt&1) st[i]=false;
        }
        memset(f,0,sizeof f);
        f[0][0]=1;
        for(int i=1;i<=m;i++)//枚举每一列
            for(int j=0;j<1<<n;j++)//枚举第i行的状态
                for(int k=0;k<1<<n;k++)//枚举第i-1行的状态
                    if((j&k)==0&&st[j|k])
                    f[i][j]+=f[i-1][k];
        cout<<f[m][0]<<endl;
    }
    return 0;
}

4.状态压缩：哈密顿图

#include <bits/stdc++.h>

using namespace std;
const int N=20,M=1<<N;
int n;
int w[N][N];
int f[M][N];
int main()
{
    cin>>n;
    for(int i=0;i<n;i++)
        for(int j=0;j<n;j++)
            cin>>w[i][j];
    memset(f,0x3f,sizeof f);
    f[1][0]=0;
    for(int i=0;i<1<<n;i++)
        for(int j=0;j<n;j++)
            if(i>>j&1)
                for(int k=0;k<n;k++)
                    if(i-(1<<j)>>k&1)
                        f[i][j]=min(f[i][j],f[i-(1<<j)][k]+w[k][j]);
    cout<<f[(1<<n)-1][n-1]<<endl;
    return 0;
}

5.树状DP：没有上司的舞会

#include <bits/stdc++.h>

using namespace std;
const int N=6010;
int n;
int h[N],e[N],ne[N],idx;
int w[N],f[N][2];
bool st[N];//判断是否有根节点
void add(int a,int b)
{
    e[idx]=b;ne[idx]=h[a];h[a]=idx++;
}
void dfs(int u)
{
    f[u][0]=0;
    f[u][1]=w[u];
    for(int i=h[u];i!=-1;i=ne[i])
    {
        int j=e[i];
        dfs(j);
        f[u][0]+=max(f[j][0],f[j][1]);
        f[u][1]+=f[j][0];
    }
}
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++) scanf("%d",&w[i]);
    memset(h,-1,sizeof h);
    for(int i=0;i<n-1;i++)
    {
        int a,b;
        scanf("%d%d",&a,&b);
        add(b,a);
        st[a]=true;
    }
    int root=1;//按照序号寻找根节点
    while(st[root]) root++;
    dfs(root);
    cout<<max(f[root][0],f[root][1])<<endl;
    return 0;
}

5.记忆化搜索：滑雪

#include <bits/stdc++.h>

using namespace std;
const int N=310;
int n,m;
int g[N][N];
int f[N][N];
int dx[4]={-1,0,1,0},dy[4]={0,1,0,-1};
int dp(int x,int y)
{
    int &v=f[x][y];//用v代替f[x][y]
    if(v!=-1) return v;//如果v被计算过了
    v=1;
    for(int i=0;i<4;i++)
    {
        int a=x+dx[i],b=y+dy[i];
        if(a>=1&&a<=n&&b>=1&&b<=m&&g[x][y]>g[a][b])
            v=max(v,dp(a,b)+1);
    }
    return v;
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
        scanf("%d",&g[i][j]);
    memset(f,-1,sizeof f);
    int res=0;
    for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
        res=max(res,dp(i,j));
    printf("%d\n",res);
    return 0;
}