HUSTOJ 入门题目集2 问题D：免费馅饼

题目描述

都说天上不会掉馅饼，但有一天gameboy正走在回家的小径上，忽然天上掉下大把大把的馅饼。说来gameboy的人品实在是太好了，这馅饼别处都不掉，就掉落在他身旁的10米范围内。馅饼如果掉在了地上当然就不能吃了，所以gameboy马上卸下身上的背包去接。但由于小径两侧都不能站人，所以他只能在小径上接。由于gameboy平时老呆在房间里玩游戏，虽然在游戏中是个身手敏捷的高手，但在现实中运动神经特别迟钝，每秒种只有在移动不超过一米的范围内接住坠落的馅饼。现在给这条小径如图标上坐标：

为了使问题简化，假设在接下来的一段时间里，馅饼都掉落在0-10这11个位置。开始时gameboy站在5这个位置，因此在第一秒，他只能接到4,5,6这三个位置中其中一个位置上的馅饼。问gameboy最多可能接到多少个馅饼？（假设他的背包可以容纳无穷多个馅饼）

输入

输入数据有多组。每组数据的第一行为以正整数n(0<n<100000)，表示有n个馅饼掉在这条小径上。在结下来的n行中，每行有两个整数x,T(0<T<100000),表示在第T秒有一个馅饼掉在x点上。同一秒钟在同一点上可能掉下多个馅饼。n=0时输入结束。

输出

每一组输入数据对应一行输出。输出一个整数m，表示gameboy最多可能接到m个馅饼。

提示：本题的输入数据量比较大，建议用scanf读入，用cin可能会超时。

样例输入

6
5 1
4 1
6 1
7 2
7 2
8 3
0

样例输出

4

Solution

首先我们确定这是一道DP题目，我们可以通过设立f[i][j]表示第i秒移动到坐标j处所能获得的最大馅饼数，再设a[i][j]表示第i秒坐标j处掉落的馅饼数，不难得出动态转移方程：

$$f_{i,j}=max(f_{i-1,j-1}，f_{i-1,j}，f_{i-1,j+1})+a_{i,j}$$

然而题目规定了gameboy只能从坐标5出发，那么前几秒内活动范围是受限的，我们不妨从最后一次掉落馅饼的时间出发，一路往前DP，一直到第0秒，最后输出f[0][5]也是可以的，不过动态转移方程现在变为：

$$f_{i,j}=max(f_{i+1,j-1}，f_{i+1,j}，f_{i+1,j+1})+a_{i,j}$$

这道题的数据比较宽松，t最大才到100k，数组完全开得起，所以先贴上clear_conscience的代码（tql%%%

//记得特判0和10边界，这段代码已经将f数组向右移动了一位防止读到数组外了
#include<cstdio>
#include<cstring>

const int N = 1e5 + 5;
int a[N][12], n, T, f[N][12], ans;

int max(int a, int b) {return a < b ? b : a;}

int main()
{
	freopen("task2.in","r",stdin);
	scanf("%d", &n);
    while(n)
	{
		memset(f, -100, sizeof f), memset(a, 0, sizeof a), ans = T = 0;
        for(int x, t, i = 1; i <= n; i++)
            scanf("%d%d", &x, &t), ++a[t][x + 1], T = max(T, t);
        f[0][6] = 0;
        for(int i = 1; i <= T; i++)
            for(int j = 1; j <= 11; j++)
                f[i][j] = max(f[i - 1][j], max(f[i - 1][j - 1], f[i - 1][j + 1])) + a[i][j];
        for(int i = 1; i <= 11; i++) ans = max(ans, f[T][i]);
        printf("%d\n", ans), scanf("%d", &n);
    }
    return 0;
}

但是我们不能只甘心于要开辣么大的空间而且冗余度这么高，每次还要遍历从100000秒到1秒，太久了！

所以我们考虑滚动数组并且跳过没有馅饼下落的时间，只需要定义一个f[2][11]的数组就能满足需求

先将读入的馅饼按照下落时间从晚到早依次排序，每次通过指针将下落时间相同的馅饼挑出来，定义cnt[i]表示该时间点在坐标i处下落的馅饼个数，统计出该时间点的cnt，直接进行转移即可

要注意，由于该过程中我们是两个有馅饼下落的时间点间进行动态转移，这个过程中能移动不止一步，所以定义cha表示该时间点和上个时间点的时间差，cha就是两个时间点间gameboy能够移动的步数，然后定义能转移给位置i处的左边界l和右边界r，直接for循环进行转移。

详见AC代码：

#include<bits/stdc++.h>
using namespace std;
int n,tou,wei,pin,cha,l,r;
struct Node{int x,t;}a[100002];
int f[2][11],cnt[11];
bool cmp(Node a1,Node a2)
{
	return a1.t>a2.t;
}
int main()
{
	scanf("%d",&n);
	while(n!=0)
	{
		for(int i=1;i<=n;i++)
			scanf("%d%d",&a[i].x,&a[i].t);
		sort(a+1,a+1+n,cmp);
		memset(f,0,sizeof f);
		a[n+1].t=0;a[0].t=100010;
		tou=1,wei=1,pin=1;
		while(tou<=n)
		{
			memset(cnt,0,sizeof cnt);
			while(a[tou].t==a[wei].t&&wei<=n)
			{
				cnt[a[wei].x]++;
				wei++;
			}
			cha=abs(a[tou].t-a[tou-1].t);
			tou=wei;
			for(int i=0;i<=10;i++)
			{
				l=max(0,i-cha),r=min(10,i+cha);
				f[pin][i]=0;
				for(int j=l;j<=r;j++)
				{
					f[pin][i]=max(f[pin][i],f[pin^1][j]);
				}
				f[pin][i]+=cnt[i];
			}
			pin=pin^1;
		}
		cha=abs(0-a[n].t);
		l=max(0,5-cha),r=min(10,5+cha);
		f[pin][5]=0;
		for(int j=l;j<=r;j++)
		{
			f[pin][5]=max(f[pin][5],f[pin^1][j]);
		}
		printf("%d\n",f[pin][5]);
		scanf("%d",&n);
	}
	return 0;
}

总结

虽然空间省了不少，但是时间上似乎并没有太大优势，因为每组数据都需要一次快排，所以也要O(n log n+n)，不过面对时间范围和跨度较大的数据会比较有优势吧