hdu 3872 Dragon Ball

题目链接:http://acm.hdu.edu.cn/showproblem.php?pid=3872

题目思路:单调队列加线段树优化,单调队列的作用是将序列分成最大值不同的若干段。

#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<string>
#include<queue>
#include<algorithm>
#include<vector>
#include<stack>
#include<list>
#include<iostream>
#include<map>
using namespace std;
#define inf 0x3f3f3f3f
#define Max 100100
int max(int a,int b)
{
    return a>b?a:b;
}
int min(int a,int b)
{
    return a<b?a:b;
}
int dp[Max],t,n,typ[Max],val[Max],rec,q[Max],lef[Max];
struct node
{
    int l,r,minx;
    int mid()
    {
        return (l+r)>>1;
    }
}T[4*Max];
void build(int l,int r,int rt)
{
    T[rt].l=l;T[rt].r=r;
    T[rt].minx=0;
    if(l==r)
    {
       // T[rt].max=val[l];
        return ;
    }

    int mid=T[rt].mid();
    build(l,mid,rt<<1);
    build(mid+1,r,rt<<1|1);

}
void modify(int l,int rt,int data)
{
    if(T[rt].l==T[rt].r)
    {
        T[rt].minx=data;
        return;
    }
    int mid=T[rt].mid();
    if(l<=mid)
    modify(l,rt<<1,data);
    else
    modify(l,rt<<1|1,data);
    T[rt].minx=min(T[rt<<1].minx,T[rt<<1|1].minx);
}
void query(int l,int r,int rt)
{
    if(T[rt].l==l&&T[rt].r==r)
    {
        rec=min(rec,T[rt].minx);
        return;
    }
    int mid=T[rt].mid();
    if(l>=mid+1)
    query(l,r,rt<<1|1);
    else if(r<=mid)
    query(l,r,rt<<1);
    else
    {
        query(l,mid,rt<<1);
        query(mid+1,r,rt<<1|1);
    }
}
int main()
{
    int i,j;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d",&n);
        for(i=1;i<=n;i++)
        {
            scanf("%d",&typ[i]);
        }
        for(i=1;i<=n;i++)
        {
            scanf("%d",&val[i]);
        }
        dp[0]=0;
        int top=0;int tail=1;
        build(0,n,1);
        q[0]=0;
        memset(lef,0,sizeof(lef));
        for(i=1;i<=n;i++)
        {
           // int maxn=val[i];
           while(tail>top+1&&val[i]>val[q[tail-1]])
           tail--;
           q[tail]=i;
           tail++;
           dp[i]=inf;
           for(j=tail-1;j>top&&q[j]>lef[typ[i]];j--)
           {
               rec=inf;
               query(max(q[j-1],lef[typ[i]]),q[j]-1,1);
             //  printf("a");
               dp[i]=min(rec+val[q[j]],dp[i]);
           }
           modify(i,1,dp[i]);
           lef[typ[i]]=i;
         //  printf("i %d dp %d\n",i,dp[i]);
        }
        printf("%d\n",dp[n]);
    }
}


你可能感兴趣的:(优化,struct,query,Build)