洛谷 P10523 [XJTUPC2024] Everyone‘s ALL IN 题解

分析

显然，直接套两层 for 循环累加求和会 TLE。

可以设每次比试的两个骑士团中分别有 $a$ 和 $b$ 个人，第一个骑士团的成员的编号依次为 $c_1,c_2,c_3,\dots ,c_a$，第二个骑士团的成员的编号依次为 $d_1,d_2,d_3,\dots ,d_b$，如果这两个骑士团进行比试，将可以获得 $c_1\times d_1+c_1\times d_2+c_1\times d_3+\cdots +c_1\times d_b+c_2\times d_1+c_2\times d_2+c_2\times d_3+\cdots +c_2\times d_b+\cdots +c_a\times d_1+c_a\times d_2+c_a\times d_3+\cdots +c_a\times d_b$ 个源石锭，根据乘法分配律化简后得 $\left(\sum _{i=1}^a{c}_i\right)\times \left(\sum _{i=1}^b{d}_i\right)$，对于其他的任意两个骑士团进行比试也是一样，所以只需要将每个骑士团的成员的编号之和预处理出来即可。

Code

#include<iostream>
#define int long long//不开 long long 见祖宗
using namespace std;
int n,m,a,x,y,cnt[1000001];
signed main() {
	ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
	cin>>n>>m;
	for(int i=1;i<=n;++i) cin>>a,cnt[a]+=i;//累加编号和
	while(m--) {
		cin>>x>>y;
		cout<<cnt[x]*cnt[y]<<"\n";//直接输出乘积
	}
	return 0;
}