1280 前缀后缀集合

1280 前缀后缀集合

题目来源：  Codility

基准时间限制：1 秒 空间限制：131072 KB 分值: 40  难度：4级算法题

 收藏

 关注

一个数组包含N个正整数，其中有些是重复的。一个前缀后缀集是满足这样条件的下标对(P,S)， 0<= P,S < N 满足数组元素A[0..P]的值也在A[S..N - 1]的值中出现，并且A[S..N - 1]中的值也再A[0..P]中出现。换句话说前缀的集合A[0..P]与后缀集合A[S..N - 1]包含完全相同的值。求这样的前缀后缀集合的数量。

例如：3 5 7 3 3 5，共有14个集合符合条件：(1, 4), (1, 3), (2, 2), (2, 1), (2, 0), (3, 2), (3, 1), (3, 0), (4, 2), (4, 1), (4, 0), (5, 2), (5, 1), (5, 0)

本题由   @javaman 翻译。

Input

第1行：一个数N, 表示数组的长度(1 <= N <= 50000)。
第2 - N + 1行：每行1个数，对应数组中的元素Ai。(1 <= Ai <= 10^9)

Output

输出符合条件的集合数量。

Input示例

6
3
5
7
3
3
5

Output示例

14

暴力

#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<map>
using namespace std;
const int maxm = 50005;
map<int, int>p;
int a[maxm], b[maxm], flag[maxm] = { 0 }, f[maxm] = { 0 };
int main()
{
	int n, i, j, k, len = 0;
	long long sum;
	scanf("%d", &n);
	for (i = 0;i < n;i++)
	{
		scanf("%d", &a[i]);
		if (!p[a[i]])
		{
			p[a[i]] = 1;
			b[len++] = a[i];
		}
	}
	sort(b, b + len);
	for (i = 0;i < len;i++)
		p[b[i]] = i;
	int ans = 0, s = n - 1, x = 0, y = 0;
	sum = 0;
	for (i = 0;i < n;i++)
	{
		if (flag[p[a[i]]])
		{
			sum += ans;
			continue;
		}
		flag[p[a[i]]] = 1;
		x++;
		ans = 0;
		int answer = 0;
		for (j = s;j >= 0;j--)
		{
			if (!flag[p[a[j]]])
				break;
			if (answer)
			{
				ans++;
				s = j - 1;
				continue;
			}
			if (!f[p[a[j]]])
			{
				y++;
				f[p[a[j]]] = 1;
			}
			if (x == y)
			{
				answer = 1;
				ans++;
				s = j - 1;
			}
		}
		sum += ans;
	}
	printf("%lld\n", sum);
	return 0;
}