题意:
就是给你一个n×n的矩阵,问你n的k次幂是多少。
思考:
代码:
#include
#define fi first
#define se second
#define pb push_back
#define db double
#define int long long
#define PII pair<int,int >
#define mem(a,b) memset(a,b,sizeof(a))
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
using namespace std;
const int mod = 1e9+7,inf = 1e18;
const int N = 2e5+10,M = 2010;
int T,n,m,k;
int va[N];
struct matx{
int n,m,a[M][M],b[M][M],c[M][M];
void init()
{
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++) c[i][j] = (i==j);
}
void ksm1()
{
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
b[i][j] = c[i][j];
}
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
c[i][j] = 0;
for(int k=1;k<=n;k++)
c[i][j] = (c[i][j]+a[i][k]*b[k][j]%mod)%mod;
}
}
}
void ksm2()
{
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
b[i][j] = a[i][j];
}
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
a[i][j] = 0;
for(int k=1;k<=n;k++)
a[i][j] = (a[i][j]+b[i][k]*b[k][j]%mod)%mod;
}
}
}
void ksm()
{
while(m)
{
if(m&1) ksm1();
ksm2();m >>= 1;
}
}
}mat;
signed main()
{
IOS;
cin>>n>>m;
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
cin>>mat.a[i][j];
}
mat.n = n,mat.m = m;mat.init();mat.ksm();
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
cout<<mat.c[i][j]<<" ";
cout<<"\n";
}
return 0;
}
题意:
给你一个数列a,1<=i<=3时a[i] = 1,i>=4时a[i] = a[i-1]+a[i-3]。现在让你求这个数列的第n项,问你是多少。
思考:
代码:
#include
#define fi first
#define se second
#define pb push_back
#define db double
#define int long long
#define PII pair<int,int >
#define mem(a,b) memset(a,b,sizeof(a))
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
using namespace std;
const int mod = 1e9+7,inf = 1e18;
const int N = 2e5+10,M = 10;
int T,n,m,k;
int va[M][M];
int anw[M][M];
struct matx{
int n,m,a[M][M],b[M][M],c[M][M];
void init()
{
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++) c[i][j] = (i==j);
}
void ksm1()
{
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
b[i][j] = c[i][j];
}
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
c[i][j] = 0;
for(int k=1;k<=n;k++)
c[i][j] = (c[i][j]+a[i][k]*b[k][j]%mod)%mod;
}
}
}
void ksm2()
{
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
b[i][j] = a[i][j];
}
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
a[i][j] = 0;
for(int k=1;k<=n;k++)
a[i][j] = (a[i][j]+b[i][k]*b[k][j]%mod)%mod;
}
}
}
void ksm()
{
while(m)
{
if(m&1) ksm1();
ksm2();m >>= 1;
}
}
}mat;
signed main()
{
IOS;
cin>>T;
while(T--)
{
cin>>n;
if(n<=3) cout<<1<<"\n";
else
{
mat.a[1][1] = 1,mat.a[1][2] = 0,mat.a[1][3] = 1;
mat.a[2][1] = 1,mat.a[2][2] = 0,mat.a[2][3] = 0;
mat.a[3][1] = 0,mat.a[3][2] = 1,mat.a[3][3] = 0;
mat.n = 3,mat.m = n-3;mat.init();mat.ksm();
va[1][1] = 1,va[2][1] = 1,va[3][1] = 1;
for(int i=1;i<=3;i++)
{
for(int j=1;j<=1;j++)
{
anw[i][j] = 0;
for(int k=1;k<=3;k++)
anw[i][j] = (anw[i][j]+mat.c[i][k]*va[k][j]%mod)%mod;
}
}
cout<<anw[1][1]<<"\n";
}
}
return 0;
}
题意:
就是给你n个点,刚开始在第1点,然后问你经过k天后也回到1点,中间可以随便走,但是不能有两个相邻天并且点也相同的情况。问你方案数有多少,并取模。
思考:
代码:
#include
#define fi first
#define se second
#define pb push_back
#define db double
#define int long long
#define PII pair<int,int >
#define mem(a,b) memset(a,b,sizeof(a))
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
using namespace std;
const int mod = 998244353,inf = 1e18;
const int N = 2e5+10,M = 10;
int T,n,m,k;
int va[M][M];
int anw[M][M];
struct matx{
int n,m,a[M][M],b[M][M],c[M][M];
void init()
{
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++) c[i][j] = (i==j);
}
void ksm1()
{
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
b[i][j] = c[i][j];
}
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
c[i][j] = 0;
for(int k=1;k<=n;k++)
c[i][j] = (c[i][j]+a[i][k]*b[k][j]%mod)%mod;
}
}
}
void ksm2()
{
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
b[i][j] = a[i][j];
}
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
a[i][j] = 0;
for(int k=1;k<=n;k++)
a[i][j] = (a[i][j]+b[i][k]*b[k][j]%mod)%mod;
}
}
}
void ksm()
{
while(m)
{
if(m&1) ksm1();
ksm2();m >>= 1;
}
}
}mat;
signed main()
{
IOS;
cin>>n>>m;
mat.a[1][1] = n-2,mat.a[1][2] = n-1;
mat.a[2][1] = 1,mat.a[2][2] = 0;
mat.n = 2;mat.m = m;mat.init();mat.ksm();
va[1][1] = 0,va[2][1] = 1;
for(int i=1;i<=2;i++)
{
for(int j=1;j<=1;j++)
{
for(int k=1;k<=2;k++)
anw[i][j] = (anw[i][j]+mat.c[i][k]*va[k][j]%mod)%mod;
}
}
cout<<anw[2][1];
return 0;
}
题意:
就是给你n个城市,机器人在1号城市上,机器人有三种行为,停留在原地,去某个相邻的城市,或者自爆。机器人每秒都会行动一次,第0秒在1号点。现在问你经过了t秒,机器人的行为方案树是多少。
思考:
代码:
矩阵
#include
#define fi first
#define se second
#define pb push_back
#define db double
#define int long long
#define PII pair<int,int >
#define mem(a,b) memset(a,b,sizeof(a))
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
using namespace std;
const int mod = 2017,inf = 1e18;
const int N = 2e5+10,M = 100;
int T,n,m,k;
int va[N];
struct matx{
int n,m,a[M][M],b[M][M],c[M][M];
void init()
{
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++) c[i][j] = (i==j);
}
void ksm1()
{
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
b[i][j] = c[i][j];
}
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
c[i][j] = 0;
for(int k=1;k<=n;k++)
c[i][j] = (c[i][j]+a[i][k]*b[k][j]%mod)%mod;
}
}
}
void ksm2()
{
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
b[i][j] = a[i][j];
}
for(int i=1;i<=n;i++)
{
for(int j=1;j<=n;j++)
{
a[i][j] = 0;
for(int k=1;k<=n;k++)
a[i][j] = (a[i][j]+b[i][k]*b[k][j]%mod)%mod;
}
}
}
void ksm()
{
while(m)
{
if(m&1) ksm1();
ksm2();m >>= 1;
}
}
}mat;
signed main()
{
IOS;
cin>>n>>m;
for(int i=1;i<=m;i++)
{
int x,y;
cin>>x>>y;
mat.a[x][y] = mat.a[y][x] = 1;
}
for(int i=1;i<=n+1;i++) mat.a[i][i] = 1;
for(int i=1;i<=n;i++) mat.a[i][n+1] = 1;
cin>>k;
mat.n = n+1,mat.m = k;mat.init();mat.ksm(); //注意是n+1个点,因为多了一个自爆点。
int ans = 0;
for(int i=1;i<=n+1;i++) ans = (ans+mat.c[1][i])%mod;
cout<<ans;
return 0;
}
直接dp
#include
#define fi first
#define se second
#define pb push_back
#define db double
#define ll long long
#define PII pair<int,int >
#define mem(a,b) memset(a,b,sizeof(a))
#define IOS std::ios::sync_with_stdio(false),cin.tie(0),cout.tie(0);
using namespace std;
const int mod = 2017,inf = 1e18;
const int N = 1e6+1,M = 31;
int T,n,m,k;
int dp[N][M];
vector<int > e[M];
signed main()
{
IOS;
cin>>n>>m;
for(int i=1;i<=m;i++)
{
int a,b;
cin>>a>>b;
e[a].pb(b);
e[b].pb(a);
}
for(int i=1;i<=n;i++)
{
e[i].pb(0);
e[i].pb(i);
}
e[0].pb(0);
cin>>k;
dp[0][1] = 1;
for(int i=0;i<=k;i++)
{
for(int a=0;a<=n;a++)
{
for(auto b:e[a])
dp[i+1][b] = (dp[i+1][b]+dp[i][a])%mod;
}
}
int ans = 0;
for(int i=0;i<=n;i++) ans = (ans+dp[k][i])%mod;
cout<<ans;
return 0;
}