POJ 3735

矩阵快速幂模板。
和我上一次写的不同,这里对于稀疏矩阵进行了优化。
这里就把每一个松鼠的表达式写出来就好了,就可以建立关系矩阵,注意判断条件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;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值