BZOJ 4893 项链分赃

Description

有一串长度为n的项链，上面有红绿蓝三种颜色的珠子，每种颜色的珠子数目都是偶数，现在要你把它切几刀分成

若干段，把其中一些段分给海盗1，剩余的段分给海盗2，要求两个海盗分得的每种颜色的珠子数量都相同，请输出

最少需要切多少刀。

Input

第一行一个整数n，表示项链的长度。

第二行n个0~2的整数，分别表示红绿蓝三种颜色。

Output

一行一个整数，为最少切多少刀。

Sample Input

6
0 2 2 1 0 1

Sample Output

2
样例解释：切两刀，分成{0,2},{2,1,0},{1}三份，第二份给海盗1，剩下给海盗2即可。

HINT

n<=100000

Source

By Monster_Yi

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

思路+pair~

lty实在是太神了……这脑洞……

证明详见出题人的博客：http://www.cnblogs.com/juruolty/p/6806029.html，讲得非常清楚~

#include<cstdio>
#include<iostream>
#include<map>
using namespace std;

int n,a[100001],c[3],num[3];

map<pair<int,pair<int,int> >,int> b;

int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0' || ch>'9') {if(ch=='-') f=-1;ch=getchar();}
	while(ch>='0' && ch<='9') {x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
	return x*f;
}

int main()
{
	n=read();
	for(int i=1;i<=n;i++) a[i]=read(),c[a[i]]++;c[1]>>=1;c[2]>>=1;c[0]>>=1;
	for(int i=1;i<=n>>1;i++)
	{
		num[a[i]]++;
		if(num[1]==c[1] && num[2]==c[2] && num[0]==c[0])
		{
			puts("1");return 0;
		}
	}
	num[1]=num[2]=num[0]=0;
	for(int i=1;i<=n;i++)
	{
		num[a[i]]++;
		if(b.count(make_pair(num[0]-c[0],make_pair(num[1]-c[1],num[2]-c[2]))))
		{
			puts("2");return 0;
		}
		b[make_pair(num[0],make_pair(num[1],num[2]))]=1;
	}
	puts("3");
	return 0;
}