牛客练习赛1

A矩阵

第一感觉是二维前缀和，但是因为字符的大小不同，最终前缀和的值的表示有很大可能发生冲突，所以，我们改变策略，用二维前缀和hash

代码如下

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <unordered_set>

using namespace std;

typedef unsigned long long ULL;

const int N = 510, base = 131;

int n, m;
ULL h[N][N], p[N * N];
char str[N];
unordered_set<ULL> S;

ULL calc(ULL f[], int l, int r) {
  return f[r] - f[l - 1] * p[r - l + 1];
}

bool check(int len) {
    S.clear();
    for (int j = len; j <= m; j ++ ) {
      int l = j - len + 1, r = j;
      ULL s = 0;
      for (int i = 1; i <= n; i ++ ) {
        s = s * p[len] + calc(h[i], l, r);
        if (i > len) s -= calc(h[i - len], l, r) * p[len * len];//如果行数大于len，就要将头的减去，头进行了len次len倍的扩大
        if (i >= len) {
          if (S.count(s)) return true;//如果之前出现过，就true
          S.insert(s);
        }
      }
    }
    return false;
}

int main() {
  ios::sync_with_stdio(false); cin.tie(0);

  cin >> n >> m;

  p[0] = 1;
  for (int i = 1; i <= n * m; i ++ ) p[i] = p[i - 1] * base;
  for (int i = 1; i <= n; i ++ ) {
    cin >> str + 1;
    for (int j = 1; j <= m; j ++ ) h[i][j] = h[i][j - 1] * base + str[j] - 'a' + 1;//记录的是每一行自己的一维前缀和hash值
  }
  //二分一下答案
  int l = 1, r = min(n, m);
  while (l < r) {
    int mid = (r + l + 1) >> 1;
    if (check(mid)) l = mid;
    else r = mid - 1;
  }

  cout << l << endl;

  return 0;
}

树

我们用k中燃料将n个节点进行染色，但是，只能怪染一块，意思就是只能将相邻的节点染色，不相邻的节点颜色一定不同。
这也就是我们题目所说颜色相同的点x与y，他们之间的所有点都必须颜色相同。
这个题可以从两个方向考虑。
法一1.我们对这个树拆边，将树分成不超过k个连通块。
将n个节点分成i块（n个节点被i中颜色染上）
则我们要从n-1个边中截掉i-1条边
即：

从n-1条边里，选出i条边进行染色，而且染色时有顺序关系
法二2.用DP来做dp[i][j]表示，染了i个节点，使用了j中颜色
当我们染到第i个节点时，我们面临两种选择
1.我们接着使用上一次的染料f[i][j] = f[i-1][j]
2.我们使用新的染料，在剩下的染料中选一种f[i][j] = f[i-1][j - 1]*(k-j+1)
所以，f[i][j] = f[i-1][j-1] * (k - j + 1) + f[i - 1][j]

法一

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

typedef long long LL;

const int N = 310, mod = 1e9 + 7;

int n, k;
int g[N][N];
LL f[N][N];
LL c[N][N];
LL a[N];

int main() {
  cin >> n >> k;

  for (int i = 1; i < n; i ++ ) {
    int x, y; cin >> x >> y;
    g[x][y] = 1;
  }

  for (int i = 0; i <= N; i ++ )
    for (int j = 0; j <= i; j ++ )
      if (!j) c[i][j] = 1;
      else c[i][j] = (c[i - 1][j - 1] + c[i - 1][j]) % mod;

  a[0] = 1;
  for (int i = k, j = 1; i >= 1; i -- , j ++)
      a[j] = (a[j - 1] * i) % mod;

  LL ans = 0;
  for (int i = 1; i <= min(n, k); i ++ ) {
    ans = (ans + (c[n - 1][i - 1] * a[i]) % mod) % mod;
  }

  cout << ans << endl;

  return 0;
}

法二

#include <iostream>
#include <c
stdio>
#include <cstring>
#include <algorithm>

using namespace std;

typedef long long LL;

const int N = 310, mod = 1e9 + 7;

int n, k;
int g[N][N];
LL f[N][N];

int main() {
  cin >> n >> k;

  for (int i = 1; i < n; i ++ ) {
    int x, y; cin >> x >> y;
    g[x][y] = 1;
  }

  f[0][0] = 1;
  for (int i = 1; i <= n; i ++ )
    for (int j = 1; j <= k; j ++ ) {
      f[i][j] = (f[i - 1][j] + f[i - 1][j - 1] * (k - j + 1)) % mod;
    }

  LL ans = 0;
  for (int i = 1; i <= k; i ++ )
      ans = (ans + f[n][i]) % mod;

  cout << ans << endl;
  return 0;
}