leetcode 第 59 场双周赛 5837. 划分数字的方案数 题解

前置知识：后缀数组，RMQ区间最值，简单dp。

首先我们会想到使用dp的思路来解决，dp[i][j] 表明当前第i个位置为结尾往前j个位置截取出来当作一个数的方案数，dp[i][j] 就是对于 i-j+1 位置为结尾上小于当前截取的值的方案数的和，因为每次要选取不同的j，这样就是n的平方的时间复杂度，还要每次比对数的大小是n的时间复杂度，最终的时间复杂度就是n的三次方。

仔细思考一下，发现只能简化数字的对比，字符串排序就能得到大小，只要求出 rank[i-j+1] +1 和 rank[i-j-j+1] 之间height 数组的最小值，然后就是分情况讨论height数组大小和rank 大小的情况。 区间最值使用RMQ就可以了。最终时间查询时间复杂度会降到log(n)。

代码：

class Solution {
public:
    const int N = 10000;
    int mod = 1e9+7;
    char s[10000]; int rank[10000],y[10000],tmp[10000]; 
    int c[10000],SA[10000],n,m,Height[10000];
    void Sort(){
        for(int i=1;i<=m;i++) c[i] = 0;
        for(int i=1;i<=n;i++) c[rank[i]]++;
        for(int i=2;i<=m;i++) c[i] += c[i-1];
        for(int i=n;i>=1;i--) SA[c[rank[y[i]]]--] = y[i];
    }
    void get_SA(){
        for(int i=1;i<=n;i++) rank[i] = s[i] , y[i] = i;
        Sort();
        for(int k=1;k<=n;k<<=1){
            int cnt = 0;
            for(int i=n-k+1;i<=n;i++) y[++cnt] = i;
            for(int i=1;i<=n;i++) if(SA[i]>k) y[++cnt] = SA[i] - k;
            Sort(); swap(rank,tmp); int num = 1; rank[SA[1]] = 1;
            for(int i=2;i<=n;i++){
                if(tmp[SA[i]] == tmp[SA[i-1]] && tmp[SA[i]+k] == tmp[SA[i-1]+k])
                    rank[SA[i]] = num;
                else rank[SA[i]] = ++num;
            } m = num;
        }
    }
    void get_Hi(){
        int k = 0;
        for(int i=1;i<=n;i++){
            if(rank[i]==1) continue;
            int j = SA[rank[i]-1]; if(k) k--;
            while(i+k<=n && j+k<=n && s[i+k]==s[j+k]) k++; 
            Height[rank[i]] = k;
        }
    }



    
    int minn[3505][100];

    void getbestarr()//n为给定的数组的长度  
    {  
         int tem = (int)floor(log2((double)n));//因为区间的最长长度是2^tem==n嘛  
       for(int i=1;i<=n;i++)  
            minn[i][0]  = Height[i];  
        for(int j=1;j<=tem;j++) //下标从1开始  
         for(int i=1;i+(1<<j)-1<=n;i++)  
        {  
             minn[i][j] = min(minn[i][j-1],minn[i+(1<<(j-1))][j-1]);  //最小值  
        }  
    }  

    
    int query(int a,int b)//getwhat表示你是想取最大还是最小  
    {  
        int k = log2(b-a+1);  
      
        return min(minn[a][k],minn[b-(1<<k)+1][k]);  
    }  

    int dp[3505][3505];
    
    int numberOfCombinations(string num) {
        for(int i=0;i<num.size();i++){
            s[i + 1] = num[i];
        }
        n = num.size();
        m = 127; get_SA();  get_Hi();
        getbestarr();
        //for(int i=1;i<=n;i++) cout<<rank[i]<<endl;
        
        memset(dp,0,sizeof(dp)); dp[0][0] = 0;
        for(int i=1;i<=n;i++) dp[0][i] = 1;
        
        for(int i=1;i<=n;i++){
            for(int j=1;j<=i;j++){
                if(s[i-j+1] == '0') {
                    dp[i][j]+=dp[i][j-1];
                    dp[i][j] %=mod;
                    continue;
                }
                if(j > i-j ) {
                    dp[i][j] += dp[i-j][j];
                    dp[i][j]+=dp[i][j-1];
                    dp[i][j] %=mod;
                    continue;
                } 
                int rk1 = rank[i-j+1];
                int rk2 = rank[i-j-j+1];
                
                int re = query(min(rk1,rk2) + 1,max(rk1,rk2));
               // cout<<rk1<<' '<<rk2<<' '<<re<<endl; 
                if(re >= j){
                    dp[i][j] += dp[i-j][j];
                }else {
                    if(rk1 > rk2) dp[i][j] += dp[i-j][j];
                    else dp[i][j] += dp[i-j][j-1];
                }
                dp[i][j]+=dp[i][j-1];
                dp[i][j] %=mod;
                
            }
            
            for(int j=i+1;j<=n;j++) dp[i][j] += dp[i][j-1],dp[i][j] %= mod;
        }
        
        return dp[n][n];
    }
};