容易发现对于每一个数,我们只需要知道他是否大于中位数(b),所以对于每一个数特殊处理一下,大于中位数的填1,小于的填-1,这样问题就转化为了能选多少个子序列使得和为0。
考虑到每一个数的范围都在1~n之间,所以使用哈希的思想。
设pos表示中位数的位置
sum[i]表示1~pos和为i的子序列的个数
首先预处理出来1~pos之间的sum,然后枚举右端点:
假设从pos到目前位置的区间和为i,那么ans+=sum[-i]即可。
注意:C++中不能用负数做下标,所以用1、2记录正负(注意0的情况)
数组开大10个就过了o(╯□╰)o
BZOJ现在又挂了。。。等上去了在贴代码。。。
#define MAX_M 10000+10
#include<stdio.h>
#include<stdlib.h>
int dp[MAX_M],x[MAX_M],y[MAX_M],time[MAX_M];
int main()
{
int n,m,ans=0,a,b,c,cnt=0;
scanf("%d %d",&n,&m);
for(int i=1;i<=m;i++)
{
scanf("%d%d%d",&a,&b,&c);
if(b>0 && b<=n &&c>0 &&c<=n)
{
time[++cnt]=a; x[cnt]=b; y[cnt]=c;
}
}
for(int i=1;i<=cnt;i++)
{
for(int j=i;j>=1;j--)
{
if(dp[j]<dp[i]) continue;
if(abs(x[j]-x[i])+abs(y[j]-y[i])<=abs(time[i]-time[j]) && (dp[j]+1>dp[i]))
dp[i]=dp[j]+1;
}
if(ans<dp[i]) ans=dp[i];
}
printf("%d",ans);
return 0;
}