题目
长为n(3<=n<=1e9)的数组,第i个数ai在[0,m](m<=1e9)之间
对于i∈[1,n],均满足a[i]+(a[i-1]&a[i+1])=m,求这样可能的数组的方案数
特别地,认为a[0]=a[n],a[n+1]=a[1],即这个数组是个环形的数组
思路来源
官方题解
题解
从末位考虑,
1. 如果m=0,只能全是0,方案数为1
2. 如果m=1,由于1+(1&1)=2,0+(0&x)=0,所以不能有三个1相邻,不能有两个0相邻
将相邻位的数字每两个看成一个点,即从01/10/11出发,
01可以转移到10或11,11可以转移到10,10只能转移到01,矩阵快速幂求出长为n的方案数
3. 如果m为>=2的偶数,考虑末位是1还是0,
①如果是1,只能是1+(1&1)=2,此时还给倒数第二位贡献了1,需要再递归求m/2-1的方案
②如果是0,只能是全0,需要再递归求m/2的方案
有f(m)=f(m/2)+f(m/2-1)
4. 如果m为>=3的奇数,末位只能是1,并且由于没有进位,可以分开来看
有f(m)=f(1)+f((m-1)/2)
代码
#include <bits/stdc++.h>
#include<iostream>
#include<cstdio>
#include<vector>
#include<map>
using namespace std;
#define rep(i,a,b) for(int i=(a);i<=(b);++i)
#define per(i,a,b) for(int i=(a);i>=(b);--i)
typedef long long ll;
typedef double db;
typedef pair<int,int> P;
#define fi first
#define se second
#define pb push_back
#define dbg(x) cerr<<(#x)<<":"<<x<<" ";
#define dbg2(x) cerr<<(#x)<<":"<<x<<endl;
#define SZ(a) (int)(a.size())
#define sci(a) scanf("%d",&(a))
#define pt(a) printf("%d",a);
#define pte(a) printf("%d\n",a)
#define ptlle(a) printf("%lld\n",a)
#define debug(...) fprintf(stderr, __VA_ARGS__)
using namespace std;
using namespace std;
typedef long long ll;
const int N=1e5+10,mod=998244353;
struct mat {
static const int MAXN=3;
ll c[MAXN][MAXN];
int m, n;
mat(){
memset(c, 0, sizeof(c));
m=n=MAXN;
}
mat(int a, int b) : m(a), n(b) {
memset(c, 0, sizeof(c));
}
void clear(){
memset(c, 0, sizeof(c));
}
mat operator * (const mat& temp) {
mat ans(m, temp.n);
for (int i = 0; i < m; i ++)
for (int j = 0; j < temp.n; j ++){
for (int k = 0; k < n; k ++){
ans.c[i][j] += c[i][k] * temp.c[k][j];
ans.c[i][j]%=mod;
}
}
return ans;
}
mat operator ^(ll n){
mat M(*this),ans(M.m, M.m);
for (int i = 0; i < M.m; i ++)
ans.c[i][i] = 1;
while (n > 0) {
if (n & 1) ans = ans * M;
M = M * M;
n >>= 1;
}
return ans;
}
}b;
//01 10 11 没有相邻两个0或者三个1的方案数
int n,m,ans;
int sol(int m){
if(m==0)return 1;//全0
if(m==1)return ans;
if(m%2==0)return (sol(m/2)+sol(m/2-1))%mod;//进位和不进位
return 1ll*ans*sol(m/2)%mod;
}
int main(){
scanf("%d%d",&n,&m);
b.c[0][1]=b.c[0][2]=1;//01->10 01->11
b.c[1][0]=1;//10->01
b.c[2][1]=1;//11->10
b=b^n;
rep(i,0,2){
ans=(ans+b.c[i][i])%mod;
}
printf("%d\n",sol(m));
return 0;
}