洋灰三角 矩阵快速幂

洋灰三角 矩阵快速幂

我的思路：
先定义一个矩阵，然后通过矩阵的相乘，实现n>>1的效果。

我的解法：（满分答案）

//#include<bits/stdc++.h>
#include<iostream>
#include<algorithm>
#include<string>
#include<string.h>
#include<map>
#include<set>
#include<vector>
#include<iomanip>
#include<sstream>
#define _for(i,a,b) for(int i=a;i<b;i++)
#define _rep(i,a,b) for(int i=a;i<=b;i++)
const int N = 1e9 + 7;
typedef long long ll;
using namespace std;
struct jz {					//矩阵快速幂
	ll mes[3][3];
	jz() { memset(mes, 0, sizeof(mes)); }
	void setBase(int k) {
		mes[0][0] = 1; mes[0][1] = k; mes[0][2] = 1;
		mes[1][0] = 0; mes[1][1] = k; mes[1][2] = 1;
		mes[2][2] = 1;
	}
	void setAns(int sum, int x,int p) {
		mes[0][0] = sum; mes[1][0] = x; mes[2][0] = p;
	}
	
}cs,ans;
jz xc(jz a,jz b) {
	jz c;
		_for(i, 0, 3) {
			_for(j, 0, 3) {
				_for(k, 0, 3) {
					c.mes[i][j] = (c.mes[i][j]+a.mes[i][k] * b.mes[k][j])%N;
				}
			}
		}
		return c;
	}
jz solve(int n) {
	while (1) {
		if (n == 0) {
			return ans;
		}
		if (n & 1)ans = xc(cs,ans);		//判断是否相乘
		cs = xc(cs,cs);				//利用矩阵的平方效果
		n >>= 1;
	}
}
int main() {
	int n, k, p;
	cin >> n >> k >> p;
	cs.setBase(k);
	ans.setAns(k+p+1,k+p,p);
	if (n == 1) {
		cout << "1" << endl;
	}
	else if (n == 2) {
		cout << k + p+1 << endl;
	}
    else{
	    ans = solve(n - 2);
        cout << ans.mes[0][0] << endl;
    }
	return 0;
}

链接：https://ac.nowcoder.com/acm/contest/136/J
来源：牛客网

洋灰是一种建筑材料，常用来筑桥搭建高层建筑，又称，水泥、混凝土。
WHZ有很多铸造成三角形的洋灰块，他想把这些洋灰三角按照一定的规律放到摆成一排的n个格子里，其中第i个格子放入的洋灰三角数量是前一个格子的k倍再多p个，特殊地，第一个格子里放1个。
WHZ想知道把这n个格子铺满需要多少洋灰三角。
输入描述:
第一行有3个正整数n，k，p。
输出描述:
输出一行，一个正整数，表示按照要求铺满n个格子需要多少洋灰三角，由于输出数据过大，你只需要输出答案模1000000007(1e9+7)后的结果即可。
示例1
输入
复制
3 1 1
输出
复制
6
说明
洋灰三角铺法：1 2 3，总计6个
示例2
输入
复制
3 2 2
输出
复制
15
说明
洋灰三角铺法：1 4 10，总计15个
示例3
输入
复制
3 3 3
输出
复制
28
说明
洋灰三角铺法：1 6 21，总计28个
备注:
对于100%的测试数据：
1 ≤ n ≤ 1000000000
1 ≤ k,p ≤ 1000