uva 11916 大步小步算法

// UVa11916 Emoogle Grid
// Rujia Liu
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<set>
#include<map>
using namespace std;

const int MOD = 100000007;
const int maxb = 500 + 10;
int n, m, k, b, r, x[maxb], y[maxb];
set<pair<int, int> > bset;

int pow_mod(int a, long long p) {
  if(p == 0) return 1;
  int ans = pow_mod(a, p/2);
  ans = (long long)ans * ans % MOD;
  if(p%2) ans = (long long)ans * a % MOD;
  return ans;
}

int mul_mod(int a, int b) {
  return (long long)a * b % MOD;
}

int inv(int a) {
  return pow_mod(a, MOD-2);
}

int log_mod(int a, int b) {
  int m, v, e = 1, i;
  m = (int)sqrt(MOD);
  v = inv(pow_mod(a, m));
  map <int,int> x;
  x[1] = 0;
  for(i = 1; i < m; i++){ e = mul_mod(e, a); if (!x.count(e)) x[e] = i; }
  for(i = 0; i < m; i++){
    if(x.count(b)) return i*m + x[b];
    b = mul_mod(b, v);
  }
  return -1;
}

// 计算可变部分的方案数
int count() {
  int c = 0; // 有k种涂法的格子数
  for(int i = 0; i < b; i++) {
    if(x[i] != m && !bset.count(make_pair(x[i]+1, y[i]))) c++; // 不可涂色格下面的可涂色格
  }
  c += n; // 第一行所有空格都有k种涂法
  for(int i = 0; i < b; i++)
    if(x[i] == 1) c--; // 扣除那些不能涂色的格子

  // ans = k^c * (k-1)^(mn - b - c)
  return mul_mod(pow_mod(k, c), pow_mod(k-1, (long long)m*n - b - c));
}

int doit() {
  int cnt = count();
  if(cnt == r) return m; // 不变部分为空

  int c = 0;
  for(int i = 0; i < b; i++)
    if(x[i] == m) c++; // 可变部分第一行中有k种涂法的格子数
  m++; // 多了一行（可变部分的第一行）
  cnt = mul_mod(cnt, pow_mod(k, c));
  cnt = mul_mod(cnt, pow_mod(k-1, n - c));
  if(cnt == r) return m; // 此时cnt为不变部分和可变部分第一行的方案总数

  return log_mod(pow_mod(k-1,n), mul_mod(r, inv(cnt))) + m;
}

int main() {
  int T;
  scanf("%d", &T);
  for(int t = 1; t <= T; t++) {
    scanf("%d%d%d%d", &n, &k, &b, &r);
    bset.clear();
    m = 1;
    for(int i = 0; i < b; i++) {
      scanf("%d%d", &x[i], &y[i]);
      if(x[i] > m) m = x[i]; // 更新不变部分的行数
      bset.insert(make_pair(x[i], y[i]));
    }
    printf("Case %d: %d\n", t, doit());
  }
}

书上的代码照旧：

题目大意：有一问题，在M行N列的网格上涂K种颜色，其中有B个格子不用涂色，其它每个格子涂一种颜色，同一列的上下两个相邻的格子不能涂相同的颜色。给出M，N，K和B个格子的位置，求出总方案数模掉1e8+7的结果R。现在已知R，求最小的M。


解题思路：有确定不用涂色格子的区域作为不变部分，总数通过计算为tmp，外加可变部分的第一行，方案数为cnt，可变部分除第一行外，每加一行都将总数乘以 ( − 1 ) ,既有


c n ∗ P = R mod ( 1 e 8 + 7 )
P = c n − 1 ∗ R mod ( 1 e 8 + 7 )


就是大步小步算法求M。