bzoj1702[Usaco2007 Mar]Gold Balanced Lineup 平衡的队列*

bzoj1702[Usaco2007 Mar]Gold Balanced Lineup 平衡的队列

题意：

N头牛，一共K种特色。每头牛有多种特色。[i,j]段被称为balanced当且仅当K种特色在[i,j]内拥有次数相同。求最大的[i,j]段长度。n≤100000，k≤30。

题解：

得到式子：a[i][l]-a[j][l]=a[i][l-1]-a[j][l-1],l在2..k之间，移项得a[i][l]-a[i][l-1]=a[j][l]-a[j][l-1],l在2..k之间，故可以定义一个结构体，里面包含所有的a[i][l]-a[i][l-1],l在2..k之间，然后用set查找满足所有元素相等的最小j即可。

代码：

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <set>
#define inc(i,j,k) for(int i=j;i<=k;i++)
#define maxn 100010
using namespace std;

inline int read(){
    char ch=getchar(); int f=1,x=0;
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1; ch=getchar();}
    while(ch>='0'&&ch<='9')x=x*10+ch-'0',ch=getchar();
    return f*x;
}
int sum[maxn][50],n,k,ans;
struct nd{
    int id; int num[50];
    bool operator < (const nd &a)const{inc(i,1,k-1)if(num[i]!=a.num[i])return num[i]<a.num[i]; return 0;}
};
multiset<nd>st; nd a;
int main(){
    n=read(); k=read();
    inc(i,1,n){
        int x=read(); inc(j,1,k)sum[i][j]=sum[i-1][j]; for(int j=k-1;j>=0;j--)if(x&(1<<j))sum[i][j+1]++;
    }
    inc(j,1,k-1)a.num[j]=0; a.id=0; st.insert(a);
    inc(i,1,n){
        inc(j,1,k-1)a.num[j]=sum[i][j]-sum[i][j+1]; a.id=i;
        multiset<nd>::iterator sti=st.find(a); if(sti==st.end())st.insert(a);else{
            ans=max(ans,i-sti->id);
        }
    }
    printf("%d",ans); return 0;
}

20160929