贡献法
最近做了一些题目,发现有些题目是正着想复杂度会超时,从对答案的贡献的角度考虑会清晰很多。
C. Sequence Pair Weight
这道题目正常取枚举区间计算区间内有多少相同点对显然是来不及的,光是枚举区间就会花费\(O(n^2)\)的时间复杂度
我们可以考虑每个点对对于答案的贡献,也就是说每个点对所在的不同区间的个数
对于一个点对\(i \quad j 答案是i*(n-j+1)\)
\(可以直接枚举点对吗?这样显然也是来不及的,考虑如何优化,可以取枚举i \quad j中的一个\)
\(考虑枚举i的话我们能计算除所有他后面出现过的j的和吗?这样显然不好计算\)
\(考虑枚举j对于每一个j我们计算所有满足\displaystyle\sum_{i=1}^ni \quad 并且a[i]=a[j]\quad i<=j\)
对于这样的情况我们是比较好计算的,我们可以用一个map存一下之前的所有i的和
#include <bits/stdc++.h>
#define int long long
#define rep(i,a,b) for(int i = (a); i <= (b); ++i)
#define fep(i,a,b) for(int i = (a); i >= (b); --i)
#define ls p<<1
#define rs p<<1|1
#define PII pair<int, int>
#define pll pair<long long, long long>
#define ll long long
#define ull unsigned long long
#define db double
#define endl '\n'
#define debug(a) cout<<#a<<"="<<a<<endl;
#define IOS ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
#define INF 0x3f3f3f3f
#define x first
#define y second
using namespace std;
const int N=1e5+10;
int n,a[N];
void solve()
{
cin>>n;
map<int,int>sum;
ll ans=0;
rep(i,1,n)
{
cin>>a[i];
ans+=sum[a[i]]*(n-i+1);
sum[a[i]]+=i;
}
cout<<ans<<endl;
rep(i,1,n) a[i]=0;
}
signed main()
{
IOS
// freopen("1.in", "r", stdin);
int t;
cin>>t;
while(t--)
solve();
return 0;
}
```

浙公网安备 33010602011771号