码迷,mamicode.com
首页 > 其他好文 > 详细

【bzoj4596】黑暗前的幻想乡

时间:2018-06-25 22:46:58      阅读:208      评论:0      收藏:0      [点我收藏+]

标签:代码   容斥   xxx   else   math   div   c++   符号   add   

Portal -->bzoj4596

Solution

  这题的话。。因为\(N\)比较小啊所以我们可以大力容斥(尽管实际算下来复杂度有点爆炸不过实测是能过的qwq)

  枚举包含了哪些颜色的边,每次重新填矩阵然后矩阵树定理高消求一波行列式然后乘上个容斥系数加到答案里面去就好了

  关于容斥原理这个东西。。(怎么感觉快忘光了)

  其实也不用想太多,反正是从最终状态(包含所有的,在这里是包含\(n-1\)种)开始往后推符号为一正一负就好了:
\[ ans=ans_{n-1}-ans_{n-2}+ans_{n-3}... \]
  然后代码大概长这个样子

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int N=20,MOD=1e9+7;
struct xxx{
    int x,y,nxt;
}a[100010];
int A[N][N],h[N],ok[N];
int n,m,tot,ans;
void add(int x,int y,int col);
int solve(int n);
int Abs(int x){return x>0?x:-x;}
bool in(int st,int x){return st>>(x-1)&1;}
int work();

int main(){
#ifndef ONLINE_JUDGE
    freopen("a.in","r",stdin);
#endif
    int x,y;
    scanf("%d",&n);
    memset(h,-1,sizeof(h));
    tot=0;
    for (int i=1;i<n;++i){
        scanf("%d",&m);
        for (int j=1;j<=m;++j){
            scanf("%d%d",&x,&y);
            add(x,y,i);
        }
    }
    ans=0;
    int all=1<<n-1,cnt,tmp,n1;
    for (int i=1;i<all;++i){
        memset(A,0,sizeof(A));
        cnt=0;
        for (int j=1;j<n;++j)
            if (in(i,j)){
                for (int k=h[j];k!=-1;k=a[k].nxt){
                    x=a[k].x; y=a[k].y;
                    ++A[x][x]; ++A[y][y];
                    --A[x][y]; --A[y][x];
                }
            }
            else ++cnt;
        tmp=solve(n-1);
        if (cnt&1) 
            ans=(ans-tmp+MOD)%MOD;
        else 
            ans=(ans+tmp)%MOD;
    }
    printf("%d\n",ans);
}

void add(int x,int y,int col){
    a[++tot].x=x; a[tot].y=y; a[tot].nxt=h[col]; h[col]=tot;
}

int solve(int n){
    int id,ret=1,tmp;
    for (int i=1;i<=n;++i){
        for (id=i;id<=n;++id)
            if (A[id][i]) break;
        if (id>n) continue;
        if (id!=i){
            ret=-ret;
            for (int j=i+1;j<=n;++j) swap(A[i][j],A[id][j]);
        }
        for (int j=i+1;j<=n;++j){
            while (A[j][i]){
                tmp=A[j][i]/A[i][i];
                for (int k=1;k<=n;++k)
                    A[j][k]=(1LL*A[j][k]+MOD-1LL*tmp*A[i][k]%MOD)%MOD;
                if (A[j][i]==0) break;
                ret=-ret;
                for (int k=1;k<=n;++k)
                    swap(A[j][k],A[i][k]);
            }
        }
    }
    for (int i=1;i<=n;++i)
        ret=1LL*ret*A[i][i]%MOD;
    return (ret+MOD)%MOD;
}

【bzoj4596】黑暗前的幻想乡

标签:代码   容斥   xxx   else   math   div   c++   符号   add   

原文地址:https://www.cnblogs.com/yoyoball/p/9226226.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!