【Kickstart】2019 Round C - Catch Some

解法

假如最后必须返回终点，可以DP
每个颜色记录狗的位置，从小到大排序
对于f[i][s]表示，只考虑前i个颜色时，访问s个狗需要的最小时间
那么有:
f[i][s] = min(f[i-1][s], f[i-1][s-j-1]+2*P[i][j] for j in range(P[i])
但是现在需要考虑最后不回家，那么需要增加一维，f[i][s][0]表示只考虑前i个颜色并且没有选择最后颜色时，访问s个狗的最小时间
最后从f[i][k][1]里找最小值

注意位置可能重叠！！！！

#include 
#include 
#include 
#include 
#include 
//#include 
#include 
#include 
#include 
#include 

#define MAXN 1010
#define ZERO 10e-6
#define NINF -100000
#define INF 0x7fffffff


using namespace std;

typedef long long lld;

int n,k;
int A[MAXN], P[MAXN];
multiset<int> C[MAXN];
unordered_map<int,int> idx;
int f[MAXN][MAXN][2];

int min(int a,int b){
    return a<b?a:b;
}

int add(int a, int b) {
    if(a==INF || b==INF) return INF;
    return a+b;
}

int solve() {
    idx.clear();
    int m = 0;
    for (int i=0;i<n;++i) {
        if (idx.find(A[i]) == idx.end()) {
            idx[A[i]] = m++;
        }
    }
    for(int i=0;i<m;++i) C[i].clear();
    for (int i=0;i<n;++i) {
        C[idx[A[i]]].insert(P[i]);
    }
    for(int i=0;i<=m;++i)
        for(int s=0;s<=k;++s) {
            f[i][s][0] = INF;
            f[i][s][1] = INF;
        }
    f[0][0][0] = 0;
    f[0][0][1] = 0;
    int ans = INF;
    for(int i=1;i<=m;++i) {
        for(int s=0;s<=k;++s) {
            f[i][s][0] = f[i-1][s][0];
            f[i][s][1] = f[i-1][s][1];
            int cnt = 0;
            for(auto &p:C[i-1]) {
                cnt++;
                if (s<cnt) break;
                f[i][s][0] = min(f[i][s][0], add(f[i-1][s-cnt][0],2*p));
                f[i][s][1] = min(f[i][s][1], min(add(f[i-1][s-cnt][1],2*p),add(f[i-1][s-cnt][0],p)));
            }
//            printf(" (%d,%d)",f[i][s][0],f[i][s][1]);
        }
//        printf("\n");
        ans = min(ans, f[i][k][1]);
    }
    return ans;
}

int main() {
    int t;
    scanf("%d",&t);
    for(auto round=1;round<=t;++round) {
        scanf("%d%d",&n,&k);
        for(int i=0;i<n;++i) {
            scanf("%d",&P[i]);
        }
        for(int i=0;i<n;++i) {
            scanf("%d",&A[i]);
        }
        printf("Case #%d: %d\n",round, solve());
    }
}