题目大意
有一个长度为
n
n
n 的数轴(看作是
n
n
n 个格子排成一行),其中有
m
m
m 个交界位置被标记了。你要用若干正方形去覆盖这个数轴(如下图),有 3 个规定:
1、正方形边长必须是正整数
2、数轴要被恰好覆盖,即不能有空、不能有地方被多个正方形覆盖。
3、被标记的位置不能是正方形的交界。
一种方案的价值是所有正方形的面积的积。求所有合法方案的价值和。
n
<
=
1
0
9
,
m
<
=
1
0
5
n<=10^9,~m<=10^5
n<=109, m<=105
题解
~~~~~~ 这个很妙啊。。。
~~~~~~ 首先瞎写一个dp: f [ i ] = ∑ j = 0 i − 1 f [ j ] ∗ ( i − j ) 2 f[i]=\sum_{j=0}^{i-1}f[j]*(i-j)^2 f[i]=∑j=0i−1f[j]∗(i−j)2,然后发现这个平方非常不好搞。
~~~~~~
于是转化模型,把
x
2
x^2
x2 看作是长度为
x
x
x 的段内,恰好放一个红球和一个蓝球(可以放在同一位置)的方案数。
~~~~~~
于是可以设
f
[
i
]
[
j
]
(
j
∈
[
0
,
2
]
)
f[i][j](j∈[0,2])
f[i][j](j∈[0,2]) 表示到了第
i
i
i 个格子,当前段内有
j
j
j 个球,的方案数。转移就是讨论一下 9 个转移,然后写出转移矩阵就可以矩阵乘法了。
~~~~~~
至于标记的位置,实际上是那些位置的
f
[
i
]
[
2
]
f[i][2]
f[i][2] 的转移要改一下,不能新开一个正方形。
~~~~~~
所以时间是
O
(
m
l
o
g
n
)
O(m~log~n)
O(m log n)
代码
#include<bits/stdc++.h>
#define fo(i,a,b) for(int i=a;i<=b;i++)
using namespace std;
typedef long long LL;
const int maxm=1e5+5;
const LL mo=1e9+7;
struct ARR{
LL n[3][3];
};
ARR re;
ARR operator * (const ARR &a,const ARR &b)
{
fo(i,0,2)
fo(j,0,2)
{
re.n[i][j]=0;
fo(k,0,2) (re.n[i][j]+=a.n[i][k]*b.n[k][j])%=mo;
}
return re;
}
int m;
LL n,x[maxm];
ARR C,ans;
void mi(LL y)
{
for(; y; y>>=1, C=C*C) if (y&1) ans=ans*C;
}
int main()
{
scanf("%lld %d",&n,&m);
fo(i,1,m) scanf("%d",&x[i]);
x[++m]=n;
sort(x+1,x+1+m);
ans=(ARR){{{1,0,0},{0,0,0},{0,0,0}}};
x[0]=-1;
fo(i,1,m)
{
LL len=x[i]-(x[i-1]+1);
C=(ARR){{{1,2,1},{0,1,1},{1,2,2}}};
mi(len);
if (i==m) break;
C=(ARR){{{1,2,1},{0,1,1},{0,0,1}}};
ans=ans*C;
}
printf("%lld\n",ans.n[0][2]);
}