poj 3680 Intervals (最小费用最大流)

建图，先将0到1，,1到2，2到3，，，N到N+1之间都建立费用为0容量为k的边。

再将输入进来的a和b之前建立容量为1费用为-w的边。

建图之后就是最小费用最大流了。

#include <iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define maxnode 50005
#define maxedge 20005
using namespace std;

int t,s;

struct A
{
    int num,loc;
}arr[4005];

bool cmp(A a,A b)
{
    return a.num<b.num;
}

struct B
{
    int a,b,w;
}line[maxedge];

struct EDGE
{
    int st,to,next,cf,w;
}edge[maxedge];

int head[maxnode],e;

void addedge(int u,int v,int val,int w)
{
    edge[e].st=u;
    edge[e].to=v;
    edge[e].next=head[u];
    edge[e].cf=val;
    edge[e].w=w;
    head[u]=e++;
    edge[e].st=v;
    edge[e].to=u;
    edge[e].next=head[v];
    edge[e].cf=0;
    edge[e].w=-w;
    head[v]=e++;
}

int q[maxnode],qsize,dis[maxnode],pre[maxnode],flow[maxnode];
bool vis[maxnode];

bool spfa()
{
    //memset(vis,0,sizeof(vis));
    memset(dis,127,sizeof(dis));
    memset(pre,-1,sizeof(pre));
    int i,v,u,j;
    qsize=0;
    dis[s]=0;
    q[qsize++]=s;
    vis[s]=1;
    flow[s]=999999999;
    for(i=0;i<qsize;i++){
        u=q[i];
        vis[u]=0;//注意spfa出队时要把标记置为0，之前忘写了，结果一直T
        for(j=head[u];j!=-1;j=edge[j].next){
            v=edge[j].to;
            if(edge[j].cf>0 && dis[v]>dis[u]+edge[j].w){
                dis[v]=dis[u]+edge[j].w;
                pre[v]=j;
                flow[v]=min(flow[u],edge[j].cf);
                if(!vis[v]){
                    vis[v]=1;
                    q[qsize++]=v;
                }
            }
        }
    }
    return pre[t]!=-1;
}

int solve()
{
    int ret=0;
    while(spfa()){
        int tmp=t;
        while(tmp!=s){
            edge[pre[tmp]].cf-=flow[t];
            edge[pre[tmp]^1].cf+=flow[t];
            tmp=edge[pre[tmp]].st;
        }
        ret+=dis[t];
    }
    return ret;
}

int main()
{
    int T;
    scanf("%d",&T);
    while(T--){
        int n,k,i,a,b,w,ct=0;
        e=0;
        memset(head,-1,sizeof(head));
        scanf("%d%d",&n,&k);
        s=0;
        for(i=1;i<=n;i++){
            scanf("%d%d%d",&a,&b,&w);
            arr[ct].num=a;
            arr[ct++].loc=-i;
            arr[ct].loc=i;
            arr[ct++].num=b;
            line[i].w=w;
        }
        sort(arr,arr+ct,cmp);
        int tmp=-1,number=0;
        for(i=0;i<ct;i++){
            if(arr[i].num!=tmp){
                number++;
                tmp=arr[i].num;
            }
            if(arr[i].loc<0){
                line[-arr[i].loc].a=number;
            }
            else{
                line[arr[i].loc].b=number;
            }
        }
        for(i=1;i<=n;i++){
            addedge(line[i].a,line[i].b,1,-line[i].w);
        }
        for(i=0;i<=number;i++){
            addedge(i,i+1,k,0);
        }
        t=number+1;
        int ans=solve();
        printf("%d\n",-ans);
    }
    return 0;
}