首页 > 其他分享 >abc100d<枚举>

abc100d<枚举>

时间:2024-01-15 11:57:54浏览次数:19  
标签:abc100d LL int score 正负 ans include

题目

D - Patisserie ABC

思路

  • 枚举三组权重和的正负情况,一共2^3种可能;
  • 对于每种情况,直接计算对应得分,取最大的m个蛋糕的得分求和;
  • 对所有情况的得分取最大即可。

总结

代码

点击查看代码
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
using LL = long long;
const int N = 1010;
LL w[N][3];

void solv()
{
    int n, m;
    cin >> n >> m;
    for (int i = 1; i <= n; i ++)
    {
        for (int j = 0; j < 3; j ++)
            cin >> w[i][j];
    }

    LL ans = 0;
    // 枚举三个权重的正负情况,一共2^3种可能
    for (int i = 0; i < 8; i ++)
    {
        // 得到当前组的正负情况
        LL sel[3];
        for (int j = 0; j < 3; j ++)
            sel[j] = (i >> j & 1) ? 1 : -1;

        // 根据正负情况,计算每个蛋糕的分数
        vector<LL> tmp;
        LL score;
        for (int j = 1; j <= n; j ++)
        {
            score = 0;
            for (int k = 0; k < 3; k ++)
                score += sel[k] * w[j][k];
            tmp.push_back(score);
        }

        // 按得分降序排序,取前m个
        sort(tmp.begin(), tmp.end(), greater<LL>());
        score = 0;
        for (int j = 0; j < m; j ++) 
            score += tmp[j];

        ans = max(ans, score);
    }

    cout << ans << '\n';
}

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int T = 1;
    // cin >> T;
    while (T--)
        solv();
    return 0;
}

标签:abc100d,LL,int,score,正负,ans,include
From: https://www.cnblogs.com/o2iginal/p/17965069

相关文章