HDU 5171 规律+矩阵快速幂(水)

HDU 5171
题意:给出序列a,操作:每次从序列选两个数相加后,将新数加入序列a.
n,a[i]<=1e5,k<=1e9.问操作k次后 序列的和最大为多少?


序列和最大,每次选当前最大和次大相加, k如果在1e5左右 优先队列直接模拟, 现在k<=1e9.
当前最大和次大分别为a,b..则操作第1,2,3,..k次操作后加入的数为
:a+b,2a+b,3a+2b,5a+3b,8a+5b,...发现第k次a和b的系数为fib(k+1),fib(k).
新添加的数之和为 a*(f[2]+f[3]+..f[k+1])+(f[1]+f[2]+..f[k])*b.

根据(f[1]+f[2]+..f[k]) = f[k+2]-1 则用矩阵快速算出fib后计算即可.

#include <bits/stdc++.h>
using namespace std;
typedef pair<int,int> ii;
typedef long long ll;
const int N=2e5+5,M=2;
const ll mod=10000007;
ll n,k,a[N];
struct matrix{
	ll a[3][3];
	void init()
	{

		a[1][1]=1;
		a[1][2]=a[2][1]=1;
		a[2][2]=0;
	}
};
matrix product(matrix a,matrix b)//矩阵乘法
{
	matrix c;
	for(int i=1;i<=2;i++)
	{
		for(int j=1;j<=2;j++)
		{
			c.a[i][j]=0;
			for(int k=1;k<=2;k++)
			{
				c.a[i][j]+=(a.a[i][k]*b.a[k][j]);
			}
			c.a[i][j]%=mod;
		}
	}
	return c;
}
matrix mul(matrix s,int k)
{
	matrix ans;
	ans.init();
	while(k)
	{
		if(k%2)
		ans=product(ans,s);

		s=product(s,s);
		k=k/2;
	}
	return ans;
}
int main()
{
    while(cin>>n>>k)
    {
        ll ans=0,y1=0,y2=0;
        for(int i=1;i<=n;i++)
        {
            scanf("%lld",&a[i]),ans=(ans+a[i])%mod;
            if(y1<=a[i])
                y2=y1,y1=a[i];
            else if(y2<a[i])
                y2=a[i];
        }
        matrix s;
        s.init();
        s=mul(s,k+2);
        ll x1=s.a[1][2]-1,x2=s.a[2][2]-1;
        ll res=(x1*y1)%mod+(x2*y2)%mod;
        res=(res-y1+mod)%mod;
        ans=(ans+res)%mod;
        cout<<ans<<endl;
    }
    return 0;
}