✅ 魔法元素的总能量
🌟 故事背景
魔法学院的仓库里存放着 nn 种魔法元素,每种元素都有自己的能量值,能量值按 等比数列 排列:
E1,E2,E3,…,EnEk=a1×qk−1Ek=a1×qk−1其中:
这些元素被划分到两个收藏箱:
- 有 A 种元素被放入 火箱;
- 有 B 种元素被放入 水箱;
- 有 C 种元素被同时放入 火箱和水箱。
魔法师想要计算:
至少放入一种箱子的元素的能量总和是多少?
📥 输入格式
n a1 q
A B C
含义:
- n:元素总数(序号 1 到 n)
- a1:首项
- q:公比
- A,B,C:两个箱子各自元素数和交集数
📤 输出格式
X
S
- 第一行:至少放入一种箱子的元素个数 X=A+B−C
- 第二行:前 X 个元素能量和 S
📋 输入样例
10 2 3
12 15 8
计算步骤:
- 容斥原理:
X=A+B−C=12+15−8=19
- 等比数列求和(取前 X 项):
S=a_1×q−1qX−1,(q=1)代入a1=2,q=3,X=19:
S=2×3−1319−1(具体数值请程序计算)
输出样例:
19
<计算出的 S>
✨ 提示公式
容斥原理:
X=A+B−CX = A + B - C等比数列求和:
S=a_1×q−1qX−1,q=1q=1时S=a1×Xq=1 时 S=a1×X
📌 数据范围
1≤n≤105,1≤a_1,q≤106(如果 X 很大,可能需要大数或取模处理)