[GDKOI2015] 看门狗 线段树优化dp

Description

Input

输入样例第一行包含两个数： n、 m， n 表示有 n 个房间以及 n 个需要被干扰的设备。
接下来有 m 行， 描述可干扰关系。 每一行两个数 a、 b 表示在房间 a 设置干扰装置可以干扰到设备 b。
接下来一行 n 个数，第 i 个数表示干扰第 i 个设备能得到的报酬。
接下来一行 n 个数，第 i 个数表示在第 i 个房间设置装置能得到的经验值。

Output

一个数表示皮尔斯能得到的最大的金钱经验和。

Sample Input

3 3
1 1
1 3
3 2
1 2 3
1 2 3

Sample Output

7

Data Constraint

对于 30%数据： 1 ≤ n ≤ 100
对于 100%数据： 1 ≤ n ≤ 100000， 0 ≤ m ≤ 500000, 0 ≤ Pi, Vi ≤ 1000

分析：很显然可以dp，设i,j为一排选到第i个，另一排选到第j个的答案，有
f[i][j]=max(f[i-1][j],f[i][j],f[i-1][j-1]+v[i]+p[j]) 最后这个要在右边情况下才可以转移

考虑到边比较少，枚举i，多数情况下都是
f[i][j]=max(f[i-1][j],f[i][j-1])
如果滚动一下i，那么相当于把x~(y-1)同时可以用f[i][x]改掉就可以了，可以用线段树优化。

代码：

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>

const int maxn=1e5+7;
const int maxe=500005;

using namespace std;

struct node{
    int x,y;
}e[maxe];

int n,m,i,j,k,c;
int v[maxn],w[maxn],t[maxn*10];

bool cmp(node x,node y)
{
    if (x.x==y.x) return x.y>y.y;
    return x.x<y.x;
}

int getmax(int p,int l,int r,int x,int y)
{
    if ((l==x) && (r==y))
    {
        return t[p];    
    }
    int mid=(l+r)/2;
    if (y<=mid) return getmax(p*2,l,mid,x,y);
    else
    {
        if (x>mid) return getmax(p*2+1,mid+1,r,x,y);
        else return max(getmax(p*2,l,mid,x,mid),getmax(p*2+1,mid+1,r,mid+1,y));
    }
}

void change(int p,int l,int r,int x,int k)
{
    if (l==r)
    {
        t[p]=max(t[p],k);
        return;
    }
    int mid=(l+r)/2;
    if (x<=mid) change(p*2,l,mid,x,k);
           else change(p*2+1,mid+1,r,x,k);
    t[p]=max(t[p*2],t[p*2+1]);       
}

int main()
{
    scanf("%d%d",&n,&m);
    for (i=1;i<=m;i++)
    {
        scanf("%d%d",&e[i].x,&e[i].y);
    }   
    for (i=1;i<=n;i++) scanf("%d",&v[i]);
    for (i=1;i<=n;i++) scanf("%d",&w[i]);

    sort(e+1,e+m+1,cmp);
    k=1;
    for (i=1;i<=n;i++)
    {

        while (e[k].x==i)
        {
            j=e[k].y;
            c=getmax(1,0,n,0,j-1);
            change(1,0,n,j,c+v[j]+w[i]);
            k++;
        }
    }
    printf("%d",t[1]);
}