UVA 11916 Emoogle Grid (BSGS)

最新推荐文章于 2020-09-21 17:42:59 发布

gongyuandaye

最新推荐文章于 2020-09-21 17:42:59 发布

阅读量155

点赞数

分类专栏：数论文章标签： BSGS

本文链接：https://blog.csdn.net/qq_43680965/article/details/105939800

版权

数论专栏收录该内容

38 篇文章 0 订阅

订阅专栏

题意：原问题是有M*N个方格，K种颜色，有B个方格不能涂色，且垂直相连的两个方格不能涂同一种颜色。问有多少种涂色方案。现在问题改为已知N,K,B以及原问题的结果r求最小的可能的M。

题解：BSGS
首先考虑没有不能涂色的方格的情况，那么第一行的每一个方格都可以任意涂色，就是kⁿ，而剩余的方格除了对应的上一行的颜色都可以随意填，(k - 1)^(n * m - n)，两式相乘就是答案。
当出现了不能填颜色的方格时，如果它的下面有方格，这个方格就可以任意涂色了。这个时候判断一下多少个方格可以涂k种颜色，多少个方格可以涂k-1种颜色。

求最小行数时要用到BSGS （a^x≡b(mod p)中x的最小值）

具体看代码注释。
注意模数是1e8+7。

#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#include<algorithm>
#include<queue>
#include<stack>
#include<cmath>
#include<vector>
#include<fstream>
#include<set>
#include<map>
#include<sstream>
#include<iomanip>
#define ll long long
using namespace std;
const int mod = 1e8 + 7;
ll t, n, k, b, r, x, y;
struct node {
	ll a, b;
	node() {}
	node(int a, int b) :a(a), b(b) {}
	bool operator<(const node m)const {
		return b == m.b ? a < m.a : b < m.b;
	}
}block[555];
ll quick_pow(ll a, ll b) {
    ll res = 1, tmp = a;
    while (b) {
        if (b & 1) res = res * tmp % mod;
        tmp = tmp * tmp % mod;
        b >>= 1;
    }
    return res;
}
void exgcd(ll a, ll b, ll& d, ll& x, ll& y) {
    if (b == 0) {
        d = a;
        x = 1;
        y = 0;
    }
    else {
        exgcd(b, a % b, d, y, x);
        y -= a / b * x;
    }
}
ll BSGS(ll a, ll b, ll p) {
    ll m = (ll)ceil(sqrt(p + 0.5));
    map<ll, int> rid;
    ll tmp = 1;
    for (int i = 0; i < m; ++i) {
        if (rid.find(tmp) == rid.end()) rid[tmp] = i; //需要判断是否已经出现过
        tmp = tmp * a % p;
    }
    ll tmp2 = quick_pow(tmp, p - 2); //模质数求逆元
    for (int i = 0; i < m; ++i) {
        if (rid.find(b) != rid.end()) return rid[b] + i * m;
        b = b * tmp2 % p;
    }
}

int main() {
	scanf("%d", &t);
    int cas = 0;
	while (t--) {
		scanf("%lld%lld%lld%lld", &n, &k, &b, &r);
        ll maxr = 1; //注意初始化
		for (int i = 1; i <= b; i++) {
			scanf("%d%d", &x, &y);
			block[i] = node(x, y);
            maxr = max(maxr, x);
		}
        sort(block + 1, block + b + 1);
        ll num = 0, bottom = 0;  //num为可以涂k个颜色的数量
        for (int i = 1; i <= b; i++) {
            if (block[i].a > 1) {
                if (i == 1) num++;
                else if (block[i - 1].b != block[i].b) num++;
                else if (block[i - 1].a + 1 != block[i].a) num++;
            }
            if (block[i].a == maxr) bottom++;
        }
        num += n - bottom;
        //temp为答案
        ll temp = quick_pow(k, num) * quick_pow(k - 1, n * maxr - b - num) % mod;
        if (temp == r) {  //答案刚好满足
            printf("Case %d: %lld\n", ++cas, maxr);
            continue;
        }
        //不满足，继续增加行数
        //先增加一行（maxr + 1）得到的答案
        temp = temp * quick_pow(k, bottom) % mod * quick_pow(k - 1, n - bottom) % mod;
        ll x, y, d;
        exgcd(temp, mod, d, x, y);
        x = (x * (r / d) % mod + mod) % mod;  //注意x为负数情况
        ll ans = BSGS(quick_pow(k - 1, n), x, mod);  //求还需要的最小行数ans
        printf("Case %d: %lld\n", ++cas, ans + maxr + 1);
    }
	return 0;
}