【GDKOI2015】星球杯

题目

给出n位选手的两轮得分，以及所属类别的信息

安排每位选手分别参加哪一轮比赛

找出最佳方案，使得1国的选手得分总和最高

分析

利用反证法可得，必然存在一种最优方案，使得N个选手都进入决赛。
故可用二维dp[i][j]表示前i名选手，有j名在第一轮预赛晋级，i－j名选手在第二轮预赛晋级所得的最大值。
时间复杂度O（N ^ 2）
枚举第一轮排名为第k名的选手（分数为x），则必存在一种最优方案有：
1. 对于分数少于x的0国选手，都安排在第一轮；
2.对于分数少于x的1国选手，都安排在第二轮。
1. 第一轮分数低于x的1国选手对答案无贡献，所以我们先筛除这部分选手；
2. 剩下：1国选手 和 第一轮分数不低于x的0国选手；
3. 对于剩余部分的选手，按照第二轮分数从高到低排序
设F[i][j] : 在前 i 个选手中，有 j 个选手分配到第一轮预赛并排名在前k位
对于0国选手（第一轮得分必不小于x）：
a. 分配到第1轮(j < k)：F[i+1][j+1] = max(F[i+1][j+1], F[i][j]);
b.分配到第2轮：F[i+1][j] = max(F[i+1][j], F[i][j]);
对于1国的选手：
a.第一轮分数小于x：
若i－j < k, F[i+1][j] ＝max(F[i+1][j], F[i][j] + SecRound[i]);
b.第一轮分数大于等于x：
分配第一轮(j < k):
F[i+1][j+1]=max(F[i+1][j+1], F[i][j] + FirRound[i])；
分配第二轮：
a) 当 i - j < k 时, F[i+1][j] = max(F[i+1][j], F[i][j] + SecRound[i]);
b)否则，F[i+1][j] = max(F[i+1][j], F[i][j]);

代码

#include<cstdio>
#include<cstring>
#include<algorithm>

using namespace std;

const int MAXN = 205;

int F[MAXN][MAXN],A[MAXN],B[MAXN],Bel[MAXN],Ord[MAXN],N,K;

bool cmp(int a,int b) {return B[a] > B[b];}

int max(int a,int b) 
{
    if (a == -1) return b;if (b == -1) return a;
    return a > b ? a : b;
}

int main()
{
    scanf("%d%d", &N, &K);
    for(int i = 1;i <= N;i ++) scanf("%d%d%d", &A[i], &B[i], &Bel[i]);
    int Ans = 0;
    for(int i = 1;i <= N;i ++)
    {
        int cnt = 0,X = A[i];
        for(int j = 1;j <= N;j ++) 
        if (j != i)
        {
            if (Bel[j]) Ord[++ cnt] = j; else
            if (A[j] > A[i]) Ord[++ cnt] = j;
        }
        sort(Ord + 1,Ord + cnt + 1,cmp);
        memset(F,255,sizeof F);
        F[1][0] = 0;
        for(int j = 1;j <= cnt;j ++)
            for(int ch = 0;ch <= N;ch ++)
            if (F[j][ch] == -1) continue; else
            if (!Bel[Ord[j]]) 
                F[j + 1][ch + 1] = max(F[j + 1][ch + 1],F[j][ch]),F[j + 1][ch] = max(F[j + 1][ch],F[j][ch]); else
            {
                if (A[Ord[j]] > X) 
                {
                    F[j + 1][ch + 1] = max(F[j + 1][ch + 1],F[j][ch] + A[Ord[j]]); 
                    F[j + 1][ch] = max(F[j + 1][ch],F[j][ch] + ((j - ch) <= K ? B[Ord[j]] : 0));
                } else
                {
                    F[j + 1][ch] = max(F[j + 1][ch],F[j][ch] + ((j - ch) <= K ? B[Ord[j]] : 0));
                }
            }
        if (F[cnt + 1][K - 1] != -1 && Bel[i]) F[cnt + 1][K - 1] += X;
        Ans = max(Ans,F[cnt + 1][K - 1]);
    }
    printf("%d\n", Ans);
}