CodeChef:Little Elephant and Colored Coins

类似墨墨的等式

设f[2][j][k]表示a[i].c是否和当前颜色相同，到当前枚举到的颜色为止，颜色数为j，对mnv取模为k的最小数

这是个无限循环背包，用spfa优化

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
#include<map>
using namespace std;
typedef long long LL;
const int _=1e2;
const int maxn=+;
const int maxv=*1e5+_;
LL inf;

LL f[][maxn][maxv];//a[i].c是否和当前颜色相同，到当前枚举到的颜色为止，颜色数为j，对mnv取模为k的最小数
struct SPFA
{
    int op,j,k;
    SPFA(){} SPFA(int OP,int J,int K){op=OP,j=J,k=K;}
}list[*maxn*maxv];int head,tail;bool v[][maxn][maxv];
void spfa(int cc,int mnv,int dv)
{
    head=,tail=; list[tail++]=SPFA(,,);
    for(int j=;j<=cc;j++)
        for(int k=;k<mnv;k++)
        {
            if(f[][j][k]!=inf)list[tail++]=SPFA(,j,k),v[][j][k]=true;
            if(f[][j][k]!=inf)list[tail++]=SPFA(,j,k),v[][j][k]=true;
        }
    while(head!=tail)
    {
        int op=list[head].op,j=list[head].j,k=list[head].k;
        int tp=,tj=j+-op,tk=(k+dv)%mnv;
        if(f[tp][tj][tk]>f[op][j][k]+dv)
        {
            f[tp][tj][tk]=f[op][j][k]+dv;
            if(v[tp][tj][tk]==false)
            {
                v[tp][tj][tk]=true;
                list[tail]=SPFA(tp,tj,tk);
                tail++;if(tail==*maxn*maxv)tail=;
            }
        }
        v[op][j][k]=false;
        head++;if(head==*maxn*maxv)head=;
    }
}

struct node{int v,c;}a[maxn];
bool cmp(node n1,node n2){return n1.c<n2.c;}
int main()
{
    freopen("a.in","r",stdin);
    freopen("a.out","w",stdout);
    int n,mnv=(<<);
    scanf("%d",&n);
    for(int i=;i<=n;i++)
        scanf("%d%d",&a[i].v,&a[i].c),mnv=min(mnv,a[i].v);
    sort(a+,a+n+,cmp);

    int cc=;
    memset(f,,sizeof(f));inf=f[][][];
    f[][][]=;
    for(int i=;i<n;i++)//开始放第i+1种
    {
        if(a[i+].c!=a[i].c)
        {
            for(int j=;j<=cc;j++)
                for(int k=;k<mnv;k++)
                    f[][j][k]=min(f[][j][k],f[][j][k]);
        }
        spfa(cc,mnv,a[i+].v);
        if(a[i+].c!=a[i].c)cc++;
    }
    for(int j=;j<=cc;j++)
        for(int k=;k<mnv;k++)
            f[][j][k]=min(f[][j][k],f[][j][k]);

    int Q,u;LL x;
    scanf("%d",&Q);
    while(Q--)
    {
        scanf("%lld",&x);u=x%mnv;
        bool bk=false;
        for(int i=n;i>=;i--)
            if(f[][i][u]<=x){printf("%d\n",i);bk=true;break;}
        if(!bk)puts("-1");
    }

    return ;
}