模拟题:Matrix

【题意】

先给出十个 1010 的矩阵 T[1]...T[10] 。(所有测试点这 10 个矩阵都相同)
然后给一个长度为 n 的序列a,将矩阵 T[a1],T[a2]...T[an] 排成一排。
按顺序执行100次操作,对于第 i 次操作,给定一个[1,n+i1]之间的正整数 x ,和一个矩阵G,然后再序列中第 x 个矩阵后紧挨着插入矩阵G.
计算新矩阵序列从左到右的乘积。
所有运算在模 109+7 意义下。

数据范围:
T[1...10] 每个矩阵中数字大小为 [1,mod) .
后插入的每个矩阵数字大小为 [1,10]
30% 1<=n<=1000
100% 1<=n<=106
时限 1s ,空间 256MB

【Trick!】

根据给定的十个矩阵,发现这十个矩阵满足交换律。。

【思路】

。。是否看出来上面的东西。。就是30分和100分的差别。。
平衡树单点插入构造出序列,然后用个前缀和维护一下,再找到每个插入点的位置。矩阵快速幂加速矩乘。。就可以过了。
(感觉这个题就难在看出交换律。。然而这么煞笔的平衡树我调了2个多小时。。自己做模拟也没有给十个矩阵的文件。。╮(╯▽╰)╭30就30把。。)

【代码】

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <queue>
#include <cstring>
#define N 1000105
#define mod 1000000007
#define INF 1000000001
using namespace std;
typedef long long ll;
typedef pair<int,int> pa;

int read()
{
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-') f=-1;ch=getchar();}
    while(isdigit(ch)){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return x*f;
} 

int n,rt,cnt,tot;
int a[N],fa[N],son[N][2],sz[N],b[N],sum[N][11],tmp[105],rank[N];

class Matrix{
    public:
        int v[11][11];
}T[111],ans;

Matrix operator *(Matrix a,Matrix b){
    Matrix rtn;
    for(int i=1;i<=10;i++)
    for(int j=1;j<=10;j++)
    {
        rtn.v[i][j]=0;
        for(int k=1;k<=10;k++)
            rtn.v[i][j]=(rtn.v[i][j]+(ll)a.v[i][k]*b.v[k][j]%mod)%mod;
    }
    return rtn;
}

Matrix Qpow(Matrix x,int y){
    Matrix rtn;
    for(int i=1;i<=10;i++) 
    for(int j=1;j<=10;j++) rtn.v[i][j]=(i==j)?1:0;
    while(y)
    {
        if(y&1) rtn=rtn*x;
        x=x*x;y>>=1;
    }
    return rtn;
}

void pushup(int k){
    int l=son[k][0],r=son[k][1];
    sz[k]=sz[l]+sz[r]+1;
}

void Build(int l,int r,int f)
{
    if(l>r) return;
    if(l==r)
    {
        fa[l]=f;son[f][l>f]=l;
        sz[l]=1;return;
    }
    int mid=l+r>>1;
    Build(l,mid-1,mid);Build(mid+1,r,mid);
    fa[mid]=f;pushup(mid);son[f][mid>f]=mid;
}

void Rotate(int x,int &k)
{
    int y=fa[x],z=fa[y],l,r;
    l=son[y][1]==x;r=l^1;
    if(y==k) k=x;
    else son[z][son[z][1]==y]=x;
    fa[x]=z;fa[y]=x;fa[son[x][r]]=y;
    son[y][l]=son[x][r];son[x][r]=y;
    pushup(y);pushup(x);
}

void Splay(int x,int &k)
{
    while(x!=k)
    {
        int y=fa[x],z=fa[y];
        if(y!=k) {
            if(son[z][0]==y^son[y][0]==x) Rotate(x,k);
            else Rotate(y,k);
        }
        Rotate(x,k);
    }
}

int Find(int k,int x)
{
    int l=son[k][0],r=son[k][1];
    if(sz[l]+1==x) return k;
    if(sz[l]+1>x) return Find(l,x);
    return Find(r,x-sz[l]-1);
}

void Insert(int k,int now)
{
    int x=Find(rt,k);
    int y=Find(rt,k+1);
    Splay(x,rt);
    Splay(y,son[x][1]);
    son[y][0]=now;fa[now]=y;sz[now]=1;
    pushup(y);pushup(x);
}

void print(Matrix a)
{
    for(int i=1;i<=10;i++)
    {
        for(int j=1;j<=10;j++) printf("%d ",a.v[i][j]);
        printf("\n");
    }
    printf("\n");
}

void Get_Rank(int k,int Delta)
{
    if(!k) return;
    int l=son[k][0],r=son[k][1];
    rank[k]=sz[l]+1+Delta;
    Get_Rank(l,Delta);
    Get_Rank(r,Delta+sz[l]+1);
}

int main()
{
    //freopen("D:\\oi\\in.txt","r",stdin);
    for(int i=1;i<=10;i++)
    for(int j=1;j<=10;j++)
    for(int k=1;k<=10;k++) T[i].v[j][k]=read();
    for(int i=1;i<=10;i++) T[0].v[i][i]=1;
    n=read();
    for(int i=1;i<=n;i++) a[i]=read();n++;
    Build(1,n,0);rt=(n+1)>>1;
    for(int i=1;i<=100;i++)
    {
        static int x;x=read();
        a[n+i]=10+i;
        for(int j=1;j<=10;j++)for(int k=1;k<=10;k++) T[10+i].v[j][k]=read();
        Insert(x,n+i);
    }
    Get_Rank(rt,0);
    for(int i=1;i<=100+n;i++) b[rank[i]]=a[i];
    for(int i=1;i<100+n;i++)
    {
        for(int j=1;j<=10;j++) sum[i][j]=sum[i-1][j];
        if(b[i]<=10&&b[i]) sum[i][b[i]]++;
        if(b[i]>10) tmp[++tot]=i; 
    }
    for(int i=1;i<=10;i++)
    for(int j=1;j<=10;j++) ans.v[i][j]=(i==j)?1:0;
    for(int i=1;i<=tot;i++)
    {
        for(int j=1;j<=10;j++)
            ans=ans*Qpow(T[j],sum[tmp[i]-1][j]-sum[tmp[i-1]][j]);
        ans=ans*T[b[tmp[i]]];
    }
    if(tmp[tot]!=99+n) 
        for(int j=1;j<=10;j++) 
            ans=ans*Qpow(T[j],sum[99+n][j]-sum[tmp[tot]][j]);
    print(ans);
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值