51nod 1836 战忽局的手段 矩阵乘法

题意

众所周知，有一个神秘的组织——战忽局，在暗中保护着我们。在局中任职的官员都有着极强的忽悠技巧，不只能用预言，还能用往事忽悠人。如今某外星间谍已经获得了战忽局曾经参与的n次事件的资料，局座发现了这件事，于是决定再次用忽悠来保证战忽局的安全。局座将发表m次演讲，每一天他都会从n事件中等概率地挑选一件混淆众人，由于局座每天很忙，不能把之前将的事件都记录下来，因此他可能会重复选择某一件事。现在局座想知道，m次演讲过后，期望能使多少事件混淆众人。
T<=1000,n,m<=10^18

分析

设f[i]表示i次演讲后期望时间。
通过推式子不难得到f[i]=f[i-1]*(n-1)/n+1。
直接矩阵乘法优化即可。
由于精度问题要用__float128.

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
using namespace std;

typedef long long LL;

struct Matrix{__float128 a[3][3];}a;

void mul(Matrix &c,Matrix a,Matrix b)
{
    for (int i=1;i<=2;i++)
        for (int j=1;j<=2;j++)
            c.a[i][j]=0;
    for (int i=1;i<=2;i++)
        for (int k=1;k<=2;k++)
            for (int j=1;j<=2;j++)
                c.a[i][j]+=a.a[i][k]*b.a[k][j];
}

Matrix ksm(Matrix x,LL y)
{
    Matrix ans=x;y--;
    while (y)
    {
        if (y&1) mul(ans,ans,x);
        mul(x,x,x);y>>=1;
    }
    return ans;
}

int main()
{
    int T;
    scanf("%d",&T);
    while (T--)
    {
        LL n,m;
        scanf("%lld%lld",&n,&m);
        a.a[1][1]=a.a[1][2]=1.0;
        a.a[2][1]=0;a.a[2][2]=(__float128)1.0*(n-1)/n;
        a=ksm(a,m);
        printf("%.7lf\n",(double)a.a[1][2]);
    }
    return 0;
}