UVA 10817 十一 Headmaster's Headache

Headmaster's Headache

Time Limit:3000MS     Memory Limit:0KB     64bit IO Format:%lld & %llu

Submit Status Practice UVA 10817

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
const int inf=1e9+7;
const int mm=65536;

int dp[105][65590];
int appcost[105],appnum[105][30];

int main()
{
    int s,n,m,x,y;
    int i,j,k;
    int sum0,k0;
    char a[103];
    while(scanf("%d %d %d",&s,&m,&n)!=EOF)
    {
        if(s==0)
            break;
        memset(appnum,0,sizeof(appnum));
        
        sum0=0;k0=0;
        for(i=1;i<=m;i++)
        {
            scanf("%d",&x);
            sum0=sum0+x;
            gets(a);
            for(j=0;a[j]!='';j++)
            {
                if('1'<=a[j] && a[j]<='9')
                {
                    y=a[j]-'0';y--;
                    if(!((k0>>(2*y)) & 1))
                    {
                        k0=k0 | (1<<(2*y));
                    }
                    else if(!((k0>>(2*y+1)) & 1))
                    {
                        k0=k0 | (1<<(2*y+1));
                    }
                }
            }
        }
        for(i=1;i<=n;i++)
        {
            scanf("%d",&appcost[i]);
            gets(a);
            int num=0;
            for(j=0;a[j]!='';j++)
            {
                if('1'<=a[j] && a[j]<='9')
                {
                    //y=a[j]-'0';y--;
                    num++;
                    appnum[i][num]=a[j]-'0';
                }
            }
            appnum[i][0]=num;
        }

        int S=1<<(2*s);
        for(i=0;i<S;i++)
        {
            dp[0][i]=inf;
        }
        dp[0][k0]=sum0;

        for(i=1;i<=n;i++)
        {
            for(j=0;j<S;j++)
            {
                dp[i][j]=dp[i-1][j];
            }

            for(j=0;j<S;j++)
            {
                if(dp[i-1][j]<inf)
                {
                    k=j;
                    for(int l=1;l<=appnum[i][0];l++)
                    {
                        y=appnum[i][l]-1;
                        if(!((k>>(2*y)) & 1))
                        {
                            k=k | (1<<(2*y));
                        }
                        else if(!((k>>(2*y+1)) & 1))
                        {
                            k=k | (1<<(2*y+1));
                        }
                    }
                    dp[i][k]=min(dp[i][k],dp[i-1][j]+appcost[i]);
                }
            }
        }
        printf("%d
",dp[n][(1<<(2*s))-1]);
    }
    return 0;
}