CF1425D Danger of Mad Snakes 二维前缀和+组合数

题目链接

题意

1000*1000的矩阵中n条蛇，你有m个武器可以攻击蛇，武器只能放在有蛇的地方，每个武器有r的攻击距离，对于每种可能的武器放置方案，产生的攻击值是所有蛇的攻击力的和的平方，求所有方案的攻击值的和。

做法

显然我们不能单独考虑每种方案，方案数太多了。我们每次枚举两条蛇，计算包含这两条蛇的方案数，这两条蛇产生的贡献是总方案数*两条蛇的攻击力的积。包含两条蛇的总方案数 = 所有的方案数-不包含第一条蛇的方案，-不包含第二条蛇的方案+不包含两条蛇的方案。就是一个简单的容斥。

不包含某条蛇的方案=C（蛇的总数-可以攻击的到这条蛇的武器的数量，m）

不包含某两条蛇的方案=C（蛇的总数-只能攻击到其中一条蛇武器是数量+可以同时攻击到两条蛇的数量）。

求可能攻击到的某条蛇的武器的数量用二维前缀和维护一下就可以，注意可能不存在能同时攻击到两条蛇的武器，发现两个矩阵没有交点的时侯直接返回0.

代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
const int INF = 0x3f3f3f3f;
const double Pi = acos(-1);
namespace {
  template <typename T> inline void read(T &x) {
    x = 0; T f = 1;char s = getchar();
    for(; !isdigit(s); s = getchar()) if(s == '-') f = -1;
    for(;  isdigit(s); s = getchar()) x = (x << 3) + (x << 1) + (s ^ 48);
    x *= f;
  }
}
#define fio ios::sync_with_stdio(false);cin.tie(0);cout.tie(0);
#define _for(n,m,i) for (register int i = (n); i <  (m); ++i)
#define _rep(n,m,i) for (register int i = (n); i <= (m); ++i)
#define _srep(n,m,i)for (register int i = (n); i >= (m); i--)
#define _sfor(n,m,i)for (register int i = (n); i >  (m); i--)
#define lson rt << 1, l, mid
#define rson rt << 1 | 1, mid + 1, r
#define lowbit(x) x & (-x)
#define pii pair<int,int>
#define fi first
#define se second
const int N = 2e3+5;
const LL Mod = 1e9+7;
int sum[N][N];
int qry(int l, int r, int x,int y) {
  if(l > x || r > y) return 0;
  l = max(l, 1); r = max(r, 1);
  x = min(x, 1000); y = min(y, 1000);
  return sum[x][y] - sum[l-1][y] - sum[x][r-1] + sum[l-1][r-1];
}
int x[N], y[N], v[N];
int c[N][N];
int main() {
  int n, m, r; scanf("%d %d %d", &n, &m, &r);
  _for(0, n, i) scanf("%d %d %d", x + i, y + i, v + i), ++sum[x[i]][y[i]];
  _for(1, N, i) _for(1, N, j) sum[i][j] += sum[i-1][j] + sum[i][j-1] - sum[i-1][j-1];
  _rep(0, n, i) {
    c[i][0] = 1;
    _rep(1, i, j) c[i][j] = (c[i-1][j] + c[i-1][j-1]) % Mod;
  } 
  LL ans = 0;
  _for(0, n, i) _for(0, n, j) {
    int ni = qry(x[i]-r, y[i]-r, x[i]+r, y[i]+r); // 包含i 
    int nj = qry(x[j]-r, y[j]-r, x[j]+r, y[j]+r); // 包含j 
    int nij = qry(max(x[i], x[j])-r, max(y[i], y[j])-r, min(x[i], x[j])+r, min(y[i], y[j])+r); // 包含i和j 
    int num = (c[n][m] - c[n-ni][m] - c[n-nj][m] + c[n-ni-nj+nij][m]) % Mod;// 包含i，j的方案数 
    ans = (1ll * num * v[i] % Mod * v[j] + ans) % Mod;
  }
  if(ans < 0) ans += Mod;
  cout << ans << endl;
}