好好玩的贪心

题目描述

xiao拿到了一个长度为n的数列，数列中每个数ai都满足1<=ai<=m。请你帮他找到一个连续子序列，使其中出现次数最多的数的出现次数 与 出现次数最少的数的的出现次数 的差最大。输出这个差。
2<=n<=100000
2<=m<=10

输入

多组数据，第一行为数据组数T（1<=t<=10）。每组数据第一行包含两个数n,m。第二行为这个数列，每两个数之间用空格隔开。

输出

输出格式
每组数据输出一个数，即最大的差值。

样例输入

2
10 2
1 1 2 2 1 2 1 1 1 1
10 3
2 3 2 1 1 1 1 1 1 1

样例输出

4
6

提示

121111中 出现次数最多的为1，出现5次。最少的为2，出现1次，而不是3,4，5……因为他们没有出现。

题解：

设最后选出的区间是[L;R]，其中出现最多的数出现x 次，出现最少的数出现y 次，则x -y 一定是
该区间内最大的出现次数之差。在[1;m] 枚举i 和j，假设它们分别是答案[L;R] 中出现最多的数和
最少的数。那么，可以看到答案在i 出现时增加1,j 出现时减少1，其他数出现时不变。问题就转化
成了在一串0;-1; +1 上的最大子区间和问题。
如果i 和j 确实分别是答案[L;R] 中出现最多的数和最少的数，则最大子区间和就是答案。否则，最
大子区间和一定比答案小。因此直接对于所有i 和j 求最大区间和，从中取最大值，最终一定取到最
大的合法答案。
时间复杂度：O(nm2)

代码：

#include<stdio.h>
#define maxn 100010
int max(const int a[],int n){
    int i,cur=0,sum=0,p=0;
    for(i=0;i<n;i++){
        if(a[i]==1)p++;
        cur+=a[i];
        if(cur>sum){
            sum=cur;
        }
        if(cur<0){
            cur=0;
        }
    }
    if(p==sum)return sum-1;
    return sum;
}
int cmp(int a,int b){
    return a>b?a:b;
}
int a[maxn],b[maxn];
void work(const int a[],int n,int i,int j){
    for(int k=0;k<n;k++){
        if(a[k]==i)b[k]=1;
        else if(a[k]==j)b[k]=-1;
        else b[k]=0;
    }
}
int judge(int n){
    int f1=0,f2=0;
    for(int i=0;i<n;i++){
        if(b[i]==1)f1=1;
        if(b[i]==-1)f2=1;
    }
    if(f1&&f2)return 1;
    return 0;
}
int main(){
    int T;scanf("%d",&T);
    while(T--){
        int n,m;
        int ans=0;
        scanf("%d%d",&n,&m);
        for(int i=0;i<n;i++){
            scanf("%d",&a[i]);
        }
        for(int i=1;i<=m;i++){
            for(int j=1;j<=m;j++){
                if(i!=j){
                    work(a,n,i,j);
                    if(judge(n)){
                        ans=cmp(ans,max(b,n));
                    }
                }
            }
        }
        if(m==1)printf("0\n");
        else
        printf("%d\n",ans);
    }
    return 0;
}