codeforces 895D

(組合數學)
題意:給定兩個由全英文字母構成的字符串s1 ,s2 ,保證字典序s1<s2 ,求有多少個s3 ,使得s1<s3<s2 並且s3s1 的字母所組成的一個排列。

思路:
  首先統計s1 串有哪些字母,每個字母出現了多少次,並用一個num數組保存下來。這樣我們起碼知道s3由哪些字母構成。然後想到可以先算字典序比s1 小的字符串的數目,再算出字典序比s2 小的字符串的數目,那麼這兩個結果一減,就是答案了。
  第一種數目比較好求,直接求出s1 是它本身排列的第幾位即可。而s3 由於不是由s1 轉換而成的就有點不好辦,於是想到可以將s3 轉換成s1 字母構成的某種形式,再利用相同的方法求。
  在求本身排列是第幾位的過程中,用i 遍歷0len ,依次求出:從當前位比較結果來看,有多少串字典序小於s1 。然後對於每個i ,求出當前位的結果(這裏用到了排列組合知識+一個小優化)。最後相加得到答案。

(公式懶得寫了..這個編輯器有點不太會用。話說,還是要多敲題,這題思路雖然出來了,敲得賊慢…)

代碼:

#include <cstdio>
#include <cstring>
#include <algorithm>
#define LL long long

using namespace std;
const int maxn = 1000010;
const LL mod = 1e9 + 7;

bool s2_smaller, s2_bigger;
int num[30], sum[30];
char s1[maxn], s2[maxn];
LL fac[maxn], inv_fac[maxn];

LL pow(LL a, LL b) {
    LL ret = 1;
    while(b) {
        if(b&1)
            ret = ret * a % mod;
        a = a * a % mod;
        b >>= 1;
    }
    return ret;
}

void pre_treat(int len) {
    fac[0] = inv_fac[0] = 1;
    for(int i=1; i<=len; i++) {
        fac[i] = (LL)i * fac[i-1] % mod;
        inv_fac[i] = pow((LL)fac[i], mod-2);
    }
}

void init(int len) {
    for(int i=0; i<len; i++)
        num[s1[i]-'a'] ++;
    sum[0] = num[0];
    for(int i=1; i<26; i++)
        sum[i] = sum[i-1] + num[i];
}

void transform_s2(int len) {
    init(len);
    for(int i=0; i<len; i++) {
        int alpha = s2[i] - 'a';
        if(!s2_smaller && !s2_bigger) {
            if(num[alpha])
                num[alpha] --;
            else {
                for(int j=alpha-1; j>=0; j--) if(num[j]) {
                    s2_smaller = true; //puts("small");
                    s2[i] = j + 'a'; num[j] --;
                    break ;
                }
                if(s2_smaller) continue ;
                for(int j=alpha+1; j<26; j++) if(num[j]) {
                    s2_bigger = true; //puts("big");
                    s2[i] = j + 'a'; num[j] --;
                    break ;
                }
                if(s2_bigger) continue ;
            }
        }
        else if(s2_smaller) {
            for(int j=25; j>=0; j--) if(num[j]) {
                s2[i] = j + 'a'; num[j] --;
                break ;
            }
        }
        else {//s2_bigger
            for(int j=0; j<26; j++) if(num[j]) {
                s2[i] = j + 'a'; num[j] --;
                break ;
            }
        }
    }
}

LL solve(char *s, int len) {
    init(len);
    LL ret = 0;
    for(int i=0; i<len; i++) {
        int alpha = s[i] - 'a';
        if(alpha > 0 && sum[alpha-1] > 0) {
            LL t1 = fac[len-i-1], t2 = 0;
            for(int j=0; j<26; j++) if(num[j]) {
                t1 = t1 * inv_fac[num[j]] % mod;
                if(j < alpha)
                    t2 = (t2 + num[j]) % mod;
            }
            ret = (ret + t1 * t2 % mod) % mod;
        }
        //printf("------i:%d ret:%I64d\n",i,ret);
        num[alpha] --;
        for(int j=alpha; j<26; j++)
            sum[j] --;
    }
    return ret;
}

int main() {
    //freopen("test.txt","r",stdin);
    s2_smaller = s2_bigger = false;
    scanf("%s%s",s1,s2);
    int len = strlen(s1);
    pre_treat(len);
    transform_s2(len);
    LL l = solve(s1, len), r = solve(s2, len);
    //printf("l:%I64d r:%I64d\n",l,r);
    LL ans = (r - l + mod) % mod;
    if(!s2_smaller && !s2_bigger)
        ans = (ans - 1 + mod) % mod;
    else if(s2_smaller)
        ans = ans;
    else
        ans = (ans - 1 + mod) % mod;
    printf("%I64d\n",ans);
    return 0;
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章