题目链接:https://nanti.jisuanke.com/t/31715
题意:
给你n个部件,每个部件最多用k次,现在有m种组装子部件的方式可以赚得额外的权值,但是每种方法最多只会被用一次。组装子部件需要ai,bi两个部件(两个都会被用一次),可以得到wi的价值。
做法:
知道是裸题有点小难受,当时根本就没看(当时也不会这个k区间)。这个题的连边有点不一样,是从u连向v+1的,为什么呢。因为两边的点都会被用到,所以覆盖到的其实是u,v的区间。之前的题是v可以马上开始另一件任务,是[ )的关系,这个题是两边都会覆盖到,是[ ] 的关系,所以需要我们要把区间设置成v+1。然后套个板子就A啦啦。
至于建图,https://blog.csdn.net/qq_41955236/article/details/82744109这里讲的比较详细。
#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;
ll cost;
}e[maxm];
map<int,int> mp;
void add(int from,int to,int cap,ll 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,ll &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;
}
ll mcmf(int s,int t){
int flow=0;
ll cost=0;
while(spfa(s,t,flow,cost)){
//cout<<flow<<" "<<cost<<endl;
}
return cost;
}
struct nodes{
int u,v;
ll c;
}ee[2500];
int main(){
int t;
cin>>t;
while(t--){
int tmp[1005],numtmp=0; mp.clear();
scanf("%d%d%d",&n,&k,&m);
cnt=0;
memset(head,-1,sizeof(head));
sp=0; tp=n+1;
for(int i=0;i<m;i++){
scanf("%d%d%lld",&ee[i].u,&ee[i].v,&ee[i].c);
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<m;i++){
int u=mp[ee[i].u],v=mp[ee[i].v];
add(u,v+1,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("%lld\n",-mcmf(sp,tp));
}
return 0;
}