[agc013e]Placing Squares

前言

看到以为是整数拆分什么的搞了好久…

题目大意

给你一个n，你要计算所有序列(a[1]…a[x])的贡献和。
设s[i]表示序列前i个的前缀和，序列必须要满足：
1，s[x]=n;
2，给一个b[1..m]，要求对于任意i，s[i]不能是b中的元素。
一个序列的贡献为所有元素平方的乘积。
（原本的问题是在一条1*n的纸带上段进行正方形底边完全覆盖）
n<=1e9，m<=1e5

解题思路

关键是转化模型。把问题转化成：考虑有n个空格，你可以在两个空格中间放隔板，其中有一些位置不能放，两端必须放。在每个隔出来的区间中，你必须放两个可区分的球。问最后的方案数。易发现等价。
然后你设f[i][0/1/2]表示做完第i个空格，当前隔板区间有几个球的方案数。
弄两个转移矩阵，一个是只能放球，一个是又能放球又能放隔板。
你把n分成m+1段，中间用矩阵乘法加速即可。
时间复杂度O(mlogn)

代码

#include<cstdio> 
#include<algorithm>
#include<cstring>
#include<cmath>
#include<set>
#include<map>
using namespace std;
#define fo(i,j,k) for(i=j;i<=k;i++)
#define fd(i,j,k) for(i=j;i>=k;i--)
#define cmax(a,b) (a=(a>b)?a:b)
#define cmin(a,b) (a=(a<b)?a:b)
typedef long long ll;
const int N=1e6+5,M=2e6+5,mo=1e9+7;
int f[3],A[3][3],B[3][3],C[3][3],D[3][3],a[3][3],b[3][3],c[3][3],g[3];
int d[N],n,m,x,i,j,k,pos,l;
void ksm(int x)
{
    int i,j,k;
    fo(i,0,2) fo(j,0,2) c[i][j]=C[i][j];
    while(x)
    {
        if (x&1)
        {
            fo(i,0,2) fo(j,0,2) D[i][j]=0;
            fo(i,0,2) fo(j,0,2) fo(k,0,2)
                D[i][j]=(D[i][j]+1ll*a[i][k]*c[k][j])%mo;
            fo(i,0,2) fo(j,0,2) a[i][j]=D[i][j];
        }
        x>>=1;
        fo(i,0,2) fo(j,0,2) D[i][j]=0;
        fo(i,0,2) fo(j,0,2) fo(k,0,2)
            D[i][j]=(D[i][j]+1ll*c[i][k]*c[k][j])%mo;
        fo(i,0,2) fo(j,0,2) c[i][j]=D[i][j];
    }
}
int main()
{
    freopen("t8.in","r",stdin);
    //freopen("t8.out","w",stdout);
    scanf("%d %d",&n,&m);
    fo(i,1,m) scanf("%d",d+i);
    pos=0;// finished pos
    f[0]=1;
    A[0][1]=2;A[1][2]=1;A[0][2]=1;
    A[0][0]=1;A[1][1]=1;A[2][2]=1;
    B[2][0]=1;B[0][0]=1;B[1][1]=1;B[2][2]=1;
    fo(i,0,2) fo(j,0,2) 
        fo(k,0,2) 
            C[i][j]+=A[i][k]*B[k][j];
    d[++m]=n;
    fo(l,1,m)
    {
        x=d[l]-pos-1;
        fo(i,0,2) fo(j,0,2) b[i][j]=a[i][j]=0;
        fo(i,0,2) a[i][i]=1;
        if (x)
            ksm(x);//use C, prod in a
        fo(i,0,2) fo(j,0,2) fo(k,0,2) b[i][j]=(b[i][j]+1ll*a[i][k]*A[k][j])%mo;
        fo(i,0,2) g[i]=0;
        fo(i,0,2) fo(j,0,2) g[j]=(g[j]+1ll*f[i]*b[i][j])%mo;
        fo(i,0,2) f[i]=g[i];
        pos=d[l];
    }
    printf("%d",f[2]);
}