Gym - 102040B Counting Inversion (数位dp)

题意：求[a,b]区间内的数字中正序对的个数。

具体思路参考：

https://blog.csdn.net/weixin_43135318/article/details/88061396

https://www.cnblogs.com/asdfsag/p/11278519.html

在此基础上维护一下每个状态中大于每个数字的数字出现的次数即可。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll N=20;
ll bit[N],nbit,f[N][2][2],g[N][2][2][N],G[N][2][2],vis[N][2][2],ka,a,b,ka2;
void dfs(ll u,ll ze,ll lim) {
    if(vis[u][ze][lim]==ka)return;
    vis[u][ze][lim]=ka;
    G[u][ze][lim]=0;
    for(ll i=0; i<=9; ++i)g[u][ze][lim][i]=0;
    if(u==0) {f[u][ze][lim]=1; return;}
    f[u][ze][lim]=0;
    for(ll i=0; i<=(lim?bit[u]:9); ++i) {
        ll v=u-1,ze2=ze&&i==0,lim2=lim&&i==bit[u];
        dfs(v,ze2,lim2);
        f[u][ze][lim]+=f[v][ze2][lim2];
        for(ll j=0; j<=9; ++j)g[u][ze][lim][j]+=g[v][ze2][lim2][j];
        G[u][ze][lim]+=G[v][ze2][lim2];
        if(!ze2) {
            for(ll j=0; j<=i; ++j)g[u][ze][lim][j]+=f[v][ze2][lim2];
            G[u][ze][lim]+=g[v][ze2][lim2][i+1];
        }
    }
}
ll solve(ll x) {
    ++ka;
    for(nbit=0; x; x/=10)bit[++nbit]=x%10;
    dfs(nbit,1,1);
    return G[nbit][1][1];
}
int main() {
    ll T;
    for(scanf("%lld",&T); T--;) {
        scanf("%lld%lld",&a,&b);
        printf("Case %lld: %lld\n",++ka2,solve(b)-solve(a-1));
    }
}