UCF Local Programming Contest Round 1A C.Unique Values (思维)

题目链接: Unique Values

大致题意

给出一段长度为n的序列, 问有多少个连续子序列中不含有相同的元素.

解题思路

思维

我们从前向后扫描序列, 设当前位置为pos, 统计当子序列右端点为pos时的答案. 用last记录上一次含有相同元素的位置.

这样[last + 1, pos]为一段不含有相同元素的连续子序列, 则以pos作为结尾的连续子序列有: [pos, pos], [pos - 1, pos], …, [last + 1, pos]. 共有num = pos - (last + 1) + 1 = pos - last个.

最终答案, 我们把每次得到的num_pos相加即可.

---

我们可以证明上述统计方式是不重不漏的.

因为我们枚举了所有右端点的情况, 而合法的连续子序列的右端点必然被枚举到了, 保证不漏.

每次枚举的区间的右端点都在改变, 因此可以保证每次统计出的num_pos是不相交的, 保证不重.

AC代码

#include <bits/stdc++.h>
#define rep(i, n) for (int i = 1; i <= (n); ++i)
using namespace std;
typedef long long ll;
int main()
{
	unordered_map<ll, int> mp;

	int n; cin >> n;
	ll res = 0; int last = 0;
	rep(i, n) {
		ll x; scanf("%lld", &x);

		last = max(last, mp[x]);
		mp[x] = i;
		res += i - last;
	}

	cout << res << endl;

	return 0;
}

END