hihoCoder1476 矩阵计数 容斥原理

题意

• 参见http://hihocoder.com/problemset/problem/1476

思路

• 最开始我用的是dp求解的这个题，后来听大神说可以容斥原理来解，就又想了想求解了一下~
• dp的思路，详见：
  http://blog.csdn.net/luke2834/article/details/63675388
• 这题容斥的基本思想是，全部是白块的矩形数，减去因一块黑块而减少的矩形数，加上因两块黑块而减少的矩形数，以此类推。即枚举到奇数个黑块时减，偶数个黑块时加。
• 不难发现如果都是白块的矩形数为n * (n+1) / 2 * m * (m+1) / 2
• 接下来就是如何找到因k个黑块而减少的矩形数了
• 首先明确一个事实，一个矩形被它的左上角方块和右下角方块唯一确定
• 先考虑只有一个黑块的情况，可以发现，因这个黑块，整个矩形被分为了四个部分，即黑块的左上集合s1，右上s2，左下s3，右下s4。可以发现被这个黑块影响而不能构成矩形的是左上角方块位于s1，右下角方块位于s4的矩形，那么用s1的方块个数*s4的方块个数就可以得到我们所求的了。
• 再考虑有多个黑块的情况，实际上被这多个黑块共同影响的矩形的左上角位于所有小黑块s1的交集，右下角位于所有小黑块s4的交集，交集通过对x和y分别取最大最小很容易得到。

实现

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int inf = 0x3f3f3f3f;
vector<int> xp,yp;
int main(){
    int n,m,k;
    cin>>n>>m>>k;
    for (int i=0;i<k;i++){
        int x,y;
        cin>>x>>y;
        xp.push_back(x);
        yp.push_back(y);
    }
    ll ans = (ll)(n+1) * n * m * (m+1) / 4;
    for (int st=1;st<(1<<k);st++){
        int s = st;
        int minx=inf, miny=inf, maxx=0, maxy=0;
        int i = 0, cnt = 0;
        while (s > 0){
            if (s & 1){
                minx = min (minx, xp[i]);
                maxx = max (maxx, xp[i]);
                miny = min (miny, yp[i]);
                maxy = max (maxy, yp[i]);
                cnt++;
            }
            i++;
            s>>=1;
        }
        ll sum = (ll)minx * miny * (n - maxx + 1) * (m - maxy + 1);
        if (cnt & 1){
            ans -= sum;
        }
        else{
            ans += sum;
        }
    }
    cout << ans << endl;
    return 0;
}