AT_abc318_e [ABC318E] Sandwiches 题解

没啥困难的。

枚举 $i$，然后找出所有在 $i$ 后面可以出现的 $k$。对于某个 $k$，设 $(i,k)$ 中有 $x$ 个 $a_j = a_i$，那么 $k$ 的贡献是 $k-1-i-x$。

发现后面 $x$ 是一个等差数列，前面可以把 $i$ 提出来，$k-1$ 可以动态维护和，就做完了。

#include <bits/stdc++.h>
using namespace std;
//#define int long long

const int N = 3e5 + 5, MOD = 1e9 + 7; // Remember to change

int n, m, q, t, a[N];
set<int> st[N];
long long nowsum[N];

namespace FastIo
{
	#define QUICKCIN ios::sync_with_stdio(0), cin.tie(0), cout.tie(0)
	int read()
	{
		char ch = getchar();
		int x = 0, f = 1;
		while ((ch < '0' || ch > '9') && ch != '-') ch = getchar();
		while (ch == '-')
		{
			f = -f;
			ch = getchar();
		}
		while (ch >= '0' && ch <= '9')
		{
			x = (x << 1) + (x << 3) + (ch ^ 48);
			ch = getchar();
		}
		return x * f;
	}
	template<class T>
	void write(T x)
	{
		if (x < 0)
		{
			putchar('-');
			x = -x;
		}
		if (x > 9) write(x / 10);
		putchar(x % 10 + '0');
	}
	template<class T>
	void writeln(T x)
	{
		write(x);
		putchar('\n');
	}
}

int main()
{
	ios::sync_with_stdio(0), cin.tie(nullptr), cout.tie(nullptr);
	cin>>n;
	for(int i=1;i<=n;i++) cin>>a[i],st[a[i]].insert(i),nowsum[a[i]]+=(i-1);
	long long ans=0LL;
	for(int i=1;i<=n;i++)
	{
		nowsum[a[i]]-=(i-1);
		st[a[i]].erase(st[a[i]].begin());
		long long l=1,r=st[a[i]].size()-1;
		long long ss=(l>r?0LL:(l+r)*(r-l+1)/2);
		ans+=nowsum[a[i]]-(long long)st[a[i]].size()*(i)-ss;
	}
	cout<<ans<<"\n";
	return 0;
}