啊啊啊边界判错了搞死我了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. vue学习记录③(路由)

    上篇文章我们用vue-cli脚手架工具做了个简单的hello world页面,但是我们破坏了原来的流程,而正常的访问页面应该是通过路由来实现的. 那么什么是路由呢? 路由就是通过不同的url来访问不同 ...

  2. C# 文件绝对路径与相对路径的转换

    class Program { const string CONFIG_PATH = @"C:\SoftWare\Config.xml"; const string IMAGE_P ...

  3. asp.net动态添加控件学习

    看了老师的教程后,自己一点感悟记录下来: 1.在页面提交后,动态生成的控件会丢失, 但如果生成控件的代码在pageload中,就可以,原理是每次生成页面都执行生成. 2.动态按件或页面原来控件, 在页 ...

  4. CG-CTF simple-machine

    运行一下,输入flag: 用ida打开: input_length和input_byte_804B0C0为重命名的变量:现在一个个看调用的函数. sub_8048526(): 这个函数使用了mmap分 ...

  5. Jenkins 使用Tfs 插件出现 MappingConflictException 错误问题解决

    FATAL: com.microsoft.tfs.core.clients.versioncontrol.exceptions.MappingConflictException: The path C ...

  6. kvm热添加和热迁移

    a.热添加磁盘 1.创建磁盘 qemu-img create -f qcow2 web01-add01.qcow2 5G 2.附加磁盘设备 virsh attach-disk web01 /opt/w ...

  7. Exception in thread "main" org.I0Itec.zkclient.exception.ZkAuthFailedException: Authentication failure is thrown while creating kafka topic

    Exception in thread "main" org.I0Itec.zkclient.exception.ZkAuthFailedException: Authentica ...

  8. ABP之模块系统

    简介 ASP.NET Boilerplate提供了构建模块的基础结构,并将它们组合在一起以创建应用程序. 模块可以依赖于另一个模块. 通常,一个程序集被视为一个模块. 如果创建具有多个程序集的应用程序 ...

  9. WiFi攻击中“核武器”

    3·15晚会上,央视曝光了WiFi探针盒子通过手机MAC地址.大数据匹配获取手机用户个人信息的典型案例. 其中,曝光的“声牙科技有限公司”号称有全国6亿手机用户的个人信息,包括手机号,只要将获取到的手 ...

  10. 遍历HashMap的四种方式

    转至:https://www.cnblogs.com/Berryxiong/p/6144086.html public static void main(String[] args) { Map< ...