Wannafly挑战赛6 C-逆序对【思维&组合数学】

题目描述
求所有长度为n的01串中满足如下条件的二元组个数：设第i位和第j位分别位ai和aj，则ai=1,aj=0{i < j}。答案对1e9+7取模。

输入描述: 输入一个n。
输出描述: 输出答案对1e9+7取模
示例1
输入
3
输出
6
说明

备注:
n <= 1018

题解：

分析：
对每一位的贡献进行考虑，考虑每一位为0时的贡献：假设第i位为0{i > 1}，则其前i-1位的总状态数为pow(2, i - 1)，所有状态下的1和0的最终出现次数相同，总状态下的01总次数为(i - 1)pow(2, i - 1)，所以1出现的状态总数为(i - 1)/2*pow(2, i - 1)，且第i位之后的状态为pow(2, n - i)，相乘为(i - 1) pow(2, n - 2)。
最后求

$$\sum_{i = 2}^{n}{(i - 1) * 2^{n - 2}}$$

#include <cstdio> 
typedef long long LL;
const LL mod = 1e9 + 7;

LL quickpow(LL a, LL b, LL c) {
    LL ans = 1;
    while(b) {
        if(b & 1) ans = ans * a % c;
        a = a * a % c;
        b >>= 1;
    }
    return ans;
}

int main() {
    LL n;
    scanf("%lld", &n);
    if(n <= 1) printf("0\n");
    else if(n == 2) printf("1\n");
    else printf("%lld\n", n % mod * ((n - 1) % mod) % mod * quickpow(2, n - 3, mod) % mod);
    return 0;
}