Tinkoff Internship Warmup Round 2018 and Codeforces Round #475 (Div. 2) C. Alternating Sum

最新推荐文章于 2018-07-24 21:21:18 发布

原创最新推荐文章于 2018-07-24 21:21:18 发布 · 727 阅读

0 ·

CC 4.0 BY-SA版权

数据结构与算法专栏收录该内容

63 篇文章

订阅专栏

本文提供了一个针对C.AlternatingSum问题的C++实现方案。该算法使用快速幂运算来解决特定数学序列求和问题，并考虑了边界条件q为1时的特殊情况。

C. Alternating Sum
time limit per test1 second
memory limit per test256 megabytes
inputstandard input
outputstandard output

这里写图片描述

注意判断q为1的情况
代码：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll model = 1e9 + 9;
ll qpow(ll a, ll b, ll mod)
{
	ll ret = 1;
	while (b)
	{
		if (b & 1) ret = ret * a % mod;
		b >>= 1;
		a = a * a % mod;
	}
	return ret;
}
int main(void)
{
	int n, a, b, k;
	scanf("%d%d%d%d", &n, &a, &b, &k);
	char str[100005];
	scanf("%s", str);
	int i;
	ll res = 0;
	for (i = 0; i <= k - 1; i++)
	{
		if (str[i] == '-')
		{
			res = (res - qpow(a, n-i, model) * qpow(b, i, model) % model  + model) % model;
		}
		else
		{
			res = (res + qpow(a, n-i, model) * qpow(b, i, model) % model  + model) % model;
		}
	}
	ll t = qpow(b, k, model) * qpow(a, k * (model - 2), model) % model;
	if (t == 1)
	{
		res = res * ((n + 1) / k) % model;
	}
	else
	{
		res = res * (qpow(t, (n+1)/k, model) - 1) % model * qpow(t-1, model - 2, model) % model;
	}
	
	cout << res;
}