bzoj 1314: River过河 优先队列

1314: River过河

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 26  Solved: 10
[Submit][Status][Discuss]

Description

ZY
带N个小Kid过河,小KID分成两种:高一年级,高二年级,由于存在代沟问题,如果同一条船上高一年级生和高二年级生数量之差超过K,就会发生不和谐的
事件.当然如果一条船上全是同一年级的,就绝对不会发生争执.现在ZY按小KID队列的顺序依次安排上船,并且不能让他们在过河时发生争执.对于当前等待
上船的小KID来说,要么让他上船,要么将停在渡口的船开走,再让他上另一条船,同一条船上的人不过超过M人.为了让所有的小KID过河,在知悉小KID
队列的情况下,最少需要多少条船.

Input

第一行给出N,M,K.含义如上所述
下行N行用来描述小KID的队列,每行一个字符”A”或者”L”

Output

最少需要多少条船

Sample Input

5 4 1
A
L
L
L
A

Sample Output

2

HINT

前三个人一条船,后两个人一条船
数据范围
30% 数据中1<=N<=1000
100%数据中1<=N<=250000,1<=M,K<=N

　　这道题过的人这么少不太正常啊。

　　容易看出，如果对输入数列换成+1/-1，然后做前缀和，当前dp[i]可以由j-i>=m and abs(a[j]-a[i])<=k 一个矩形内区域的dp值转移出来，首先kdtree肯定没有问题，看有无简单点的做法，思考整体二分，发现不太好处理，注意上述的矩形区域有着固定的宽，而且x坐标单调递增，于是很容易想到对于每一个a[i]维护优先队列了。

　　用deque<int>要mle，原因是deque封装了大量维护下标访问的量，而改成list<int>内存要好很多。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<vector>
#include<deque>
#include<list>
using namespace std;
#define MAXN 251000
#define lch (now<<1)
#define rch (now<<1^1)
#define smid ((l+r)>>1)
#define MAXT MAXN*4
#define INF 0x3f3f3f3f
int a[MAXN];
int dp[MAXN];
list<int> lst[MAXN];
int sgt[MAXT];
void Modify_sgt(int now,int l,int r,int pos,int v)
{
        if (l==r)
        {
                sgt[now]=v;
                return ;
        }
        if (pos<=smid)
                Modify_sgt(lch,l,smid,pos,v);
        else
                Modify_sgt(rch,smid+,r,pos,v);
        sgt[now]=min(sgt[lch],sgt[rch]);
}
int Query_sgt(int now,int l,int r,int x,int y)
{
        if (l==x && r==y)
                return sgt[now];
        if (y<=smid)
                return Query_sgt(lch,l,smid,x,y);
        else if (smid<x)
                return Query_sgt(rch,smid+,r,x,y);
        else
                return min(Query_sgt(lch,l,smid,x,smid),Query_sgt(rch,smid+,r,smid+,y));
}
list<int> que;

int main()
{
        freopen("input.txt","r",stdin);
        int n,m,t,x,y,z;
        scanf("%d%d%d\n",&n,&m,&t);
        char ch;
        int level=;
        int mxvvv=;
        memset(sgt,INF,sizeof(sgt));
        a[]=;
        for (int i=;i<=n;i++)
        {
                scanf("%c\n",&ch);
                if (ch=='A')
                        a[i]=;
                else
                        a[i]=-;
        }
        for (int i=;i<=n;i++)
                a[i]+=a[i-];
        for (int i=;i<=n;i++)
                level=min(a[i],level);
        for (int i=;i<=n;i++)
                mxvvv=max(a[i],mxvvv);
        level=-level;
        mxvvv+=level;
        for (int i=;i<=n;i++)
                a[i]+=level;
        dp[]=;
        lst[a[]].push_back();
        Modify_sgt(,,mxvvv,a[],dp[lst[a[]].front()]);
        que.push_back();
        for (int i=;i<=n;i++)
        {
                dp[i]=Query_sgt(,,mxvvv,max(,a[i]-t),min(mxvvv,a[i]+t))+;
                if (que.size() && a[que.front()]-a[que.front()-]==a[i]-a[i-])
                        dp[i]=min(dp[i],dp[que.front()-]+);
                while (lst[a[i]].size() && dp[lst[a[i]].back()]>=dp[i])
                        lst[a[i]].pop_back();
                lst[a[i]].push_back(i);
                Modify_sgt(,,mxvvv,a[i],dp[lst[a[i]].front()]);
                if (i>=m)
                {
                        if (lst[a[i-m]].size() && lst[a[i-m]].front()==i-m)
                        {
                                lst[a[i-m]].pop_front();
                                if (lst[a[i-m]].size())
                                        Modify_sgt(,,mxvvv,a[i-m],dp[lst[a[i-m]].front()]);
                                else
                                        Modify_sgt(,,mxvvv,a[i-m],INF);
                        }
                }
                if (i== || a[i]-a[i-]==a[i+]-a[i])
                {
                        while (!que.empty() && dp[que.back()-]>=dp[i])
                                que.pop_back();
                        que.push_back(i+);
                }else
                {
                        while (!que.empty())
                                que.pop_back();
                        que.push_back(i+);
                }
                if (i-que.front()+>=m)
                        que.pop_front();
        }
        printf("%d\n",dp[n]);
}