【LOJ】#3051. 「十二省联考 2019」皮配

LOJ#3051. 「十二省联考 2019」皮配

当时我在考场上觉得这题很不可做。。。

当然，出了考场后再做，我还是没发现学校和城市是可以分开的，导致我还是不会

事实上，若一个城市投靠了某个阵营，学校可以任意选择派系，但是反过来看，学校选择了派系，也不影响城市投靠什么阵营，而这两者共同固定了一个学校选择的导师，所以对于k = 0的情况

我们设两个dp，\(g[i][j]\)表示考虑了前i个城市，去蓝阵营的人数为j，\(h[i][j]\)表示考虑了前i个城市，去鸭派系的人数为j，最后只需要把合法的分别乘起来就好

对于\(k!=0\)的情况，没有限制的城市，和没有限制的学校，都可以按照k = 0的方案去做

然而这些限制给的是什么呢，可以认为是选了一些蓝阵营的城市，将导致我们必须选鸭派系或者必须选R派系，选了红阵营亦然。这个时候两者就有关系了，所以设\(f[i][j][k]\)是考虑了前i个有限制的城市，选了蓝阵营的城市总人数为\(j\)在，在有限制的学校中选了鸭派系的人数是\(k\)

滚动数组就可以开的下了

然后答案合并的时候是会选择g和h一段连续的区间和\(f[i][j][k]\)合并，计算上下界查区间和即可

#include <bits/stdc++.h>
#define fi first
#define se second
#define pii pair<int,int>
#define mp make_pair
#define pb push_back
#define space putchar(' ')
#define enter putchar('\n')
#define eps 1e-10
#define ba 47
//#define ivorysi
#define MAXN 50010
#define MAXM 200005
using namespace std;
typedef long long int64;
typedef unsigned int u32;
typedef double db;
template<class T>
void read(T &res) {
    res = 0;T f = 1;char c = getchar();
    while(c < '0' || c > '9') {
    if(c == '-') f = -1;
    c = getchar();
    }
    while(c >= '0' && c <= '9') {
    res = res * 10 +c - '0';
    c = getchar();
    }
    res *= f;
}
template<class T>
void out(T x) {
    if(x < 0) {x = -x;putchar('-');}
    if(x >= 10) {
    out(x / 10);
    }
    putchar('0' + x % 10);
}
const int MOD = 998244353;
int n,c,k,C0,C1,D0,D1;
vector<int> city[1005];
int f[2505][305],g[2505],h[2505],tmp[305];
int b[1005],s[1005],p[1005],all;
bool lim[1005],dl[1005];
int inc(int a,int b) {
    return a + b >= MOD ? a + b - MOD : a + b;
}
int mul(int a,int b) {
    return 1LL * a * b % MOD;
}
void update(int &x,int y) {
    x = inc(x,y);
}
void Init() {
    read(n);read(c);
    read(C0);read(C1);read(D0);read(D1);
    for(int i = 1 ; i <= c ; ++i) city[i].clear();
    memset(lim,0,sizeof(lim));memset(dl,0,sizeof(dl));
    all = 0;
    for(int i = 1 ; i <= n ; ++i) {
    read(b[i]);read(s[i]);all += s[i];
    city[b[i]].pb(i);
    }
    read(k);
    int a;
    for(int i = 1 ; i <= k ; ++i) {
    read(a);read(p[a]);
    dl[a] = 1;lim[b[a]] = 1;
    }
}
void Solve() {
    memset(g,0,sizeof(g));
    g[0] = 1;
    for(int i = 1 ; i <= c ; ++i) {
    if(!lim[i] && city[i].size() > 0) {
        int sum = 0;
        for(auto t : city[i]) {
        sum += s[t];
        }
        for(int j = C0; j >= sum ; --j) {
        update(g[j],g[j - sum]);
        }
    }
    }
    memset(h,0,sizeof(h));
    h[0] = 1;
    for(int i = 1 ; i <= n ; ++i) {
    if(!dl[i]) {
        for(int j = D0 ; j >= s[i] ; --j) {
        update(h[j],h[j - s[i]]);
        }
    }
    }
    memset(f,0,sizeof(f));
    f[0][0] = 1;
    int tot = 0,nl = 0;
    for(int i = 1 ; i <= c ; ++i) {
    if(lim[i]) {
        int sum = 0,s1 = 0;
        for(auto t : city[i]) {
        sum += s[t];
        if(dl[t]) s1 += s[t];
        }
        tot = tot + sum;tot = min(tot,C0);
        nl = nl + s1;nl = min(nl,D0);
        for(int j = tot; j >= 0 ; --j) {
        for(int r = 0 ; r <= nl ; ++r) tmp[r] = f[j][r];
        for(auto t : city[i]) {
            if(dl[t]) {
            if(p[t] >= 2) {
                if(p[t] == 2) continue;
                else {
                for(int r = nl ; r >= 0 ; --r) {
                    if(r >= s[t]) tmp[r] = tmp[r - s[t]];
                    else tmp[r] = 0;
                }
                }
            }
            else {
                for(int r = nl ; r >= s[t] ; --r) update(tmp[r],tmp[r - s[t]]);
            }
            }
        }
        for(int r = 0 ; r <= nl ; ++r) f[j][r] = tmp[r];
        if(j >= sum) {
            for(int r = 0 ; r <= nl ; ++r) tmp[r] = f[j - sum][r];
            for(auto t : city[i]) {
            if(dl[t]) {
                if(p[t] < 2) {
                if(p[t] == 0) continue;
                else {
                    for(int r = nl ; r >= 0 ; --r) {
                    if(r >= s[t]) tmp[r] = tmp[r - s[t]];
                    else tmp[r] = 0;
                    }
                }
                }
                else {
                for(int r = nl ; r >= s[t] ; --r) update(tmp[r],tmp[r - s[t]]);
                }
            }
            }
            for(int r = 0 ; r <= nl ; ++r) update(f[j][r],tmp[r]);
        }
        
        }
    }
    }
    for(int i = 1 ; i <= C0 ; ++i) update(g[i],g[i - 1]);
    for(int i = 1 ; i <= D0 ; ++i) update(h[i],h[i - 1]);
    int ans = 0;
    for(int i = 0 ; i <= C0 ; ++i) {
    for(int j = 0 ; j <= 300 ; ++j) {
        int t0 = 0,t1 = 0;
        int u = C0 - i,d = all - C1 - i;
        if(d <= u) {
        t0 = g[u];
        if(d > 0) update(t0,MOD - g[d - 1]);
        }
        u = D0 - j,d = all - D1 - j;
        if(d <= u) {
        t1 = h[u];
        if(d > 0) update(t1,MOD - h[d - 1]);
        }
        update(ans,mul(f[i][j],mul(t0,t1)));
    }
    }
    out(ans);enter;
    cerr << ans;
}
int main(){
#ifdef ivorysi
    freopen("f1.in","r",stdin);
#endif
    int T;
    read(T);
    for(int i = 1 ; i <= T ; ++i) {
    Init();
    Solve();
    }
}

转载于:https://www.cnblogs.com/ivorysi/p/10996598.html