[BZOJ5010][FJOI2017]矩阵填数(状压DP)
5010: [Fjoi2017]矩阵填数
Time Limit: 10 Sec Memory Limit: 128 MB
Submit: 90 Solved: 45
[Submit][Status][Discuss]Description
给定一个 h*w 的矩阵,矩阵的行编号从上到下依次为 1..h，列编号从左到右依次1..w。在这个矩阵中你需要在每个格子中填入 1..m 中的某个数。给这个矩阵填数的时候有一些限制，给定 n 个该矩阵的子矩阵,以及该子矩阵的最大值 v,要求你所填的方案满足该子矩阵的最大值为 v。现在,你的任务是求出有多少种填数的方案满足 n 个限制。两种方案是不一样的当且仅当两个方案至少存在一个格子上有不同的数。由于答案可能很大，你只需要输出答案 对 1,000,000,007 的取模即可。Input
输入数据的第一行为一个数 T，表示数据组数。对于每组数据，第一行为四个数 h,w,m,n。接下来 n 行，每一行描述一个子矩阵的最大值 v。每行为五个整数 x1,y1,x2,y2,v，表示一个左上角为(x1,y1),右下角为(x2,y2)的子矩阵的最大值为 v ( 1≤x1≤x2≤h, 1≤y1≤y2≤w)T≤5,1≤h,w,m≤10000,1≤v≤m,1≤n≤10Output
对于每组数据输出一行，表示填数方案 mod 1,000,000,007 后的值。Sample Input
2
3 3 2 2
1 1 2 2 2
2 2 3 3 1
4 4 4 4
1 1 2 3 3
2 3 4 4 2
2 1 4 3 2
1 2 3 4 4Sample Output
28
76475HINT
Source
容斥原理
先离散化，对于每一块，可以通过限制得出这一块最大值是多少。
答案=每个条件的子矩形最大值都小于等于这个条件的v - 有一个条件的子矩形最大值小于这个条件的v + 有两个条件的子矩形最大值小于这个条件的v - ...
我们可以O(2^n)枚举每个限制条件是否满足，然后可以根据表示这个状态的二进制数中1的个数确定这个状态的方案数贡献是正是负。
还有一个就是离散化的问题，用左闭右开。
代码用时：1h
1 #include<cstdio>
2 #include<cstring>
3 #include<algorithm>
4 #define rep(i,l,r) for (int i=l; i<=r; i++)
5 using namespace std;
6
7 const int N=25,md=1000000007;
8 int h,w,m,n,T,xx[N],yy[N],vv[N],mp[N][N],re[N][N],cnt[N],lx,ly,lv;
9 struct mat{ int x1,x2,y1,y2,v; }a[N];
10 int ksm(int a,int b){
11 int res;
12 for (res=1; b; a=(1ll*a*a)%md,b>>=1)
13 if (b & 1) res=(1ll*res*a)%md;
14 return res;
15 }
16
17 int main(){
18 freopen("bzoj5010.in","r",stdin);
19 freopen("bzoj5010.out","w",stdout);
20 for (scanf("%d",&T); T--; ){
21 scanf("%d%d%d%d",&h,&w,&m,&n);
22 xx[1]=0; xx[lx=2]=h; yy[1]=0; yy[ly=2]=w; vv[lv=1]=m;
23 rep(i,1,n){
24 scanf("%d%d%d%d%d",&a[i].x1,&a[i].y1,&a[i].x2,&a[i].y2,&a[i].v);
25 a[i].x1--; a[i].y1--;
26 xx[++lx]=a[i].x1; xx[++lx]=a[i].x2;
27 yy[++ly]=a[i].y1; yy[++ly]=a[i].y2;
28 vv[++lv]=a[i].v; vv[++lv]=a[i].v-1;
29 }
30 sort(xx+1,xx+lx+1); sort(yy+1,yy+ly+1); sort(vv+1,vv+lv+1);
31 lx=unique(xx+1,xx+lx+1)-xx-1; ly=unique(yy+1,yy+ly+1)-yy-1; lv=unique(vv+1,vv+lv+1)-vv-1;
32 rep(i,1,lx) rep(j,1,ly) re[i][j]=(xx[i]-xx[i-1])*(yy[j]-yy[j-1]);
33 rep(i,1,n){
34 a[i].x1=lower_bound(xx+1,xx+lx+1,a[i].x1)-xx;
35 a[i].x2=lower_bound(xx+1,xx+lx+1,a[i].x2)-xx;
36 a[i].y1=lower_bound(yy+1,yy+ly+1,a[i].y1)-yy;
37 a[i].y2=lower_bound(yy+1,yy+ly+1,a[i].y2)-yy;
38 a[i].v=lower_bound(vv+1,vv+lv+1,a[i].v)-vv;
39 }
40 int ans=0;
41 for (int s=0; s<(1<<n); s++){
42 rep(j,1,lx) rep(k,1,ly) mp[j][k]=lv;
43 int now=1;
44 rep(i,1,n){
45 int v=a[i].v;
46 if ((s>>(i-1))&1) now=-now,v--;
47 rep(j,a[i].x1+1,a[i].x2) rep(k,a[i].y1+1,a[i].y2) mp[j][k]=min(mp[j][k],v);
48 }
49 memset(cnt,0,sizeof(cnt));
50 rep(i,1,lx) rep(j,1,ly) cnt[mp[i][j]]+=re[i][j];
51 rep(i,1,lv) if (cnt[i]) now=1ll*now*ksm(vv[i],cnt[i])%md;
52 ans=(ans+now)%md;
53 }
54 printf("%d\n",(ans+md)%md);
55 }
56 return 0;
57 }
