项链分赃

Time Limit: 4 Sec  Memory Limit: 128 MB
Submit: 184  Solved: 116
[Submit][Status][Discuss]

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

思路

B站出题人系列；

因为某一个奇妙的 http://www.bilibili.com/video/av9885078/ ；

得证，n种宝石最多切n刀；

所以，ans∈{1,2,3};

然后就看看切几刀了；

代码实现

#include<set>
#include<cstdio>
#include<cstdlib>
#include<algorithm>
using namespace std;
const int maxn=1e5+10;
int n,a[maxn],b[3],c[3];
set<pair<int,pair<int,int> > > v;
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++) scanf("%d",&a[i]),b[a[i]]++;
    b[0]>>=1,b[1]>>=1,b[2]>>=1;
    for(int i=1;i<=n;i++){
        c[a[i]]++;
        if(b[0]==c[0]&&b[1]==c[1]&&b[2]==c[2]) return printf("1\n"),0;
    }
    c[0]=c[1]=c[2]=0;
    for(int i=1;i<=n;i++){
        c[a[i]]++;
        if(v.count(make_pair(c[0]-b[0],make_pair(c[1]-b[1],c[2]-b[2])))) return printf("2\n"),0;
        v.insert(make_pair(c[0],make_pair(c[1],c[2])));
    }
    return printf("3\n"),0;
}