码迷,mamicode.com
首页 > 其他好文 > 详细

HDU.6155.Subsequence Count(线段树 矩阵)

时间:2018-02-10 11:22:04      阅读:148      评论:0      收藏:0      [点我收藏+]

标签:turn   ret   线段树   class   getc   blog   size   位置   ons   

题目链接
首先考虑询问[1,n]怎么做
设 f[i][0/1]表示[1,i]以0/1结尾的不同子序列个数
\(if(A[i]) f[i][1] = f[i-1][0] + f[i-1][1] + 1 , f[i][0] = f[i-1][0]\)
\(\ \ if(!A[i]) f[i][0] = f[i-1][0] + f[i-1][1] + 1 , f[i][1] = f[i-1][1]\)
很整齐,我们来写成矩阵的样子: \(f[i,0]\ f[i,1]\ 1=f[i-1,0]\ f[i-1,1]\ 1\ *\ M\)
算了不写了。。这儿写比较麻烦 不妨看
最后可得原序列为0/1的位置 分别为两个仅互换1,2行 1,2列的矩阵M[0/1] [l,r]的dp值就是区间矩阵乘积的结果 ans = f[i,0]+fi,1

/*
注: Modify时标记不要改两次-- 
*/
#include<cstdio>
#include<cctype>
#include<cstring>
#include<algorithm>
#define gc() getchar()
const int N=1e5+5,mod=1e9+7;
const int M[2][3][3]={{1,0,0,1,1,0,1,0,1},{1,1,0,0,1,0,0,1,1}};//0 1

int n,m,A[N];
char s[N];
struct Matrix
{
    int mat[3][3];
//  Matrix() {memset(mat,0,sizeof mat);}
    Matrix operator *(const Matrix &a)const
    {
        Matrix res;
        for(int i=0; i<3; ++i)
            for(int j=0; j<3; ++j)
            {
                res.mat[i][j]=0;
                for(int k=0; k<3; ++k)
                    res.mat[i][j] += 1LL*mat[i][k]*a.mat[k][j]%mod,
                    res.mat[i][j] >= mod ? res.mat[i][j]-=mod : 0;//还是喜欢紧凑些-- 
            }
        return res;
    }
};
struct Seg_Tree
{
    Matrix mat[N<<2];
    bool tag[N<<2];
    inline void PushUp(int rt)
    {
        mat[rt]=mat[rt<<1]*mat[rt<<1|1];
    }
    inline void Init(int rt,int opt)
    {
        for(int i=0; i<3; ++i)
            for(int j=0; j<3; ++j)
                mat[rt].mat[i][j]=M[opt][i][j];
//      if(opt)
//          mat[rt].mat[0][0]=mat[rt].mat[0][1]=mat[rt].mat[1][1]=mat[rt].mat[2][1]=mat[rt].mat[2][2]=1;
//      else
//          mat[rt].mat[0][0]=mat[rt].mat[1][0]=mat[rt].mat[1][1]=mat[rt].mat[2][0]=mat[rt].mat[2][2]=1;
    }
    void Swap(int rt)
    {
//      for(int i=0; i<3; ++i)//swap(Line 1,Line 2)
//          std::swap(mat[rt].mat[0][i], mat[rt].mat[1][i]);
//      for(int i=0; i<3; ++i)//swap(Column 1,Column 2)
//          std::swap(mat[rt].mat[i][0], mat[rt].mat[i][1]);
        std::swap(mat[rt].mat[0][0],mat[rt].mat[1][0]),
        std::swap(mat[rt].mat[0][1],mat[rt].mat[1][1]),
        std::swap(mat[rt].mat[0][2],mat[rt].mat[1][2]),
        std::swap(mat[rt].mat[0][0],mat[rt].mat[0][1]),
        std::swap(mat[rt].mat[1][0],mat[rt].mat[1][1]),
        std::swap(mat[rt].mat[2][0],mat[rt].mat[2][1]),
        tag[rt]^=1;
    }
    void PushDown(int rt)
    {
        Swap(rt<<1), Swap(rt<<1|1);
        tag[rt]=0;
    }
    void Build(int l,int r,int rt)
    {
        tag[rt]=0;
        if(l==r)
            if(A[l]) Init(rt,1);
            else Init(rt,0);
        else
        {
            int m=l+r>>1;
            Build(l,m,rt<<1), Build(m+1,r,rt<<1|1);
            PushUp(rt);
        }
    }
    void Modify_Flip(int l,int r,int rt,int L,int R)
    {
        if(L<=l && r<=R) Swap(rt);//, tag[rt]^=1;//WA
        else
        {
            if(tag[rt]) PushDown(rt);
            int m=l+r>>1;
            if(L<=m) Modify_Flip(l,m,rt<<1,L,R);
            if(m<R) Modify_Flip(m+1,r,rt<<1|1,L,R);
            PushUp(rt);
        }
    }
    Matrix Query(int l,int r,int rt,int L,int R)
    {
        if(L<=l && r<=R) return mat[rt];
        if(tag[rt]) PushDown(rt);
        int m=l+r>>1;
        if(L<=m)
            if(m<R) return Query(l,m,rt<<1,L,R)*Query(m+1,r,rt<<1|1,L,R);
            else return Query(l,m,rt<<1,L,R);
        else return Query(m+1,r,rt<<1|1,L,R);
    }
}t;
inline int read()
{
    int now=0,f=1;register char c=gc();
    for(;!isdigit(c);c=gc()) if(c=='-') f=-1;
    for(;isdigit(c);now=now*10+c-'0',c=gc());
    return now*f;
}

int main()
{
#ifndef ONLINE_JUDGE
//  freopen("1.in","r",stdin);
#endif

    int T=read();
    while(T--)
    {
        n=read(),m=read();
        scanf("%s",s+1);
        for(int i=1; i<=n; ++i) A[i]=s[i]-'0';
        t.Build(1,n,1);
        int opt,l,r; Matrix res;
        while(m--)
        {
            opt=read(),l=read(),r=read();
            if(opt==1) t.Modify_Flip(1,n,1,l,r);
            else
                res = t.Query(1,n,1,l,r),
                printf("%d\n",(res.mat[2][0]+res.mat[2][1])%mod);
        }
    }
    return 0;
}

HDU.6155.Subsequence Count(线段树 矩阵)

标签:turn   ret   线段树   class   getc   blog   size   位置   ons   

原文地址:https://www.cnblogs.com/SovietPower/p/8438012.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!