线段树

hdu 4366 Successor http://acm.hdu.edu.cn/showproblem.php?pid=4366　　　　　　2012-12-9

题意：给你一棵树，每个节点有能力值(abi)和忠诚值(val)，询问一个节点，它的后代中能力值大于它，忠诚值最大的节点

遍历树，树状结构转换为线性结构，区间求max(val),abi>k，类似离线算法，先按abi排序，一个一个插入

#include <cstdio>
#include <vector>
#include <cstring>
#include <algorithm>
#include <map>
using namespace std;

const int N=50010<<2;
int tl[N],tr[N],val[N],abi[N],mv[N],p[N];
int ans[N];
vector<int> g[N];
map<int,int> mp;
void dfs(int u,int &d)
{
    tl[u]=d++;
    for(int i=0;i<(int)g[u].size();i++)
        dfs(g[u][i],d);
    tr[u]=d;
}
bool cmp(int a,int b)
{
    return abi[a]>abi[b] || (abi[a]==abi[b] && a<b);
}
void update(int p,int x,int rt,int l,int r)
{
    if(l==r) {mv[rt]=x; return;}
    int m=(l+r)>>1,lch=rt<<1,rch=lch|1;
    if(p<=m) update(p,x,lch,l,m);
    else update(p,x,rch,m+1,r);
    mv[rt]=max(mv[lch],mv[rch]);
}
int query(int ql,int qr,int rt,int l,int r)
{
    if(ql<=l && r<=qr) return mv[rt];
    int m=(l+r)>>1,lch=rt<<1,rch=lch|1;
    int ans=-1;
    if(ql<=m) ans=max(ans,query(ql,qr,lch,l,m));
    if(qr>m) ans=max(ans,query(ql,qr,rch,m+1,r));
    return ans;
}
int main()
{
    int T;
    scanf("%d",&T);
    while(T--)
    {
        int n,m;
        mp.clear(); mp[-1]=-1;
        scanf("%d%d",&n,&m);
        for(int i=0;i<n;i++) g[i].clear();
        for(int i=1;i<n;i++)
        {
            int fa;
            scanf("%d%d%d",&fa,&val[i],&abi[i]);
            g[fa].push_back(i);
            mp[val[i]]=i;
            p[i]=i;
        }
        int d=0;
        dfs(0,d);
        sort(p+1,p+n,cmp);
        memset(mv,-1,sizeof(mv));
        for(int i=1;i<n;i++)
        {
            ans[p[i]]=query(tl[p[i]],tr[p[i]]-1,1,1,n-1);
            update(tl[p[i]],val[p[i]],1,1,n-1);
        }
        ans[0]=-1;
        for(int i=0;i<m;i++)
        {
            int q;
            scanf("%d",&q);
            printf("%d\n",mp[ans[q]]);
        }
    }
    return 0;
}

 

 

hdu 4027 Can you answer these queries?  http://acm.hdu.edu.cn/showproblem.php?pid=4027

题意：区间开根号，区间求和

线段树中整段区间的值都为1时，就不用更新了，否则暴力更新叶子节点，一个数开几次根号就会变为1

#include <cstdio>
#include <cmath>
using namespace std;

typedef long long LL;
const int N=100010<<4;
#define lch (rt<<1)
#define rch (rt<<1|1)
LL sum[N];
void build(int rt,int l,int r)
{
    if(l==r)
    {
        scanf("%I64d",&sum[rt]);
        return;
    }
    int m=(l+r)/2;
    build(lch,l,m);
    build(rch,m+1,r);
    sum[rt]=sum[lch]+sum[rch];
}
void update(int ql,int qr,int rt,int l,int r)
{
    if(sum[rt]==r-l+1) return;
    if(l==r)
    {
        sum[rt]=(LL)(sqrt(sum[rt])+1e-8);
        return;
    }
    int m=(l+r)/2;
    if(ql<=m) update(ql,qr,lch,l,m);
    if(qr>m) update(ql,qr,rch,m+1,r);
    sum[rt]=sum[lch]+sum[rch];
}
LL query(int ql,int qr,int rt,int l,int r)
{
    if(ql<=l && r<=qr) return sum[rt];
    int m=(l+r)/2;
    LL ans=0;
    if(ql<=m) ans+=query(ql,qr,lch,l,m);
    if(qr>m) ans+=query(ql,qr,rch,m+1,r);
    return ans;
}
int main()
{
    int n,C=0;
    while(~scanf("%d",&n))
    {
        printf("Case #%d:\n",++C);
        build(1,1,n);
        int m;
        scanf("%d",&m);
        for(int i=0;i<m;i++)
        {
            int t,x,y;
            scanf("%d%d%d",&t,&x,&y);
            if(x>y) {int t=x; x=y; y=t;}
            if(t==0) update(x,y,1,1,n);
            else printf("%I64d\n",query(x,y,1,1,n));
        }
        printf("\n");
    }
    return 0;
}

 

hdu 3911 Black And White  http://acm.hdu.edu.cn/showproblem.php?pid=4027

题意：01串，操作区间取反，求区间最长的连续1的长度

区间合并，s0最长的连续0的长度，ls0左端点开始最长连续0的长度，rs0一直右端点最长连续0的长度，s1，ls1，rs1类似，def延迟标记

#include <cstdio>
#include <algorithm>
using namespace std;

const int N=100010<<2;
int s0[N],ls0[N],rs0[N],s1[N],ls1[N],rs1[N],def[N];
void pushup(int rt,int l,int r)
{
    int m=(l+r)>>1,lch=rt<<1,rch=lch|1;

    s0[rt]=max(max(s0[lch],s0[rch]),rs0[lch]+ls0[rch]);
    ls0[rt]=(ls0[lch]==m-l+1?ls0[lch]+ls0[rch]:ls0[lch]);
    rs0[rt]=(rs0[rch]==r-m?rs0[rch]+rs0[lch]:rs0[rch]);

    s1[rt]=max(max(s1[lch],s1[rch]),rs1[lch]+ls1[rch]);
    ls1[rt]=(ls1[lch]==m-l+1?ls1[lch]+ls1[rch]:ls1[lch]);
    rs1[rt]=(rs1[rch]==r-m?rs1[rch]+rs1[lch]:rs1[rch]);
}
void build(int rt,int l,int r)
{
    def[rt]=0;
    if(l==r)
    {
        int x;
        scanf("%d",&x);
        if(x==0)
        {
            s0[rt]=ls0[rt]=rs0[rt]=1;
            s1[rt]=ls1[rt]=rs1[rt]=0;
        }
        else
        {
            s0[rt]=ls0[rt]=rs0[rt]=0;
            s1[rt]=ls1[rt]=rs1[rt]=1;
        }
        return;
    }
    int m=(l+r)>>1,lch=rt<<1,rch=lch|1;
    build(lch,l,m);
    build(rch,m+1,r);
    pushup(rt,l,r);
}
void pushdown(int rt)
{
    if(def[rt]==0) return;
    int lch=rt<<1,rch=lch|1;

    swap(s0[lch],s1[lch]);
    swap(ls0[lch],ls1[lch]);
    swap(rs0[lch],rs1[lch]);
    def[lch]^=1;

    swap(s0[rch],s1[rch]);
    swap(ls0[rch],ls1[rch]);
    swap(rs0[rch],rs1[rch]);
    def[rch]^=1;

    def[rt]=0;
}
void update(int ql,int qr,int rt,int l,int r)
{
    if(ql<=l && r<=qr)
    {
        swap(s0[rt],s1[rt]);
        swap(ls0[rt],ls1[rt]);
        swap(rs0[rt],rs1[rt]);
        def[rt]^=1;
        return;
    }
    pushdown(rt);
    int m=(l+r)>>1,lch=rt<<1,rch=lch|1;
    if(ql<=m) update(ql,qr,lch,l,m);
    if(qr>m) update(ql,qr,rch,m+1,r);
    pushup(rt,l,r);
}
int query(int ql,int qr,int rt,int l,int r)
{
    if(ql<=l && r<=qr) return s1[rt];
    pushdown(rt);
    int m=(l+r)>>1,lch=rt<<1,rch=lch|1;
    int ans=0;
    if(ql<=m) ans=max(ans,query(ql,qr,lch,l,m));
    if(qr>m) ans=max(ans,query(ql,qr,rch,m+1,r));
    if(ql<=m && m<qr) ans=max(ans,min(rs1[lch],m-ql+1)+min(ls1[rch],qr-m));
    return ans;
}
int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        build(1,1,n);
        int m;
        scanf("%d",&m);
        for(int i=0;i<m;i++)
        {
            int x,l,r;
            scanf("%d%d%d",&x,&l,&r);
            if(x==1) update(l,r,1,1,n);
            else printf("%d\n",query(l,r,1,1,n));
        }
    }
    return 0;
}