Codeforces Round #510 (Div. 2) B. Vitamins

B. Vitamins

题目链接：https://codeforces.com/contest/1042/problem/B

题意：

给出几种药，没种可能包含一种或多种（最多三种）维生素，现在问要吃到这三种维生素买药最少花费是多少。

题解：

嗯...可以直接暴力：

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N = 2e5+;
int n;
int c[N];
char s[N][];
int mn[];
int main(){
    scanf("%d",&n);
    for(int i=;i<;i++) mn[i]=;
    string s1="BA",s2="AB",s3="AC",s4="CA";
    for(int i=;i<=n;i++){
        scanf("%d %s",&c[i],s[i]);
        if(strlen(s[i])==){
            if(s[i][]=='A') mn[]=min(mn[],c[i]);
            else if(s[i][]=='B') mn[]=min(mn[],c[i]);
            else mn[]=min(mn[],c[i]);
        }else if(strlen(s[i])==){
            if(s[i]==s1 || s[i]==s2)
                mn[]=min(mn[],c[i]);
            else if(s[i]==s3 || s[i]==s4){
                mn[]=min(mn[],c[i]);
            }else{
                mn[]=min(mn[],c[i]);
            }
        }else{
            mn[]=min(mn[],c[i]);
        }
    }
    int ans = ;
    ans=min(ans,mn[]+mn[]+mn[]);
    ans=min(ans,mn[]+mn[]);
    ans=min(ans,mn[]+mn[]);
    ans=min(ans,mn[]+mn[]);
    for(int i=;i<=;i++){
        for(int j=;j<=;j++){
            if(i==j) continue ;
            ans=min(ans,mn[i]+mn[j]);
        }
    }
    ans=min(ans,mn[]);
    if(ans==) cout<<"-1";
    else cout<<ans;
    return ;
}

但是还有一种更简单的方法：dp，利用位运算来做。

代码如下：

#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
typedef long long ll;
const int N = ;
int dp[],a[N];
char s[N];
int n;
int main(){
    scanf("%d",&n);
    memset(dp,INF,sizeof(dp));
    dp[]=;
    for(int i=;i<=n;i++){
        scanf("%d",&a[i]);
        scanf("%s",s);
        int len = strlen(s),x=;
        for(int j=;j<len;j++) x|=(<<(s[j]-'A'));
        for(int j=;j<;j++) dp[x|j]=min(dp[x|j],dp[j]+a[i]);
    }
    if(dp[]==INF) cout<<-;
    else cout<<dp[];
    return ;
}