zoukankan      html  css  js  c++  java
  • UOJ388 [UNR #3] 配对树

    先考虑怎样配对最优,发现对于一条边,若其两端的子树内需要配对的点的个数都为奇数,则该边会有 (1) 的贡献,否则没有贡献,得这样为最优情况。

    那么对于一棵子树,将其内部的点标记为 (1),得到一个 (01) 串,子树父边的贡献为 (01) 串中长度为偶数且区间和为奇数的区间个数。

    可以用线段树合并来优化,线段树上每个节点维护 (v(cur,0/1,0/1)),表示 (cur) 对应的区间中前缀和为偶数或奇数,位置为偶数或奇数的位置个数,设点 (x) 的父边权值为 (val),得其对答案的贡献为:

    [large val(v(rt_x,0,0)v(rt_x,1,0)+v(rt_x,0,1)v(rt_x,1,1)) ]

    为方便计算位置个数,线段树区间设为 ([1,m+1])

    #include<bits/stdc++.h>
    #define maxn 200010
    #define maxm 20000010
    #define p 998244353
    #define mid ((l+r)>>1)
    using namespace std;
    typedef long long ll;
    template<typename T> inline void read(T &x)
    {
        x=0;char c=getchar();bool flag=false;
        while(!isdigit(c)){if(c=='-')flag=true;c=getchar();}
        while(isdigit(c)){x=(x<<1)+(x<<3)+(c^48);c=getchar();}
        if(flag)x=-x;
    }
    int n,m,tot;
    ll ans;
    int rt[maxn],ls[maxm],rs[maxm],sum[maxm],v[maxm][2][2];
    struct edge
    {
        int to,nxt;
        ll v;
    }e[maxn];
    int head[maxn],edge_cnt;
    void add(int from,int to,int val)
    {
        e[++edge_cnt]={to,head[from],val},head[from]=edge_cnt;
    }
    void pushup(int cur,int l,int r)
    {
        sum[cur]=sum[ls[cur]]+sum[rs[cur]];
        int k=sum[ls[cur]]&1;
        for(int i=0;i<=1;++i)
            for(int j=0;j<=1;++j)
                v[cur][i][j]=v[ls[cur]][i][j]+v[rs[cur]][i^k][j];
        if(!ls[cur]) v[cur][0][0]+=mid/2-(l-1)/2,v[cur][0][1]+=(mid+1)/2-l/2;
        if(!rs[cur]) v[cur][k][0]+=r/2-mid/2,v[cur][k][1]+=(r+1)/2-(mid+1)/2;
    }
    void insert(int l,int r,int pos,int &cur)
    {
        if(!cur)
        {
            cur=++tot;
            v[cur][0][0]=r/2-(l-1)/2;
            v[cur][0][1]=(r+1)/2-l/2;
        }
        if(l==r)
        {
            sum[cur]++;
            return;
        }
        if(pos<=mid) insert(l,mid,pos,ls[cur]);
        else insert(mid+1,r,pos,rs[cur]);
        pushup(cur,l,r);
    }
    int merge(int x,int y,int l,int r)
    {
        if(!x||!y) return x+y;
        int cur=++tot;
        if(l==r) sum[cur]=sum[x]+sum[y];
        else
        {
            ls[cur]=merge(ls[x],ls[y],l,mid);
            rs[cur]=merge(rs[x],rs[y],mid+1,r);
            pushup(cur,l,r);
        }
        return cur;
    }
    void dfs(int x,int fa)
    {
        for(int i=head[x];i;i=e[i].nxt)
        {
            int y=e[i].to;
            if(y==fa) continue;
            dfs(y,x);
            ans=(ans+e[i].v*((ll)v[rt[y]][0][0]*v[rt[y]][1][0]%p+(ll)v[rt[y]][0][1]*v[rt[y]][1][1]%p)%p)%p;
            rt[x]=merge(rt[x],rt[y],1,m+1);
        }
    }
    int main()
    {
        read(n),read(m);
        for(int i=1;i<n;++i)
        {
            int x,y,v;
            read(x),read(y),read(v);
            add(x,y,v),add(y,x,v);
        }
        for(int i=1,x;i<=m;++i) read(x),insert(1,m+1,i,rt[x]);
        dfs(1,0),printf("%lld",ans);
        return 0;
    }
    
  • 相关阅读:
    【cocos2d-js公文】十七、事件分发机制
    UVA它11292
    定义自己的仪表板DashBoard
    ufldl学习笔记和编程作业:Feature Extraction Using Convolution,Pooling(卷积和汇集特征提取)
    互联网和移动互联网怎么赚钱?
    android布局margin和padding差异!
    解决ORA-28000: the account is locked
    网络编程基本知识
    10个优秀的 HTML5 &amp; CSS3 下拉菜单制作教程
    Codeforces554A:Kyoya and Photobooks
  • 原文地址:https://www.cnblogs.com/lhm-/p/13887290.html
Copyright © 2011-2022 走看看