题解:P10523 [XJTUPC2024] Everyone's ALL IN

· · 题解

P10523 [XJTUPC2024] Everyone's ALL IN 题解

分析

水黄没错了。

主要思路是拆一个式子,拿样例中第一天来分析。1\times 3+1\times 4+1\times 8+2\times 3+2\times 4+2\times 8+9\times 3+9\times 4+9\times 8=180 这是第一天的收益,由题可知每天的收益是由 x 骑士团的成员编号分别乘上 y 骑士团的成员编号再加起来。根据小学的乘法分配律,1\times 3+1\times 4+1\times 8+2\times 3+2\times 4+2\times 8+9\times 3+9\times 4+9\times 8=(1+2+9) \times 3+(1+2+9) \times 4+(1+2+9) \times 8,再用一次乘法分配律,(1+2+9) \times 3+(1+2+9) \times 4+(1+2+9) \times 8=(1+2+9)\times (3+4+8)。这时我们就可以得到思路了。统计每个骑士团的成员编号之和 cnt_i,每次询问用 cnt_x \times cnt_y 即可。

代码

#include<bits/stdc++.h>
#define int long long
using namespace std;
int cnt[1000005], n, m;
signed main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        int x;
        cin>>x;
        cnt[x]+=i;
    }
    while(m--)
    {
        int x, y;
        cin>>x>>y;
        cout<<cnt[x]*cnt[y]<<"\n";
    }
}