POJ 3762 区间K覆盖

版权声明:本博客内容基本为原创,如有问题欢迎联系,转载请注明出处 https://blog.csdn.net/qq_41955236/article/details/82747848

题目链接:http://poj.org/problem?id=3762

题意:

       给你n个时间段,每个区间都有一个自带的权值,你要从里面选取若干个区间,注意每个区间只可以被选择一次,使得每个时间段上没有点可以被使用超过K次并且使得到的权值最大。

       

做法:

       请看这里https://blog.csdn.net/qq_41955236/article/details/82744109

       第二篇K覆盖了。开心。


#include <map>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<queue>
using namespace std;
typedef long long ll;
const int maxn=6000;
const int maxm=20000;
const int inf=0x3f3f3f3f;
int dis[maxn];
int vis[maxn],pre[maxn];
int head[maxn],cnt;
int n,m,sp,tp,k;
struct node{
    int to,cap,next;
    int cost;
}e[maxm];
map<int,int> mp;
void add(int from,int to,int cap,int cost){
    e[cnt].to=to; e[cnt].cap=cap;
    e[cnt].cost=cost; e[cnt].next=head[from];
    head[from]=cnt++;

    e[cnt].to=from; e[cnt].cap=0;
    e[cnt].cost=-cost; e[cnt].next=head[to];
    head[to]=cnt++;
}
bool spfa(int s,int t,int &flow,int &cost){
    queue<int> q;
    memset(dis,inf,sizeof(dis));
    memset(vis,0,sizeof(vis));
    memset(pre,-1,sizeof(pre));
    dis[s]=0;  q.push(s);
    vis[s]=1;
    int d=inf;
    while(!q.empty()){
        int u=q.front();
        q.pop();
        vis[u]=0;
        for(int i=head[u];~i;i=e[i].next){
            int v=e[i].to;
            if(e[i].cap>0&&dis[v]>dis[u]+e[i].cost){
                dis[v]=dis[u]+e[i].cost;
                pre[v]=i;
                if(!vis[v]){
                    vis[v]=1;
                    q.push(v);
                }
            }
        }
    }
    if(dis[t]==inf){
        return false;
    }
    for(int i=pre[t];~i;i=pre[e[i^1].to]){
        d=min(d,e[i].cap);
    }
    for(int i=pre[t];~i;i=pre[e[i^1].to]){
        e[i].cap-=d;
        e[i^1].cap+=d;
        cost+=e[i].cost*d;
    }
    flow+=d;
    return true;
}
int mcmf(int s,int t){
    int flow=0;
    int cost=0;
    while(spfa(s,t,flow,cost)){
        //cout<<flow<<" "<<cost<<endl;
    }
    return cost;
}
struct nodes{
    int u,v;
    int c;
}ee[2500];
char s[15];
int dd(char x){
    return x-'0';
}
int deal(char s[]){
    return dd(s[7])+dd(s[6])*10+dd(s[4])*60+dd(s[3])*600+dd(s[1])*3600+dd(s[0])*36000;
}
void input(int i){
    scanf("%s",s);ee[i].u=deal(s);
    scanf("%s",s);ee[i].v=deal(s);
    scanf("%d",&ee[i].c);
}
int main(){
    int tmp[4005],numtmp=0; mp.clear();
    scanf("%d%d",&n,&k);
    cnt=0; sp=0;
    memset(head,-1,sizeof(head));
    for(int i=0;i<n;i++){
        input(i);
        tmp[numtmp++]=ee[i].u,tmp[numtmp++]=ee[i].v;
    }
    sort(tmp,tmp+numtmp);
    int cc=unique(tmp,tmp+numtmp)-tmp;
    numtmp=0;
    for(int i=0;i<cc;i++){
        mp[tmp[i]]=++numtmp;
    }
    for(int i=0;i<n;i++){
        int u=mp[ee[i].u],v=mp[ee[i].v];
        add(u,v,1,-ee[i].c);
    }
    for(int i=1;i<numtmp;i++)
        add(i,i+1,k,0);
    tp=numtmp+1;
    add(sp,1,k,0); add(numtmp,tp,k,0);
    printf("%d\n",-mcmf(sp,tp));

    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_41955236/article/details/82747848