Codeforces Round 718C (矩阵线段树)

185 篇文章 0 订阅
18 篇文章 0 订阅

Sasha has an array of integers a1, a2, ..., an. You have to performm queries. There might be queries of two types:

  1. 1 l r x — increase all integers on the segment froml to r by valuesx;
  2. 2 l r — find , wheref(x) is the x-th Fibonacci number. As this number may be large, you only have to find it modulo109 + 7.

In this problem we define Fibonacci numbers as follows: f(1) = 1, f(2) = 1, f(x) = f(x - 1) + f(x - 2) for allx > 2.

Sasha is a very talented boy and he managed to perform all queries in five seconds. Will you be able to write the program that performs as well as Sasha?

Input

The first line of the input contains two integers n andm (1 ≤ n ≤ 100 000,1 ≤ m ≤ 100 000) — the number of elements in the array and the number of queries respectively.

The next line contains n integers a1, a2, ..., an (1 ≤ ai ≤ 109).

Then follow m lines with queries descriptions. Each of them contains integerstpi,li,ri and may bexi (1 ≤ tpi ≤ 2,1 ≤ li ≤ ri ≤ n,1 ≤ xi ≤ 109). Heretpi = 1 corresponds to the queries of the first type andtpi corresponds to the queries of the second type.

It's guaranteed that the input will contains at least one query of the second type.

Output

For each query of the second type print the answer modulo 109 + 7.

Examples
Input
5 4
1 1 2 1 1
2 1 5
1 2 4 2
2 2 4
2 1 5
Output
5
7
9
Note

Initially, array a is equal to 1, 1, 2, 1, 1.

The answer for the first query of the second type is f(1) + f(1) + f(2) + f(1) + f(1) = 1 + 1 + 1 + 1 + 1 = 5.

After the query 1 2 4 2 array a is equal to 1, 3, 4, 3, 1.

The answer for the second query of the second type is f(3) + f(4) + f(3) = 2 + 3 + 2 = 7.

The answer for the third query of the second type is f(1) + f(3) + f(4) + f(3) + f(1) = 1 + 2 + 3 + 2 + 1 = 9.



题意:让你用线段树维护一个下标序列,每次区间加或者查询以这个区间所有数为下标的菲波那切数列和。



分析:线段树中每个节点存放一个矩阵就可以了。


#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstdlib>
#include <cstdio>
#include <set>
#include <map>
#include <vector>
#include <stack>
#include <cmath>
#include <queue>
#include <cstring>
#define MOD 1000000007
#define N 200005
using namespace std;
typedef long long ll;
int T,n,m,k,num,a[N];
struct Matrix
{
    ll val[3][3];
    int x ,y;
    void init()
    {
        memset(val,0,sizeof(val));
        val[1][1] = val[2][2] = 1;
    }
    Matrix(int w,int v)
    {
        x = w,y = v;
        memset(val,0,sizeof(val));
    }
    Matrix(){memset(val,0,sizeof(val));}
    friend Matrix operator *(const Matrix a,const Matrix b)
    {
        Matrix c;
        c.x = a.x,c.y = b.y;
        for(int i = 1;i <= a.x;i++)
         for(int j = 1;j <= b.y;j++)
          for(int k = 1;k <= b.x;k++)
           c.val[i][j] = (c.val[i][j] + a.val[i][k]*b.val[k][j]) % MOD;
        return c;
    }
    friend Matrix operator +(const Matrix a,const Matrix b)
    {
        Matrix c;
        c.x = a.x,c.y = b.y;
        for(int i = 1;i <= a.x;i++)
         for(int j = 1;j <= b.y;j++)
          c.val[i][j] = (a.val[i][j] + b.val[i][j]) % MOD;
        return c;
    }
};
Matrix ksm(Matrix a,ll b)
{
    Matrix c = a;
    b--;
    while(b)
    {
        if(b & 1) c = c * a;
        a = a * a;
        b >>= 1;
    }
    return c;
}
struct Segtree
{
    int l,r;
    bool down;
    Matrix dow = Matrix(2,2);
    Matrix sta = Matrix(2,2);
}tree[4*N];
void push_up(int i)
{
    tree[i].sta = tree[2*i].sta + tree[2*i+1].sta;
}
void build(int i,int l,int r)
{
    tree[i].l = l;
    tree[i].r = r;
    tree[i].down = 0;
    tree[i].dow.init();
    if(l == r)
    {
        Matrix temp = Matrix(2,2);
        temp.val[1][1] = temp.val[1][2] = temp.val[2][1] = 1;
        if(a[l] != 1) tree[i].sta = ksm(temp,a[l] - 1);
        else tree[i].sta.init();
        return;
    }
    int mid = (l+r)>>1;
    build(2*i,l,mid);
    build(2*i+1,mid+1,r);
    push_up(i);
}
void push_down(int i)
{
    if(!tree[i].down) return;
    tree[2*i].down = tree[2*i+1].down = true;
    tree[2*i].dow = tree[2*i].dow * tree[i].dow;
    tree[2*i+1].dow = tree[2*i+1].dow * tree[i].dow;
    tree[2*i].sta = tree[2*i].sta * tree[i].dow;
    tree[2*i+1].sta = tree[2*i+1].sta * tree[i].dow;
    tree[i].down = false;
    tree[i].dow.init();
}
void deal_add(int i,int x,int y,Matrix val)
{
    int l = tree[i].l,r = tree[i].r;
    if(l == x && r == y)
    {
        tree[i].sta = tree[i].sta * val;
        tree[i].dow = tree[i].dow * val;
        tree[i].down = true;
        return;
    }
    push_down(i);
    int mid = (l+r)/2;
    if(y <= mid) deal_add(2*i,x,y,val);
    else
     if(x <= mid)
     {
       deal_add(2*i,x,mid,val);
       deal_add(2*i+1,mid+1,y,val);
     }
     else deal_add(2*i+1,x,y,val);
     push_up(i);
}
ll Find(int i,int x,int y)
{
    int l = tree[i].l,r = tree[i].r;
    if(l == x && r == y) return tree[i].sta.val[1][1];
    push_down(i);
    int mid = (l+r)/2;
    if(y <= mid) return Find(2*i,x,y);
    else
     if(x <= mid) return (Find(2*i,x,mid) + Find(2*i+1,mid+1,y)) % MOD;
     else return Find(2*i+1,x,y);
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i = 1;i <= n;i++) scanf("%d",&a[i]);
    build(1,1,n);
    for(int i = 1;i <= m;i++)
    {
        scanf("%d",&k);
        if(k == 1)
        {
            int l,r,x;
            scanf("%d%d%d",&l,&r,&x);
            Matrix temp = Matrix(2,2);
            temp.val[1][1] = temp.val[1][2] = temp.val[2][1] = 1;
            temp = ksm(temp,x);
            deal_add(1,l,r,temp);
        }
        else
        {
            int l,r;
            scanf("%d%d",&l,&r);
            printf("%I64d\n",Find(1,l,r));
        }
    }
}


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值