P3413 SAC#1 - 萌數

是的,我又去偷題面了

1522713-20191028212255636-978293732.png

解析在代碼裏

#include <bits/stdc++.h>
#define int long long 
using namespace std ;
const int maxn = 1e3+105 ;
const int ha = 1e9+7 ;
int f[maxn][12][12] , ans ;
// f[i][j][k] -> 第i位上的數字是j,i-1的數字是k的不是萌數的個數是多少 
string l , r ;
int calc (string s) {
    int n = s.size() , X = -1 , Y = -1 ;// X , Y 分別記錄i位相鄰倆個數位 
    int tot = 0 , ans = 0 , flag = 0 ;
    for(int i = 0 ; i < n ; i ++) {//把字符串轉成數字,(因爲用到了補集的思想) 
        tot = (tot * 10 + s[i]-'0'+ha) % ha ;
    }
    for(int i = 1 ; i < n ; i ++) { // 處理len位之前的,保證都是合法的 
        for(int j = 1 ; j <= 9 ; j ++) {//枚舉當前位置的數字 
            for(int k = 0 ; k <= 9 ; k ++) {//枚舉下一位的數字 
                ans = (ans+f[i][j][k]) % ha ;
            }
        }
    }
    for(int i = n ; i > 1 ; i --) {//枚舉位數 
        int  v = s[n-i] - '0' ;//倒着來(從低位到高位);
        for(int j = 0 ; j < v ; j ++) {//枚舉當前位置上的每一個數字 
            if(i != n || j != 0) {//如果首位不是0(首位不能是0哎) 
                for(int k = 0 ; k <= 9 ; k ++) {//枚舉i-1位 
                    if(X != j && Y != j && j != k && k != X) {//如果不是萌數 
                        ans = (ans+f[i][j][k]) % ha ;//就加到補集裏 
                    } 
                }
            }
        } 
        if(v == X || v == Y) {
            flag = 1 ;
            break ;
        }Y = X ; X = v ;
    } 
    if(n > 1) ans += 10 ;//加上個位數 
    if(!flag) {
        for(int j = 0 ; j <= s[n-1] - '0' ; j ++) {
            if(j != Y && j != X) {
                ans = (ans+1)%ha ;
            }
        }
    }
    return (tot-ans+1+ha)%ha ;//求補集 
}
signed main () {
    cin >> l >> r ;
    for(int i = 2 ; i <= 1000 ; i ++) { //枚舉位數
        for(int j = 0 ; j <= 9 ; j ++) {//枚舉i位上的數字 
            for(int k = 0 ; k <= 9 ; k ++) {//枚舉i+1位上的數字 
                if(j != k) //如果這兩位上的數字不同 
                {
                    for(int o = 0 ; o <= 9 ; o ++) {//枚舉第i-1位的數字 
                        if(j!=o&&k!=o) {// 如果再不同的話,就不是萌數了 
                            f[i][j][k] = (f[i][j][k]+f[i-1][k][o])%ha ;
                        }
                    } 
                    if(i-1 == 1) f[i][j][k] = (f[i][j][k]+1)%ha ; //如果上一位是一位數,ans++(一位數也是非萌數) 
                } 
            }   
        } 
    }
//  int m = s.size() ;
    ans = (calc(r)-calc(l)+ha)%ha ;//[l,r] 可以差分成 [0,r] - [0,l-1] ; 
    for(int i = 1 ; i < l.size() ; i ++ ) {//但是由於l是個該死的字符串 
        if(l[i] == l[i-1] || (i > 1 && l[i] == l[i-2])) {//暴力判斷第l個數是不是 
            ans = (ans+1) % ha ;
            break ;
        }
    }
    printf("%lld\n",ans) ;
    return 0 ;
} 

溜了溜了!

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章