题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=2430
题目大意:有n坨豌豆,每坨都有w[i]个,现在要从中选择连续的若干坨,然后用一个能装p个豌豆的背包装豆子,直到剩下的豌豆数量<=p为止,还要求剩下的豌豆数量少于k,问最多能装多少包。
解题思路:题意很明了,模型也很好抽象。本题就是要选择连续的一段和为sum,使得max(sum/p) 且sum%p<=k。那么要怎么算这个sum呢?这里就大有文章可做了。
其实这类求区间和的问题很经常是用sum[i]- sum[j-1]这种方式来表达的,我们设sum[i]表示0到i坨豌豆的总数量,那么上面的sum就是sum[i]-sum[j-1]。从而得到一个式子,max(sum[i]-sum[j]) (j < i && (sum[i]-sum[j])%p<=k),表示以i为区间尾,因为豌豆数量非负数,那我们向前找个最小的下标使得差最小且符合条件即可。
我们设sum[i]为x,sum[j]为y,那么上式变成0<=(x-y)%p<=k且x-y最大,约束条件是x大等于y,我们要利用这些条件快速地找到x对应的那个最小的y。
0<=(x-y)%p<=k --> 0<=(x%p - y%p + p) % p<=k
如果x%p >= y%p,那么x%p-k<=y%p<=x%p,我们就可以根据x%p来定位y%p.
如果x%p < y%p,那么我们可以换个角度想,在以y结尾的时候就会有上面的情况。
那问题就剩下一个了,如果根据x%p来快速定位y%p。写法有多种,用线段树和树状数组都可以写,但我用线性的单调队列来写,时间78ms,是提交记录的第二名。
我先将每个和对p取余得到一个余数r数组,然后按余数大小优先位置其次的顺序排序。排完序之后就可以根据ri - k <= rj <= ri来往前找那个最小的j。遍历数组r,因为r是有序的,我用单调队列储存r小于当前的最小下标。关于单调队列,我就不多说了,百度google去。最后更新答案即可。
总复杂度还是(nlogn),主要用于排序。
测试数据:
20
2 19 3
17 3
Out:Case 1: 1
2 19 1
17 3Out:Case 2: 1
2 19 3
17 6Out:Case 3: -1
10 20 10
0 3 1 8 19 39 2 9 1 8Out:Case 4: 4
3 100 10
32 34 23Out:Case 5: -1
1 5 3
5Out:Case 6: 1
5 15 4
7 8 3 10 10Out:Case 7: 2
C艹代码:
#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
#define MAX 1000010
#define max(a,b) (a)>(b)?(a):(b)
struct node {
int x,in;
}r[MAX],qu[MAX];
__int64 sum[MAX],ans;
int head,tail;
int n,p,k,mmin[MAX];
int cmp(node a,node b) {
if (a.x == b.x) return a.in < b.in;
else return a.x < b.x;
}
void Solve_AC() {
int i,j;
head = 0,tail = 1;
for (i = 1; i <= n; ++i) {
if (tail > head) mmin[r[i].in] = -1;
else mmin[r[i].in] = qu[tail].in;
while (tail <= head
&& r[i].in < qu[head].in) head--;
qu[++head] = r[i];
while (tail <= head && r[i + 1].x - qu[tail].x > k)
tail++;
}
for (i = 1; i <= n; ++i) {
if (sum[i] % p <= k) ans = max(ans,sum[i] / p);
if (mmin[i] == -1 || mmin[i] >= i) continue;
ans = max(ans,(sum[i] - sum[mmin[i]]) / p);
}
}
int main()
{
int i,j,t,cas = 0;
scanf("%d",&t);
while (t--) {
scanf("%d%d%d",&n,&p,&k);
for (i = 1; i <= n; ++i) {
scanf("%d",&sum[i]);
r[i].in = i;
sum[i] += sum[i-1];
r[i].x = sum[i] % p;
}
sort(r+1,r+1+n,cmp);
ans = -1,Solve_AC();
printf("Case %d: %I64d\n",++cas,ans);
}
}
本文ZeroClock原创,但可以转载,因为我们是兄弟。