题目大意:N头牛,排列在一条直线上,坐标为0~10亿。牛有两个品种,求最大的一段区间,该区间内两种牛的个数相同。
题目网址:http://soj.me/1375
思路:对输入按坐标排序,扫描一次,用s[i][1]累计记录前i条记录A种牛出现的头数,s[i][2]记录B种,再归一化。找相同的s[i][2]中两头牛距离最远就行了。
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <algorithm>
const int MAXN = 50003;
const int MAXK = 2;
const int prime = MAXN;
int ans = 0;
int s[MAXN][4];
int hashtable[MAXN];
struct Input {
int type;
int xValue;
}inputs[MAXN];
int cmp( const void *a ,const void *b)
{
return (*(Input *)a).xValue > (*(Input *)b).xValue ? 1 : -1;
}
/*
int get_hash_code(int* v, int k)//对二维数组每一行的前k列进行HASH
{
int i, p = 0;
for (i = 1; i < k + 1; ++i)
{
p = ((p << 2) + (v[i] >> 4)) ^ (v[i] << 10);
}
p = p % prime;
while (p < 0)
{
p = p + prime;
}
return p;
}
*/
bool cmp(int *p,int *q)
{
return p[1]<q[1];
}
int main()
{
// freopen("1.txt", "r", stdin);
int i, j, n, k;
k=2;
scanf("%d", &n);
memset(hashtable, -1, sizeof(hashtable));
memset(s, 0, sizeof(s));
memset(inputs, 0, sizeof(inputs));
hashtable[0] = 0;
for(i=1;i<=n;i++)
{
scanf("%d %d",&inputs[i].type,&inputs[i].xValue);
}
//结构体排序
qsort(inputs,n+1,sizeof(inputs[0]),cmp);
//s[i][1]记录前i个数据中type=0的个数,s[i][2]是type=1的个数
for (i = 1; i < n + 1; ++i)
{
s[i][3]=inputs[i].xValue;
if(inputs[i].type==0){
s[i][1]=s[i-1][1]+1;
s[i][2]=s[i-1][2];
}
else {
s[i][1]=s[i-1][1];
s[i][2]=s[i-1][2]+1;
}
}
for ( i = 1; i < n + 1; ++i)
{
for (j = k; j > 0; --j)
{
s[i][j] -= s[i][1];//重点:归一化,这样s[i][1]=0
}
//只有两种牛(其中一种还归一化为0),就不用HASH了
//int p = get_hash_code(s[i], k);
int p=s[i][2];
while (hashtable[p] != -1)
{
for (j = 1; j < k + 1; ++j)
{
if (s[i][j] != s[hashtable[p]][j])
{
goto label1;
}
}
//找到s[hashtable[p][2]跟s[i][2]相同的,说明hashtable[p]+1到i这一段0和1个数相同,抵消了
ans = inputs[i].xValue - inputs[hashtable[p]+1].xValue > ans ? inputs[i].xValue - inputs[hashtable[p]+1].xValue : ans;
break;
label1:
p++;
}
if (hashtable[p] == -1)
{
hashtable[p] = i;
}
}
printf("%d\n", ans);
return 0;
}