UVA - 11916 Emoogle Grid

最新推荐文章于 2021-05-27 16:20:53 发布

freeze up

最新推荐文章于 2021-05-27 16:20:53 发布

阅读量119

点赞数

分类专栏：数论

本文链接：https://blog.csdn.net/weixin_44059127/article/details/104175022

版权

数论同时被 2 个专栏收录

7 篇文章 0 订阅

订阅专栏

BSGS

1 篇文章 0 订阅

订阅专栏

题意：

给定 $m \times n$ 的方格，其中有 $b$ 个障碍物，除了障碍物以外的格子均需被染成 $k$ 种颜色中的一种，且要求 $(x, y), (x + 1, y)$ 不能染相同颜色，染色方案数模 $1 e 8 + 7$ 为 $r$ 。现在给定 $n, k, b, r$ ，求最小的 $m$ 。 $\leq 10^8, 2 \leq k \leq 10^8, b \leq 500)$

链接：

https://vjudge.net/problem/UVA-11916

解题思路：

若没有障碍物，则除了第一行每个格子染 $k$ 种外，其他格子均只有 $k - 1$ 种。有障碍物 $(x, y)$ 则 $(x + 1, y)$ 有 $k$ 种染色可能。计算出能染 $k$ 和 $k - 1$ 的格子数量以后，问题转化为求 $a^x \equiv b~mod~p$ 的最小的 $x$ ，用 $B S G S$ 算法求解。

参考代码：

#include<bits/stdc++.h>
 
using namespace std;
typedef long long ll;
typedef pair<int, int> pii;
#define sz(a) ((int)a.size())
#define pb push_back
#define lson (rt << 1)
#define rson (rt << 1 | 1)
#define gmid (l + r >> 1)
const int maxn = 3e5 + 5;
const int inf = 0x3f3f3f3f;
const int mod = 1e8 + 7;

map<int, int> mp;
set<pii> st;
vector<pii> G;
int n, k, b, r, mx;

ll qpow(ll a, ll b){

    ll ret = 1;
    while(b){

        if(b & 1) ret = ret * a % mod;
        a = a * a % mod;
        b >>= 1;
    }
    return ret;
}

ll inv(ll a){

    return qpow(a, mod - 2);
}

ll BSGS(ll a, ll b){

    ll lim = sqrt(mod + 0.5);
    ll cur = 1, inva = 1;
    for(int i = 0; i < lim; ++i){

        if(!mp.count(cur)) mp[cur] = i;
        cur = cur * a % mod, inva = inva * a % mod;
    }
    inva = inv(inva);
    for(int i = 0; i <= lim; ++i){

        if(mp.count(b)) return i * lim + mp[b];
        b = b * inva % mod;
    }
    return -1;
}

int solve(){

    ll cntK = 0, cntK_1 = 0, cntB = 0, cntT = 0;
    for(auto &it : G){

        if(it.first == 1) ++cntT;
        if(it.first == mx) ++cntB;
        else if(st.find({it.first + 1, it.second}) == st.end()) ++cntK;
    }
    cntK += n - cntT;
    cntK_1 = n * 1ll * mx - b - cntK;
    ll ret = qpow(k, cntK) * qpow(k - 1, cntK_1) % mod;
    if(ret == r) return mx;
    ret = ret * qpow(k, cntB) % mod * qpow(k - 1, n - cntB) % mod;
    if(ret == r) return mx + 1;
    return BSGS(qpow(k - 1, n), r * inv(ret) % mod) + mx + 1;
}

int main(){
 
    // ios::sync_with_stdio(0); cin.tie(0);
    int t, cas = 0; scanf("%d", &t);
    while(t--){

        scanf("%d%d%d%d", &n, &k, &b, &r);
        G.clear(), st.clear(), mp.clear();
        mx = 1;
        for(int i = 1; i <= b; ++i){

            int x, y; scanf("%d%d", &x, &y);
            G.pb({x, y}), mx = max(mx, x);
            st.insert({x, y});
        }
        int ret = solve();
        printf("Case %d: %d\n", ++cas, ret);
    }
    return 0;
}