2020暑期训练5

矩阵快速幂

矩阵乘法和数的乘法一样都满足结合律,所以可以像写普通快速幂一样写矩阵快速幂。
例题:P3390 【模板】矩阵快速幂

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> pii;
typedef vector<int> vi;
const int INF = 0x3f3f3f3f;
const int maxn = 1e5 + 9;
const ll mod = (ll)1000000007;
inline ll read()
{
    ll data = 0, f = 1;
    char ch = getchar();
    while (!isdigit(ch))
    {
        if (ch == '-')
            f = -1;
        ch = getchar();
    }
    while (isdigit(ch))
    {
        data = (data << 3) + (data << 1) + ch - '0';
        ch = getchar();
    }
    return f * data;
}
struct matrix
{
    ll mat[105][105];
    int size;
    matrix()//这是必要的
    {
        memset(mat, 0, sizeof(mat));
        size = 0;
    }
    void create_unit(int s)
    {
        size = s;
        for (int i = 1; i <= s; ++i)
            mat[i][i] = 1;
    }
    void init()
    {
        for (int i = 1; i <= size; ++i)
            for (int j = 1; j <= size; ++j)
                mat[i][j] = read();
    }
    matrix operator*(const matrix &b)
    {
        matrix ans;
        for (int i = 1; i <= size; ++i)
            for (int j = 1; j <= size; ++j)
                for (int k = 1; k <= size; ++k)
                    ans.mat[j][k] = (ans.mat[j][k] + mat[j][i] * b.mat[i][k] % mod) % mod;
        ans.size = size;
        return ans;
    }
    /*这个没必要重载
    void operator=(const matrix &b)
    {
        size = b.size;
        for (int i = 1; i <= size; ++i)
            for (int j = 1; j <= size; ++j)
                mat[i][j] = b.mat[i][j];
    }
    */
    void print()
    {
        for (int i = 1; i <= size; ++i)
        {
            for (int j = 1; j <= size; ++j)
                cout << mat[i][j] << ' ';
            cout << '\n';
        }
    }
};
matrix a, ans;

int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("data.txt","w",stdout);
    //std::ios::sync_with_stdio(false);
    //std::cin.tie(0);
    a.size = read();
    ll k = read();
    a.init();
    ans.create_unit(a.size);
    while (k)
    {
        if (k & 1)
            ans = ans * a;
        a = a * a, k /= 2;
    }
    ans.print();
    //fclose(stdin);
    //fclose(stdout);
    return 0;
}

矩阵加速

推导数列的时候用矩阵快速幂加速
例题:P1939 【模板】矩阵加速(数列)
递推式:
[ a n a n − 1 a n − 2 ] = [ 1 0 1 1 0 0 0 1 0 ] × [ a n − 1 a n − 2 a n − 3 ] \begin{bmatrix} a _n \\ a_{n-1}\\ a_{n-2} \end{bmatrix}=\begin{bmatrix} 1&0&1 \\ 1&0&0\\ 0&1&0 \end{bmatrix}\times \begin{bmatrix} a _{n-1} \\ a_{n-2}\\ a_{n-3} \end{bmatrix} anan1an2=110001100×an1an2an3
那么只要对系数矩阵用快速幂就可以了,然后这样得到的 m a t [ 1 ] [ 1 ] mat[1][1] mat[1][1] a n + 1 a_{n+1} an+1,所以输出 m a t [ 2 ] [ 1 ] mat[2][1] mat[2][1]

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> pii;
typedef vector<int> vi;
const int INF = 0x3f3f3f3f;
const int maxn = 1e5 + 9;
const ll mod = (ll)1000000007;
inline ll read()
{
    ll data = 0, f = 1;
    char ch = getchar();
    while (!isdigit(ch))
    {
        if (ch == '-')
            f = -1;
        ch = getchar();
    }
    while (isdigit(ch))
    {
        data = (data << 3) + (data << 1) + ch - '0';
        ch = getchar();
    }
    return f * data;
}
struct matrix
{
    ll mat[105][105];
    int size;
    matrix()
    {
        memset(mat, 0, sizeof(mat));
        size = 0;
    }
    void create_unit(int s)
    {
        size = s;
        for (int i = 1; i <= s; ++i)
            mat[i][i] = 1;
    }
    void init()
    {
        for (int i = 1; i <= size; ++i)
            for (int j = 1; j <= size; ++j)
                mat[i][j] = read();
    }
    matrix operator*(const matrix &b)
    {
        matrix ans;
        for (int i = 1; i <= size; ++i)
            for (int j = 1; j <= size; ++j)
                for (int k = 1; k <= size; ++k)
                    ans.mat[j][k] = (ans.mat[j][k] + mat[j][i] * b.mat[i][k] % mod) % mod;
        ans.size = size;
        return ans;
    }
    void print()
    {
        for (int i = 1; i <= size; ++i)
        {
            for (int j = 1; j <= size; ++j)
                cout << mat[i][j] << ' ';
            cout << '\n';
        }
    }
};

int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("data.txt","w",stdout);
    //std::ios::sync_with_stdio(false);
    //std::cin.tie(0);
    int T = read();
    while (T--)
    {
        int n = read();
        if (n <= 3)
        {
            puts("1");
            continue;
        }
        matrix a, ans;
        a.size = 3, a.mat[1][1] = 1, a.mat[1][3] = 1, a.mat[2][1] = 1, a.mat[3][2] = 1;
        ans.create_unit(3);
        while (n)
        {
            if (n & 1)
                ans = ans * a;
            a = a * a, n /= 2;
        }
        cout << ans.mat[2][1] << '\n';
    }
    //fclose(stdin);
    //fclose(stdout);
    return 0;
}

单调队列优化

例题:P3800 Power收集
数据比较大,复杂度多一个log恐怕都会T,只能用单调队列。做法显然是从第二行开始给每个格子加上上一行对应的 2 × v 2\times v 2×v 长度的区间内最大值,一直到最后一行。

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> pii;
typedef vector<int> vi;
const int INF = 0x3f3f3f3f;
const int maxn = 4001;
inline int read()
{
    int data = 0, f = 1;
    char ch = getchar();
    while (!isdigit(ch))
    {
        if (ch == '-')
            f = -1;
        ch = getchar();
    }
    while (isdigit(ch))
    {
        data = (data << 3) + (data << 1) + ch - '0';
        ch = getchar();
    }
    return f * data;
}
deque<int> dq;
int n, m, v, k, dp[maxn][maxn];
void ins(int d, int pos)
{
    while (!dq.empty() && dp[d][pos] > dp[d][dq.back()])
        dq.pop_back();
    dq.push_back(pos);
}
int find(int d, int pos)
{
    if (pos + v <= m)
        ins(d, pos + v);
    while (dq.front() + v < pos)
        dq.pop_front();
    return dq.front();
}
int main()
{
    //freopen("in.txt","r",stdin);
    //freopen("data.txt","w",stdout);
    //std::ios::sync_with_stdio(false);
    //std::cin.tie(0);
    n = read(), m = read(), k = read(), v = read();
    for (int i = 1; i <= k; ++i)
    {
        int x = read(), y = read();
        dp[x][y] = read();
    }
    for (int i = 2; i <= n; ++i)
    {
        for (int j = 1; j <= v; ++j)
            ins(i - 1, j);
        for (int j = 1; j <= m; ++j)
            dp[i][j] += dp[i - 1][find(i - 1, j)];
        dq.clear();
    }
    cout << *max_element(dp[n] + 1, dp[n] + m + 1) << '\n';
    //fclose(stdin);
    //fclose(stdout);
    return 0;
}

max_element() 函数返回一个指向最大元素的位置的指针

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值