codeforces1252K Addition Robot

https://codeforces.com/contest/1252/problem/K

艹了比赛的时候区间翻转的tag写错了,翻转应该^1的不应该直接=1的,WA到结束也没找出错,我好菜啊

对于A,B的值维护一个矩阵,A操作和B操作按顺序执行就是A,B的值按照矩阵连乘来变

区间AB翻转操作就是把矩阵交换一下,用个tag^=1来记录一下

求区间的值就把这段区间的矩阵连乘的结果查询出来再跟A,B乘一下就得到最后的值了

#include<bits/stdc++.h>
#define maxl 100010
using namespace std;

const int mod=1000000007;

int n,q;
char s[maxl];
struct matrix
{
  long long a[3][3];
  matrix()
  {
    for(int i=0;i<=2;i++)
      for(int j=0;j<=2;j++)
	a[i][j]=0;
  }
  matrix operator * (const matrix &b) const
  {
    matrix t;
    for(int i=1;i<=2;i++)
      for(int j=1;j<=2;j++)
	for(int k=1;k<=2;k++)
	  t.a[i][k]=(t.a[i][k]+1ll*a[i][j]*b.a[j][k]%mod)%mod;
    return t;
  }
};
struct node
{
  int l,r,tag;
  matrix d;
}tree[maxl*4];

inline void chge(matrix & d)
{
  matrix t;
  for(int i=1;i<=2;i++)
    for(int j=1;j<=2;j++)
      t.a[i][j]=d.a[i][j];
  d.a[1][1]=t.a[2][2];d.a[1][2]=t.a[2][1];
  d.a[2][1]=t.a[1][2];d.a[2][2]=t.a[1][1];
}

inline void gank(int k)
{
  if(tree[k].l==tree[k].r)
    return;
  if(tree[k].tag)
    {
      tree[k<<1].tag^=1;
      chge(tree[k<<1].d);
      tree[k<<1|1].tag^=1;
      chge(tree[k<<1|1].d);
      tree[k].tag=0;
    }
}

inline void push_up(int k)
{
  tree[k].d=tree[k<<1].d * tree[k<<1|1].d;
}

inline void build(int k,int l,int r)
{
  tree[k].l=l;tree[k].r=r;tree[k].tag=0;
  if(l==r)
    {
      if(s[l]=='A')
	{
	  tree[k].d.a[1][1]=1;tree[k].d.a[1][2]=0;
	  tree[k].d.a[2][1]=1;tree[k].d.a[2][2]=1;
	}
      else
	{
	  tree[k].d.a[1][1]=1;tree[k].d.a[1][2]=1;
	  tree[k].d.a[2][1]=0;tree[k].d.a[2][2]=1;
	}
      return;
    }
  int mid=(l+r)>>1;
  build(k<<1,l,mid);
  build(k<<1|1,mid+1,r);
  push_up(k);
}

inline void prework()
{
  scanf("%d%d",&n,&q);
  scanf("%s",s+1);
  build(1,1,n);
}

inline void upd(int k,int l,int r)
{
  gank(k);
  if(tree[k].l==l && tree[k].r==r)
    {
      chge(tree[k].d);
      tree[k].tag^=1;
      return;
    }
  int mid=(tree[k].l+tree[k].r)>>1;
  if(r<=mid)
    upd(k<<1,l,r);
  else if(l>mid)
    upd(k<<1|1,l,r);
  else
    {
      upd(k<<1,l,mid);
      upd(k<<1|1,mid+1,r);
    }
  push_up(k);
}

inline matrix qry(int k,int l,int r)
{
  gank(k);
  if(tree[k].l==l && tree[k].r==r)
    return tree[k].d;
  matrix t;
  int mid=(tree[k].l+tree[k].r)>>1;
  if(r<=mid)
    t=qry(k<<1,l,r);
  else if(l>mid)
    t=qry(k<<1|1,l,r);
  else
    {
      t=qry(k<<1,l,mid);
      t=t*qry(k<<1|1,mid+1,r);
    }
  push_up(k);
  return t;
}

inline void mainwork()
{
  int op,l,r,a,b;
  matrix t;long long aa,bb;
  for(int i=1;i<=q;i++)
    {
      scanf("%d",&op);
      if(op==1)
		{
		  scanf("%d%d",&l,&r);
		  upd(1,l,r);
		}
      else
		{
		  scanf("%d%d%d%d",&l,&r,&a,&b);
		  t=qry(1,l,r);
		  aa=(1ll*t.a[1][1]*a%mod+1ll*t.a[2][1]*b%mod)%mod;
		  bb=(1ll*t.a[1][2]*a%mod+1ll*t.a[2][2]*b%mod)%mod;
		  printf("%lld %lld\n",aa,bb);
		}
    }
}

inline void print(){}

int main()
{
  //freopen("K.in","r",stdin);
  prework();
  mainwork();
  print();
  return 0;
}

 

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值