HDU 4341 Gold miner 分组背包 并查集

题目

题意: 黄金矿工. 将每个矿抽象成点, 给出每个矿的坐标, 得到这个矿的时间和矿的价值, 问在有限的时间内, 得到的最大价值是多少.

思路: 同一条线上的矿, 前面的不取后面的就不能取. 那么可以将同一条线上的点分到同一组, 组内按照从近到远的顺序排序, 考虑分组背包, 组内的第j个元素是第1→第j个矿石的总价值/代价, 每个组内只能取1个元素.

代码:

#include<bits/stdc++.h>
#define fuck(x) std::cout<<"["<<#x<<"->"<<x<<"]"<<endl;
using namespace std;
typedef long long ll;

const int M=2e5+5;
const int inf=1e9+5;
const int mod=1e9+7;

int n,t;
int dp[M];

int vi=0;
struct node {
    int x,y,t,v;
    int fa;
} pep[205];

bool cmp(node a,node b) {
    return a.x*a.x+a.y*a.y<b.x*b.x+b.y*b.y;
}

vector<node>v[M];
int findfa(int a) {
    int x=a;
    while(pep[x].fa!=x) {
        x=pep[x].fa;
    }
    return pep[a].fa=x;
}

int main() {
    int cz=0;
    while(~scanf("%d%d",&n,&t)) {
        memset(dp,0,sizeof(dp));
        for(int i=0; i<M; i++)
            v[i].clear();
        vi=0;
        for(int i=0; i<n; i++) {
            scanf("%d%d%d%d",&pep[i].x,&pep[i].y,&pep[i].t,&pep[i].v);
            pep[i].fa=i;//并查集维护同一条线
        }
        for(int i=0; i<n; i++) {
            for(int j=i+1; j<n; j++) {
                if(pep[i].x*pep[j].y==pep[i].y*pep[j].x) {//只要没有0,0的情况,满足x1*y2==x2*y1就能保证在一条线上
                    pep[j].fa=i;
                }
            }
        }
        for(int i=0; i<n; i++) {
            int fa=findfa(i);
            if(fa==i) {//如果是第一个新建一个分组
                v[vi++].push_back(pep[i]);
            } else {//加入已有的分组
                v[fa].push_back(pep[i]);
            }
        }
        for(int i=0; i<vi; i++) {
            sort(v[i].begin(),v[i].end(),cmp);
            for(int j=1; j<v[i].size(); j++) {
                v[i][j].t+=v[i][j-1].t;
                v[i][j].v+=v[i][j-1].v;
                //把在一条线上的点分到一个组内
                //转化为分组背包
                //取了这个点前面的所有点这个点才能取,那么这个点的价值/代价就变成前j个点的总价值/代价
            }
        }
        for(int i=0; i<vi; i++) {
            for(int V=t; V>=0; V--) {//分组背包先V:t->0后组内,保证每个dp[V]只被组内一个元素更新
                for(int j=0; j<v[i].size(); j++) {
                    if(V>=v[i][j].t)
                        dp[V]=max(dp[V],dp[V-v[i][j].t]+v[i][j].v);
                }
            }
        }

        int ans=0;
        for(int i=0; i<=t; i++) {
            ans=max(ans,dp[i]);
        }
        printf("Case %d: %d\n",++cz,ans);

    }
    return 0;
}