1004 C. Sonya and Robots

https://codeforces.com/group/qWc0vtZbjV/contest/1004/problem/C

题意：有两个机器人，从一组数组的两端向中间前进，你可以给这两个机器人设定一个数字，这样机器人在遇到数组中的数字与设定的数字相同的时候，就会停下来，避免了相撞。现给你一个数组，问有几种设定方法可以避免相撞，（左右两个机器人设定相同数字视为重复）。

如果是理想情况下，数组中数字没有重复的，答案就是1+2+3…+n-1。例如1 2 3 4，对于1来说有三种，2来说有两种，3来说有一种，一共是1+2+3。然后处理重复的情况你如果不优化的话，就会使两重循环n^2的复杂度，肯定tle，所以我们预处理一下，倒着遍历来求出前一个元素在计算时要减去的重复量。

#include <bits/stdc++.h>
using namespace std;
#define qc std::ios::sync_with_stdio(0);
long long a[100001];
long long b[100001];
long long q[100001];
long long h[100001];

int main() {
	qc;
	cin.tie(0);
	int n;
	int k = 0;
	cin >> n;
	for (int i = 1; i <= n; i++) {
		cin >> a[i];
	}
	for (int i = n; i >= 1; i--) {
		//继承
		b[i] += b[i + 1];
		//传递
		h[a[i]]++;
		if (h[a[i]] > 1)
			b[i - 1]++;
	}
	long long ans = 0;
	for (int i = 1; i <= n - 1; i++) {
		if (q[a[i]] > 0)
			continue;
		q[a[i]]++;
		ans = ans + n - i - b[i];
	}
	cout << ans;
}