loj#2269. 「SDOI2017」切树游戏

还是loj的机子快啊。。。

普通的DP不难想到,设F[i][zt]为带上根玩出zt的方案数,G[i][zt]为子树中的方案数,后面是可以用FWT优化的

主要是复习了下动态DP

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;
const int _=1e2;
const int maxn=3e4+_;
const int fbin=maxn<<2;
const int maxm=128+2;
const int mod=1e4+7;
inline int ad(int x,int y){return (x>=mod-y)?(x-mod+y):x+y;}
inline int re(int x,int y){return (x<y)?(x-y+mod):x-y;}
inline int read()
{
    int x=0; char ch=getchar();
    while(!isdigit(ch))ch=getchar();
    while(isdigit(ch)){x=x*10+ch-‘0‘;ch=getchar();}
    return x;
}
inline void write(int d)
{
    if(d>=10)write(d/10);
    putchar(d%10+‘0‘);
}
int m;

struct poly
{
    int a[maxm];
    poly(){}
    void FWT(int op)
    {
        for(int i=1;i<m;i<<=1)
            for(int j=0;j<m;j+=(i<<1))
                for(int k=0;k<i;k++)
                {
                    int a1=a[j+k],a2=a[j+k+i];
                    a[j+k]=ad(a1,a2);
                    a[j+k+i]=re(a1,a2);
                    if(op==-1)
                    {
                        a[j+k]=a[j+k]*(mod/2+1)%mod;
                        a[j+k+i]=a[j+k+i]*(mod/2+1)%mod;
                    }
                }
    }
    poly(int x){memset(a,0,sizeof(a));a[x]=1;FWT(1);}

    friend poly operator +(poly u,poly v){for(int i=0;i<m;i++)u.a[i]=ad(u.a[i],v.a[i]); return u;}
    friend poly operator -(poly u,poly v){for(int i=0;i<m;i++)u.a[i]=re(u.a[i],v.a[i]); return u;}
    friend poly operator *(poly u,poly v){for(int i=0;i<m;i++)u.a[i]=u.a[i]*v.a[i]%mod; return u;}
}o;
struct Matrix
{
    poly a,b,c,d;
    Matrix(){}
    Matrix(poly f,poly g){a=b=c=f;d=f+g;}
    Matrix(poly A,poly B,poly C,poly D){a=A,b=B,c=C,d=D;}
    friend Matrix operator +(Matrix u,Matrix v){return Matrix(u.a*v.a,u.a*v.b+u.b,u.c*v.a+v.c,u.c*v.b+u.d+v.d);}
};

//----------------------------------------def------------------------------------------------------------

namespace FASEG//每个点的每个孩子的F+1
{
    struct trnode
    {
        int lc,rc;poly p;
    }tr[10*maxn];int trlen,rt[maxn];
    void update(int now)
    {
        int lc=tr[now].lc,rc=tr[now].rc;
        if(lc&&rc)tr[now].p=tr[lc].p*tr[rc].p;
        else if(lc)tr[now].p=tr[lc].p;
        else if(rc)tr[now].p=tr[rc].p;
    }
    void insert(int &now,int l,int r,int p,poly u)
    {
        if(now==0)now=++trlen;
        if(l==r){tr[now].p=u;return ;}
        int mid=(l+r)/2;
        if(p<=mid)insert(tr[now].lc,l,mid,p,u);
        else insert(tr[now].rc,mid+1,r,p,u);
        update(now);
    }
}
namespace LINESEG//第i个位置放的是i的重儿子转移到i的转移矩阵
{
    #define lc (now<<1)
    #define rc (now<<1|1)
    #define mid ((ql+qr)/2)

    Matrix tr[fbin];
    void update(int now){tr[now]=tr[rc]+tr[lc];}
    void change(int now,int ql,int qr,int p,Matrix m)
    {
        if(ql==qr){tr[now]=m;return ;}
        if(p<=mid)change(lc,ql,mid,p,m);
        else change(rc,mid+1,qr,p,m);
        update(now);
    }
    Matrix getmatrix(int now,int ql,int qr,int l,int r)
    {
        if(ql==l&&qr==r)return tr[now];
             if(r<=mid)  return getmatrix(lc,ql,mid,l,r);
        else if(mid+1<=l)return getmatrix(rc,mid+1,qr,l,r);
        else return getmatrix(rc,mid+1,qr,mid+1,r)+getmatrix(lc,ql,mid,l,mid);
    }

    #undef lc
    #undef rc
    #undef mid
}

//---------------------------------------data struct----------------------------------------------------

int n;
struct node
{
    int x,y,next;
}a[2*maxn];int len,last[maxn];
void ins(int x,int y)
{
    len++;
    a[len].x=x;a[len].y=y;
    a[len].next=last[x];last[x]=len;
}
int fa[maxn],son[maxn],tot[maxn],dep[maxn];
void pre_tree_node(int x)
{
    tot[x]=1;
    for(int k=last[x];k;k=a[k].next)
    {
        int y=a[k].y;
        if(y!=fa[x])
        {
            fa[y]=x;
            dep[y]=dep[x]+1;
            pre_tree_node(y);
            if(son[x]==0||tot[son[x]]<tot[y])son[x]=y;
            tot[x]+=tot[y];
        }
    }
}
int z,ys[maxn],top[maxn],bot[maxn];
void pre_tree_edge(int x,int tp)
{
    ys[x]=++z;top[x]=tp;
    if(son[x]!=0)pre_tree_edge(son[x],tp),bot[x]=bot[son[x]];
    else bot[x]=x;
    for(int k=last[x];k;k=a[k].next)
    {
        int y=a[k].y;
        if(y!=fa[x]&&y!=son[x])
            pre_tree_edge(y,y);
    }
}

//-------------------------------------------cop----------------------------------------------------

int w[maxn],cnt[maxn],wch[maxn];//有多少轻孩子,爸爸的第几个轻孩子
poly F[maxn],G[maxn],g[maxn];//包括x的方案数,总方案数,只有轻儿子的总方案数
void treeDP(int x)
{
    F[x]=poly(w[x]);
    cnt[x]=1;
    for(int k=last[x];k;k=a[k].next)
    {
        int y=a[k].y;
        if(fa[x]!=y)
        {
            treeDP(y);
            F[x]=F[x]*(F[y]+o);
            G[x]=G[x]+G[y];
            if(son[x]!=y)
            {
                wch[y]=++cnt[x];
                g[x]=g[x]+G[y];
            }
        }
    }
    G[x]=G[x]+F[x];

    using namespace FASEG;
    insert(rt[x],1,cnt[x],1,poly(w[x]));
    for(int k=last[x];k;k=a[k].next)
    {
        int y=a[k].y;
        if(fa[x]!=y&&son[x]!=y)
            insert(rt[x],1,cnt[x],wch[y],F[y]+o);
    }
}

//-------------------------------------------pre------------------------------------------------

void query(int zt)
{
    poly u=G[1];u.FWT(-1);
    write(u.a[zt]),putchar(‘\n‘);
}
Matrix ma;
void change(int x,int zt)
{
    w[x]=zt;
    using namespace FASEG;
    insert(rt[x],1,cnt[x],1,poly(w[x]));

    int tx=top[x];
    while(x!=0)
    {
        LINESEG::change(1,1,n,ys[x],Matrix(tr[rt[x]].p,g[x]));
        if(fa[tx]!=0)
        {
            ma=LINESEG::getmatrix(1,1,n,ys[tx],ys[bot[tx]]);
            insert(rt[fa[tx]],1,cnt[fa[tx]],wch[tx],ma.c+o);
            g[fa[tx]]=g[fa[tx]]-G[tx];
            G[tx]=ma.d;
            g[fa[tx]]=g[fa[tx]]+G[tx];
        }
        else G[tx]=LINESEG::getmatrix(1,1,n,ys[tx],ys[bot[tx]]).d;
        x=fa[tx];tx=top[x];
    }
}

char ss[10];
int main()
{
    freopen("xor.in","r",stdin);
    freopen("xor.out","w",stdout);
    int x,y;
    n=read(),m=read();
    for(int i=1;i<=n;i++)w[i]=read();
    for(int i=1;i<n;i++)
    {
        x=read(),y=read();
        ins(x,y),ins(y,x);
    }
    pre_tree_node(1);
    z=0,pre_tree_edge(1,1);
    for(int i=0;i<m;i++)o.a[i]=1;
    treeDP(1);
    for(int i=1;i<=n;i++)
        LINESEG::change(1,1,n,ys[i],Matrix(FASEG::tr[FASEG::rt[i]].p,g[i]));

    int Q;
    Q=read();
    while(Q--)
    {
        char ch=getchar();
        while(ch!=‘C‘&&ch!=‘Q‘)ch=getchar();
        bool bk=ch==‘C‘;
        while(ch!=‘e‘&&ch!=‘y‘)ch=getchar();
        if(!bk)
            x=read(),query(x);
        else
        {
            x=read(),y=read(),change(x,y);

        /*    poly u=LINESEG::getmatrix(1,1,n,ys[1],ys[bot[1]]).d;
            u.FWT(-1);
            for(int i=0;i<=3;i++)
                printf("%d ",u.a[i]);
            puts(""); */
        }
    }

    return 0;
}

原文地址:https://www.cnblogs.com/AKCqhzdy/p/10745412.html

时间: 2024-09-29 17:33:38

loj#2269. 「SDOI2017」切树游戏的相关文章

loj#2002. 「SDOI2017」序列计数(dp 矩阵乘法)

题意 题目链接 Sol 质数的限制并没有什么卵用,直接容斥一下:答案 = 忽略质数总的方案 - 没有质数的方案 那么直接dp,设\(f[i][j]\)表示到第i个位置,当前和为j的方案数 \(f[i + 1][(j + k) \% p] += f[i][j]\) 矩乘优化一下. #include<bits/stdc++.h> #define LL long long using namespace std; const int MAXN = 2e7 + 10, mod = 20170408,

loj2004. 「SDOI2017」硬币游戏

2004. 「SDOI2017」硬币游戏 周末同学们非常无聊,有人提议,咱们扔硬币玩吧,谁扔的硬币正面次数多谁胜利. 大家纷纷觉得这个游戏非常符合同学们的特色,但只是扔硬币实在是太单调了. 同学们觉得要加强趣味性,所以要找一个同学扔很多很多次硬币,其他同学记录下正反面情况. 用 $ \texttt{H} $ 表示正面朝上, 用 $ \texttt{T} $ 表示反面朝上,扔很多次硬币后,会得到一个硬币序列.比如 $ \texttt{HTT} $ 表示第一次正面朝上,后两次反面朝上. 但扔到什么时

Loj #2541「PKUWC2018」猎人杀

Loj #2541. 「PKUWC2018」猎人杀 题目链接 好巧妙的题! 游戏过程中,概率的分母一直在变化,所以就非常的不可做. 所以我们将问题转化一下:我们可以重复选择相同的猎人,只不过在一个猎人被选择了过后我们就给他打上标记,再次选择他的时候就无效.这样与原问题是等价的. 证明: 设\(sum=\sum_iw_i,kill=\sum_{i被杀死了}w_i\). 攻击到未被杀死的猎人\(i\)的概率为\(P\). 则根据题意\(P=\frac{w_i}{sum-kill}\). 问题转化后:

Loj #2542. 「PKUWC2018」随机游走

Loj #2542. 「PKUWC2018」随机游走 题目描述 给定一棵 \(n\) 个结点的树,你从点 \(x\) 出发,每次等概率随机选择一条与所在点相邻的边走过去. 有 \(Q\) 次询问,每次询问给定一个集合 \(S\),求如果从 \(x\) 出发一直随机游走,直到点集 \(S\) 中所有点都至少经过一次的话,期望游走几步. 特别地,点 \(x\)(即起点)视为一开始就被经过了一次. 答案对 $998244353 $ 取模. 输入格式 第一行三个正整数 \(n,Q,x\). 接下来 \(

Loj #3111. 「SDOI2019」染色

Loj #3111. 「SDOI2019」染色 题目描述 给定 \(2 \times n\) 的格点图.其中一些结点有着已知的颜色,其余的结点还没有被染色.一个合法的染色方案不允许相邻结点有相同的染色. 现在一共有 \(c\) 种不同的颜色,依次记为 \(1\) 到 \(c\).请问有多少对未染色结点的合法染色方案? 输入格式 第一行有两个整数 \(n\) 和 \(c\),分别描述了格点图的大小和总的颜色个数. 之后两行,每行有 \(n\) 个整数:如果是 \(0\) 则表示对应结点未被染色,否

Loj #2553. 「CTSC2018」暴力写挂

Loj #2553. 「CTSC2018」暴力写挂 题目描述 temporaryDO 是一个很菜的 OIer .在 4 月,他在省队选拔赛的考场上见到了<林克卡特树>一题,其中 \(k = 0\) 的部分分是求树 \(T\) 上的最长链.可怜的 temporaryDO 并不会做这道题,他在考场上抓猫耳挠猫腮都想不出一点思路. 这时,善良的板板出现在了空中,他的身上发出璀璨却柔和的光芒,荡漾在考场上.''题目并不难.'' 板板说.那充满磁性的声音,让 temporaryDO 全身充满了力量. 他

Loj #2554. 「CTSC2018」青蕈领主

Loj #2554. 「CTSC2018」青蕈领主 题目描述 "也许,我的生命也已经如同风中残烛了吧."小绿如是说. 小绿同学因为微积分这门课,对"连续"这一概念产生了浓厚的兴趣.小绿打算把连续的概念放到由整数构成的序列上,他定义一个长度为 \(m\) 的整数序列是连续的,当且仅当这个序列中的最大值与最小值的差,不超过\(m-1\).例如 \(\{1,3,2\}\) 是连续的,而 \(\{1,3\}\) 不是连续的. 某天,小绿的顶头上司板老大,给了小绿 \(T\)

AC日记——「SDOI2017」序列计数 LibreOJ 2002

「SDOI2017」序列计数 思路: 矩阵快速幂: 代码: #include <bits/stdc++.h> using namespace std; #define mod 20170408 #define ll long long struct MatrixType { int n,m; ll ai[105][105]; void mem(int n_,int m_) { n=n_,m=m_; for(int i=0;i<=n;i++) for(int v=0;v<=m;v++

LibreOJ #2002. 「SDOI2017」序列计数

二次联通门 : LibreOJ #2002. 「SDOI2017」序列计数 /* LibreOJ #2002. 「SDOI2017」序列计数 线性筛 + 矩阵优化dp 先构造出全部情况的矩阵 用矩阵快速幂计算答案 再构造出全不是质数的矩阵 计算出答案 前一个答案减后一个答案即可 */ #include <cstdio> #include <iostream> #include <cstring> const int BUF = 12312312; char Buf[BU