矩阵快速幂模板。
和我上一次写的不同,这里对于稀疏矩阵进行了优化。
这里就把每一个松鼠的表达式写出来就好了,就可以建立关系矩阵,注意判断条件n,m,t三个要同时为零。还有就是矩阵里面有常数,取1就好了。
#include<iostream>
#include<string>
#include<algorithm>
#include<cmath>
#include<cstdlib>
#include<climits>
#include<stack>
#include<vector>
#include<queue>
#include<set>
#include<map>
#include<cstdio>
#define up(i,a,b) for(int i=a;i<b;i++)
#define dw(i,a,b) for(int i=a;i>b;i--)
#define upd(i,a,b) for(int i=a;i<=b;i++)
#define dwd(i,a,b) for(int i=a;i>=b;i--)
//#define local
typedef long long ll;
const double esp = 1e-6;
const double pi = acos(-1.0);
const long long INF = 0x3f3f3f3f;
using namespace std;
typedef pair<int, int> pir;
int n, k;
ll m;
struct mat { ll a[103][103]; };
mat mat_mul(mat x, mat y)
{
mat res;
memset(res.a, 0, sizeof(res.a));
upd(i, 0, n)
{
upd(k, 0, n)
{
if (!x.a[i][k])continue;//一定要是i,k才能优化
upd(j, 0, n)
{
res.a[i][j] += x.a[i][k] * y.a[k][j];
}
}
}
return res;
}
mat mat_pow(int k, mat mod)
{
mat res;
memset(res.a, 0, sizeof(res.a));
upd(i, 0, n)
res.a[i][i] = 1;
while (k > 0)
{
if (k & 1)res = mat_mul(res, mod);
mod = mat_mul(mod, mod);
k >>= 1;
}
return res;
}
int main()
{
while (scanf("%d %d %d", &n, &m, &k)&&(n+m+k))
{
char s[2];
mat mod;
memset(mod.a, 0, sizeof(mod.a));
map<int , int > mp;//映射,初始化自己对自己,每个松鼠只会对应一个
upd(i, 0, n)
{
mp[i] = i;
}
int x, y;
up(i,0,k)
{
cin >> s;
if (s[0] == 'g')
{
//cin >> x;
scanf("%d", &x);
x--;
mod.a[x][n]++;//这和常数项有关
}
if (s[0] == 'e')
{
scanf("%d", &x);
x--;
mod.a[x][n] = 0;
mp[x] = -1;
}
if (s[0] == 's')
{
scanf("%d %d", &x, &y);
x--; y--;
swap(mod.a[x][n], mod.a[y][n]);//常数项和前面的松鼠
swap(mp[x], mp[y]);
}
}
/*upd(i, 0, n)
{
upd(j, 0, n)
{
if (i == j)mod.a[i][j] = 1;
else mod.a[i][j] = 0;
}
}
up(i, 0, k)
{
scanf("%s", s);
if (s[0] == 'g')
{
scanf("%d", &x);
mod.a[x-1][n]++;
}
if (s[0] == 's')
{
scanf("%d %d", &x, &y);
j] = 0;
}
}
}*/
upd(i, 0, n)
{
if (mp[i] != -1)
{
mod.a[i][mp[i]] = 1;//初始化矩阵
}
}
mod = mat_pow(m, mod);//矩阵快速幂
up(i, 0, n)
{
printf("%I64d ", mod.a[i][n]);//打印常数项
}
cout << endl;
}
return 0;
}