jzoj4916. 【GDOI2017模拟12.9】完全背包问题 (背包+最短路)

题面

题解

考场上蠢了……这么简单的东西都想不到……

首先排序加去重。

先来考虑一下,形如

\[a_1x_1+a_2x_2+...a_nx_n=w,a_1<a_2<...<a_n,x_i\geq 0\]

的方程该怎么判断是否有解

首先,把两边都放到模\(a_1\)的意义下,就是

\[a_2x_2+...a_nx_n\equiv w\pmod{a_1}\]

然后是一个神仙的转化,我们设有\(a_1\)个点,分别代表\(0,1,...,a_1-1\),对于每一个点\(u\),都连出去\(n-1\)条边,分别连向\((a+a_i)\%a_1\),代价为\(a_i\)。那么我们对这个图跑一遍最短路,如果\(dis_{w\%a_1}\leq w\),那么显然可以用这条最短路上的所有边加上若干个\(a_1\)构成\(w\),否则的话肯定不行

那么对于大于等于\(L\)的部分我们可以暴力\(dp\),设\(dp_{i,j}\)表示选了\(i\)个数,其中这\(i\)个数的和模\(a_1\)后为\(j\)\(a_1\)就是最小的数),这\(i\)个数的和最小是多少,直接大力转移即可。

最后用\(dp_{i,j}\)更新\(dis_j\),然后从每一个不为\(inf\)\(dis_j\)出发,以所有\(a_i< L\)为边,跑多源最短路就行了

ps:其实严格来说如果\(a_1\geq L\)的话会发现上面的做法是错的,不过错的只有最短路那部分,暴力\(dp\)还是正确的,所以忽视这一点也没问题

//minamoto
#include<bits/stdc++.h>
#define R register
#define ll long long
#define inf 0x3f3f3f3f
#define fp(i,a,b) for(R int i=a,I=b+1;i<I;++i)
#define fd(i,a,b) for(R int i=a,I=b-1;i>I;--i)
#define go(u) for(int i=head[u],v=e[i].v;i;i=e[i].nx,v=e[i].v)
template<class T>inline bool cmin(T&a,const T&b){return a>b?a=b,1:0;}
template<class T>inline bool cmax(T&a,const T&b){return a<b?a=b,1:0;}
using namespace std;
char buf[1<<21],*p1=buf,*p2=buf;
inline char getc(){return p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++;}
ll read(){
    R ll res,f=1;R char ch;
    while((ch=getc())>'9'||ch<'0')(ch=='-')&&(f=-1);
    for(res=ch-'0';(ch=getc())>='0'&&ch<='9';res=res*10+ch-'0');
    return res*f;
}
const int N=55,M=10005;
struct node{
    int u,d;
    node(){}
    node(R int U,R int D):u(U),d(D){}
    inline bool operator <(const node &b)const{return d>b.d;}
};priority_queue<node>q;
bitset<N*M>f[N];int a[N],vis[M];
int n,m,p,l,c,lim,u,v;ll w,dis[M],dp[N][M];
void mdzz(){
    memset(dis,0x3f,sizeof(dis));
    fp(i,0,a[1]-1){
        fp(j,0,c)cmin(dis[i],dp[j][i]);
        if(dis[i]!=inf)q.push(node(i,dis[i]));
    }
    while(!q.empty()){
        u=q.top().u,q.pop();
        if(vis[u])continue;vis[u]=1;
        fp(i,2,p)if(cmin(dis[v=(u+a[i])%a[1]],dis[u]+a[i]))q.push(node(v,dis[v]));
    }
}
int main(){
//  freopen("testdata.in","r",stdin);
    freopen("bag.in","r",stdin);
    freopen("bag.out","w",stdout);
    n=read(),m=read();
    fp(i,1,n)a[i]=read();
    sort(a+1,a+1+n),n=unique(a+1,a+1+n)-a-1;
    l=read(),c=read(),p=lower_bound(a+1,a+1+n,l)-a-1;
    memset(dp,0x3f,sizeof(dp));
    dp[0][0]=0;
    fp(i,p+1,n)fp(k,0,c-1)fp(j,0,a[1]-1){
        v=(j+a[i])%a[1];
        cmin(dp[k+1][v],dp[k][j]+a[i]);
    }
    mdzz();
    while(m--){
        w=read();
        puts(dis[w%a[1]]<=w?"Yes":"No");
    }
    return 0;
}

猜你喜欢

转载自www.cnblogs.com/bztMinamoto/p/10427712.html