啊啊啊边界判错了搞死我了QAQ

这题是一个想起来很休闲写起来很恶心的背包

对于\(k=0\)的情况,可以发现选阵营和选派系是独立的,对选城市选阵营和学校选派系分别跑一遍01背包就行了

对于\(k>0\)的情况,设\(f[i][0/1][j][k]\)表示对于第\(i\)个有限制的学校,该学校选择\(0/1\)阵营时,\(C0\)阵营有\(j\)人,\(D0\)派系有\(k\)人的方案数

转移要分类讨论,有点麻烦,看代码吧

// luogu-judger-enable-o2
#include <bits/stdc++.h>
#define N 1010
#define M 2505
#define R register
#define rep(i,x,y) for(i=x;i<=y;++i)
#define des(i,x,y) for(i=x;i>=y;--i)
#define mod 998244353
using namespace std;

int f[2][2][M][M],pref0[M],preg0[M],f0[M],g0[M];
int C0,C1,D0,D1,c0[N],tmp2[N],sum2=0;
int cnts[2],cntc[2],n,c;
bool visct[N]; 

inline int add(int x,int y){ return (x+y>=mod)?x+y-mod:x+y; }
inline int sub(int x,int y){ return (x-y<0)?x-y+mod:x-y; }
struct hhw{ int b,s,id,lim; } tmp[N],a0[N],ak[N];
inline void rd(int &x){
    char c=getchar();R int y=0;
    while(c<'0'||c>'9') c=getchar();
    while(c>='0' && c<='9') y=y*10+c-'0',c=getchar();
    x=y;
}
inline void solve0(){
    R int i,j;
    f0[0]=pref0[0]=1;
    rep(i,1,cntc[0])
        des(j,C0-c0[i],0)
            f0[j+c0[i]]=add(f0[j+c0[i]],f0[j]);
    rep(i,1,C0) pref0[i]=add(pref0[i-1],f0[i]);
    g0[0]=preg0[0]=1;
    rep(i,1,cnts[0])
        des(j,D0-a0[i].s,0)
            g0[j+a0[i].s]=add(g0[j+a0[i].s],g0[j]);
    rep(i,1,D0) preg0[i]=add(preg0[i-1],g0[i]);
}

inline bool cmp(const hhw &x,const hhw &y){ return x.b<y.b; }

inline void addf(int a,int b,int c,int d,int x){
    if(c>C0||d>D0) return;
    f[a][b][c][d]=add(f[a][b][c][d],x);
}

inline void solvek(){
    f[0][0][0][0]=1;
    R int i,j,k,o,pos=1;
    sort(ak+1,ak+cnts[1]+1,cmp);
    rep(i,1,cnts[1]){
        rep(o,0,1)
            des(j,C0,0)
                des(k,sum2,0){//不能直接写D0!!!要把有限制的学校的人数统计出来!!!
                    f[pos][o][j][k]=0;
                    if(o==0){
                        if(ak[i].lim^0){
                            if(ak[i].b==ak[i-1].b)
                                addf(pos,o,j,k+ak[i].s,f[pos^1][o][j][k]);
                            else{
                                addf(pos,o,j+tmp2[ak[i].b],k+ak[i].s,f[pos^1][o][j][k]);
                                addf(pos,o,j+tmp2[ak[i].b],k+ak[i].s,f[pos^1][o^1][j][k]);
                            }
                        }
                        if(ak[i].lim^1){
                            if(ak[i].b==ak[i-1].b)
                                addf(pos,o,j,k,f[pos^1][o][j][k]);
                            else{
                                addf(pos,o,j+tmp2[ak[i].b],k,f[pos^1][o][j][k]);
                                addf(pos,o,j+tmp2[ak[i].b],k,f[pos^1][o^1][j][k]);
                            }
                        }
                    } else{
                        if(ak[i].lim^2){
                            if(ak[i].b==ak[i-1].b)
                                addf(pos,o,j,k+ak[i].s,f[pos^1][o][j][k]);
                            else{
                                addf(pos,o,j,k+ak[i].s,f[pos^1][o][j][k]);
                                addf(pos,o,j,k+ak[i].s,f[pos^1][o^1][j][k]);
                            }
                        }
                        if(ak[i].lim^3){
                            if(ak[i].b==ak[i-1].b)
                                addf(pos,o,j,k,f[pos^1][o][j][k]);
                            else{
                                addf(pos,o,j,k,f[pos^1][o][j][k]);
                                addf(pos,o,j,k,f[pos^1][o^1][j][k]);
                            }
                        }
                    }
                }
        pos^=1;
    }
}

inline int qwq(int x,int y){ return y>=0?sub(pref0[x],pref0[y]):pref0[x]; }
inline int owo(int x,int y){ return y>=0?sub(preg0[x],preg0[y]):preg0[x]; }

inline void solve(){
    R int i,x,y,k,j,sum=0,ans=0;
    rd(n),rd(c);
    rd(C0),rd(C1),rd(D0),rd(D1);
    rep(i,1,n){
        rd(tmp[i].b),rd(tmp[i].s);
        tmp[i].id=i,tmp[i].lim=-1;
        sum+=tmp[i].s;
        tmp2[tmp[i].b]+=tmp[i].s;
    }
    rd(k);
    rep(i,1,k){
        rd(x),rd(y),tmp[x].lim=y;
        visct[tmp[x].b]=1;sum2+=tmp[x].s;
    }
    rep(i,1,n){
        if(tmp[i].lim==-1) a0[++cnts[0]]=tmp[i];
        else ak[++cnts[1]]=tmp[i];
    }
    rep(i,1,c){
        if(!visct[i] && tmp2[i]) c0[++cntc[0]]=tmp2[i];
        else cntc[1]++;
    }
    if(sum-C0>C1 || sum-D0>D1){
        printf("0\n");
        return;
    }
    solve0(),solvek();
    rep(i,0,C0)
        rep(j,0,D0){
            if(sum-C1-i<=C0-i && sum-D1-j<=D0-j){
                int x=add(f[cnts[1]&1][0][i][j],f[cnts[1]&1][1][i][j]);
                ans=add(ans,1ll*x*qwq(C0-i,sum-C1-i-1)%mod*owo(D0-j,sum-D1-j-1)%mod);
            }
            f[0][0][i][j]=f[0][1][i][j]=f[1][0][i][j]=f[1][1][i][j]=0;
        }
    printf("%d\n",ans);
    rep(i,1,c) tmp2[i]=visct[i]=0;
    cntc[1]=cntc[0]=cnts[1]=cnts[0]=sum=sum2=0;
    memset(f0,0,sizeof(int)*(C0+1));
    memset(g0,0,sizeof(int)*(D0+1));
}

int main(){
    R int t;rd(t);
    while(t--) solve();
}

洛谷P5289 [十二省联考2019]皮配(01背包)的更多相关文章

  1. luogu P5289 [十二省联考2019]皮配

    传送门 首先考虑一个正常的dp,设\(f_{i,j,k}\)为前\(i\)个学校,\(j\)人在\(\color{#0000FF}{蓝阵营}\),\(k\)人在\(\color{#654321}{吔} ...

  2. luogu P5289 [十二省联考2019]皮配 背包

    LINK:皮配 我承认是一道很难的题目. 不过对于这道题 部分分的提示显得尤为重要. 首先是 40分的暴力dp 很容易想 但是不容易写. 从40分可以发现我们只需要把蓝阵营和鸭派系的人数给存在起来就行 ...

  3. 【BZOJ5498】[十二省联考2019]皮配(动态规划)

    [BZOJ5498][十二省联考2019]皮配(动态规划) 题面 BZOJ 洛谷 题解 先考虑暴力\(dp\),设\(f[i][j][k]\)表示前\(i\)所学校,有\(j\)人在某个阵营,有\(k ...

  4. 洛谷.5284.[十二省联考2019]字符串问题(后缀自动机 拓扑 DP)

    LOJ BZOJ 洛谷 对这题无话可说,确实比较...裸... 像dls说的拿拓扑和parent树一套就能出出来了... 另外表示BZOJ Rank1 tql... 暴力的话,由每个\(A_i\)向它 ...

  5. 洛谷.5283.[十二省联考2019]异或粽子(可持久化Trie 堆)

    LOJ 洛谷 考场上都拍上了,8:50才发现我读错了题=-= 两天都读错题...醉惹... \(Solution1\) 先求一遍前缀异或和. 假设左端点是\(i\),那么我们要在\([i,n]\)中找 ...

  6. 洛谷P5284 [十二省联考2019]字符串问题 [后缀树]

    传送门 思路 设\(dp_i\)表示以\(i\)结尾的\(A\)串,能达到的最长长度. 然后发现这显然可以\(i\)往自己控制的\(k\)连边,\(k\)往能匹配的\(j\)连边,就是个最长路,只要建 ...

  7. 洛谷 5291 [十二省联考2019]希望(52分)——思路+树形DP

    题目:https://www.luogu.org/problemnew/show/P5291 考场上写了 16 分的.不过只得了 4 分. 对于一个救援范围,其中合法的点集也是一个连通块. 2n 枚举 ...

  8. 洛谷P5284 [十二省联考2019]字符串问题(SAM+倍增+最长路)

    题面 传送门 题解 首先,我们把串反过来,那么前缀就变成后缀,建一个\(SAM\).我们发现一个节点的后缀是它的所有祖先 那么我们是不是直接按着\(parent\)树建边就可以了呢? 显然不是.我们假 ...

  9. Luogu5289 十二省联考2019皮配(动态规划)

    将选择导师看成先选阵营再选派系,这样有显然的O(nm2)暴力,即按城市排序后,设f[i][j][k]为前i个学校中第一个阵营有j人第一个派系有k人的方案数,暴力背包. 对于k=0,可以发现选阵营和选派 ...

随机推荐

  1. 【spring实战第五版遇到的坑】3.1中的例子报错

    按照书中的例子,一直做到第3.1章使用JDBC读写数据时,在提交设计的taco表单时,报了如下的异常信息: Failed to convert property value of type java. ...

  2. Yii2设计模式——单例模式

    应用举例 在Yii.php中: require __DIR__ . '/BaseYii.php'; // Yii框架的帮助类,提供框架基本的功能 class Yii extends \yii\Base ...

  3. Django学习之九: auth 认证组件

    目录 Django auth 认证组件 配置使用auth组件及其中间件 request.user 可以直接在template模版中使用 auth组件常用api 获取认证model类 认证检测装饰器@l ...

  4. css3自适应布局单位vw,vh你知道多少?

    视口单位(Viewport units) 什么是视口? 在桌面端,视口指的是在桌面端,指的是浏览器的可视区域:而在移动端,它涉及3个视口:Layout Viewport(布局视口),Visual Vi ...

  5. new会返回NULL空指针吗

    c++中的new会返回NULL空指针吗 https://stackoverflow.com/questions/3389420/will-new-operator-return-null On a s ...

  6. Android Studio Error:Execution failed for task ':app:preDebugAndroidTestBuild'.彻底解决的方法以及修改AScompileSDKVersion

    Error Error:Execution failed for task ':app:preDebugAndroidTestBuild'. > Conflict with dependency ...

  7. Android视频录制从不入门到入门系列教程(一)————简介

    一.WHY Android SDK提供了MediaRecorder帮助开发者进行视频的录制,不过这个类很鸡肋,实际项目中应该很少用到它,最大的原因我觉得莫过于其输出的视频分辨率太有限了,满足不了项目的 ...

  8. 从零学习Fluter(九):正式使用flutter进入商业开发

    一下为下个app 功能模块简要划分 接口文档需要说明 公告接口 messageTitle -- 消息内容 红包墙接口 使用说明 地理位置定位 shareGetMoney - 分享活动金额 lookCo ...

  9. SSM框架多数据源和AOP事务管理之间

  10. git常用命令总结--廖雪峰老师Git教程命令总结

    学习了廖雪峰老师的Git教程之后的命令总结,重点关于git和远程仓库的东西. 如果没有学过,这是传送门 下面这个图很重要 一.git初始化本地仓库和配置 echo "想输入到文件的内容,一般 ...