洛谷P1896 [SCOI2005]互不侵犯King

和前面的玉米田很接近,加了数量的要求。做的时候翻了两个ruozhi错误,一是估计数据范围很大,但没用longlong.二是没有仔细看k的范围,数组f的第三维开小了。

#include<iostream>
using namespace std;
const int maxn=512;
int n,K,num,c[maxn];
long long ans=0;
void count(){
    for(int i=0;i<512;i++){
        int j=i;
        while(j>0){	c[i]+=(j%2);j/=2;} 
    }
}
long long  f[10][512][82];
long long dp(){
    f[0][0][0]=1;
    num=(1<<n);
    for(int k=1;k<=K;k++) 
    for(int i=1;i<=n;i++){//处理每行 
        for(int j=0;j<num;j++){
            if(j&(j<<1))continue;//左边相邻,不做。 
            if(k-c[j]<0)continue;//放多了,不做             
            for(int l=0;l<num;l++){ 
                if((l&j)||((l<<1)&j)||((l>>1)&j))continue;//与上一行冲突 
                f[i][j][k]+=f[i-1][l][k-c[j]]; 
            } 
        }		
    }
    for(int i=1;i<=n;i++)
        for(int j=0;j<num;j++)ans+=f[i][j][K];
        return ans;
}
int main(){
    cin>>n>>K;
    count();
    cout<<dp();	
} 


猜你喜欢

转载自blog.csdn.net/lengxuenong/article/details/79150683
今日推荐