Codeforces 963A Alternating Sum（找规律）_alternating sum第一行包含四个整数 nn , aa , bb 和 kk 。第二行包含-CSDN博客

本文链接：https://blog.csdn.net/CSDNjiangshan/article/details/81698606

题目链接：Alternating Sum

题意

给定 $a,b$ 的值，以及 $n+1$ 项系数 $s_0,s_1,\cdots,s_n$ ，所有系数不是 $1$ 就是 $-1$ ，这 $n+1$ 项系数是以 $k$ 为循环节的，且 $k|n+1$ ，给定前 $k$ 项系数，求公式 $\sum_{i=0}^ns_ia^{n−i}b^i\mod10^9+9$ 的值。

输入

第一行为 $4$ 个整数 $n,a,b,k~(1\leq n\leq10^9,1\leq a,b\leq10^9,1\leq k\leq10^5)$ ，第二行包含一个长度为 $k$ 的字符串，字符串只包含 + 和 - 这两种字符，表示系数的正负。

输出

输出计算结果。

样例

输入

2 2 3 3
+-+

输出

提示

(\sum_{i = 0}^{n} s_{i} a^{n - i} b^{i}) = 2^{2} 3^{0} - 2^{1} 3^{1} + 2^{0} 3^{2} = 7

$(\sum_{i=0}^ns_ia^{n-i}b^i)=2^23^0-2^13^1+2^03^2=7$ 。

输入
4 1 5 1 `-`
输出
999999228
提示
$(\sum_{i=0}^ns_ia^{n-i}b^i)=-1^45^0-1^35^1-1^25^2-1^15^3-1^05^4=-781=999999228

\mod10^9+9$。|

题解

只观察每 $k$ 个数字，会发现这这些项是以 $(\frac{b}{a})^k$ 为公比的等差数列，套上等比数列求和公式就可以得到答案，注意公比为 1 <script type="math/tex" id="MathJax-Element-17">1</script> 的情况。

过题代码

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cmath>
#include <climits>
#include <cstring>
#include <string>
#include <vector>
#include <list>
#include <queue>
#include <stack>
#include <map>
#include <set>
#include <functional>
#include <algorithm>
using namespace std;

#define LL long long
const LL MOD = 1000000009;
const int maxn = 100000 + 100;
LL n, a, b, k;
char str[maxn];

LL fast_pow(LL res, LL n) {
    LL ans;
    for(ans = 1; n != 0; n >>= 1) {
        if(n % 2 == 1) {
            ans = (ans * res) % MOD;
        }
        res = (res * res) % MOD;
    }
    return ans;
}

void exgcd(LL a, LL b, LL &x, LL &y) {
    if(b == 0) {
        x = 1;
        y = 0;
        return ;
    }
    exgcd(b, a % b, y, x);
    y -= x * (a / b);
}

LL inv(LL x) {
    LL xx, yy;
    exgcd(x, MOD, xx, yy);
    xx = ((xx % MOD) + MOD) % MOD;
    return xx;
}

int main() {
    #ifdef LOCAL
    freopen("test.txt", "r", stdin);
    #endif // LOCAL

    while(scanf("%I64d%I64d%I64d%I64d", &n, &a, &b, &k) != EOF) {
        if(k >= n + 1) {
            k = n + 1;
        }
        scanf("%s", str);
        LL tmp = 0;
        LL inva = inv(a);
        LL invb = inv(b);
        LL aa = fast_pow(a, n + 1);
        LL bb = invb;
        for(LL i = 0; i < k; ++i) {
            aa = (aa * inva) % MOD;
            bb = (bb * b) % MOD;
            if(str[i] == '+') {
                tmp += (aa * bb) % MOD;
            } else {
                tmp -= (aa * bb) % MOD;
            }
            tmp = ((tmp % MOD) + MOD) % MOD;
        }
        LL q = fast_pow((inva * b) % MOD, k);
        LL nn = (n + 1) / k;
        if(q == 1) {
            tmp = (nn * tmp) % MOD;
        } else {
            LL fenzi = (((1 - fast_pow(q, nn)) % MOD) + MOD) % MOD;
            LL fenmu = ((1 - q) % MOD + MOD) % MOD;
            fenzi = (fenzi * inv(fenmu)) % MOD;
            tmp = (tmp * fenzi) % MOD;
        }
        printf("%I64d\n", tmp);
    }

    return 0;
}