题意:
给出m个视频,有两种类型,每个视频有一个开始时间和结束时间,每部视频最多给一个人看,观看后得到wi的快乐值,一次只能看一部视频。
一个人连续观看两部相同类型的视频会有一个 -W 快乐值得效果。问现在有k个人,能获得的快乐值总和最多是多少。
思路:
费用流,将人看成流量,路径模拟观看视频的过程。
每个视频拆点,连一条流量为1,费用为-w的边;源点到次源点连一条流量为k,费用为0的边;
次源点向每个视频连一条流量为1,费用为0的边;视频之间如果不交叉,相同类型连一条流量为1,费用为W的边,不同类型连一条流量为1,费用为0的边。
详情可以看注释。
代码:
#include <algorithm>
#include <iostream>
#include <iomanip>
#include <sstream>
#include <cstring>
#include <cstdlib>
#include <cctype>
#include <cstdio>
#include <string>
#include <bitset>
#include <cmath>
#include <ctime>
#include <queue>
#include <stack>
#include <list>
#include <map>
#include <set>
#define X first
#define Y second
#define lch (o<<1)
#define rch (o<<1|1)
#define pb push_back
#define lowbit(x) (x&-x)
#define pii pair<int,int>
#define sd(n) scanf("%d",&n)
#define sf(n) scanf("%lf",&n)
#define cout1(x) cout<<x<<endl
#define pdd pair<double,double>
#define ALL(x) x.begin(),x.end()
#define sdd(n,m) scanf("%d%d",&n,&m)
#define INS(x) inserter(x,x.begin())
#define mst(a,b) memset(a,b,sizeof(a))
#define sff(n,m) scanf("%lf%lf",&n,&m)
#define cout2(x,y) cout<<x<<" "<<y<<endl
#define qclear(a) while(!a.empty())a.pop()
#define sddd(n,m,k) scanf("%d%d%d",&n,&m,&k)
#define IOS std::ios::sync_with_stdio(false)
#define SRAND srand((unsigned int)(time(0)))
#define sfff(n,m,k) scanf("%lf%lf%lf",&n,&m,&k)
#define cout3(x,y,z) cout<<x<<" "<<y<<" "<<z<<endl
typedef long long ll;
typedef unsigned long long ull;
typedef unsigned int uint;
using namespace std;
const double PI=acos(-1.0);
const int INF=0x3f3f3f3f;
const double eps=1e-3;
const ll mod=1e9;
const int maxn=505;
const int maxm=100005;
struct Edge{
int to,next,cap,flow,cost;
}edge[maxm];
int h[maxn],tot;
int pre[maxn];
int dis[maxn];
bool vis[maxn];
int N;
void init(int n){
N=n;
tot=0;
mst(h,-1);
}
void addedge(int u,int v,int cap,int cost){
edge[tot].to=v;
edge[tot].cap=cap;
edge[tot].cost=cost;
edge[tot].flow=0;
edge[tot].next=h[u];
h[u]=tot++;
edge[tot].to=u;
edge[tot].cap=0;
edge[tot].cost=-cost;
edge[tot].flow=0;
edge[tot].next=h[v];
h[v]=tot++;
}
bool spfa(int s,int t){
queue<int>que;
for(int i=0;i<N;i++){
dis[i]=INF;
vis[i]=0;
pre[i]=-1;
}
dis[s]=0;
vis[s]=1;
que.push(s);
while(!que.empty()){
int u=que.front();
que.pop();
vis[u]=0;
for(int i=h[u];i!=-1;i=edge[i].next){
int v=edge[i].to;
if(edge[i].cap>edge[i].flow &&
dis[v]>dis[u]+edge[i].cost){
dis[v]=dis[u]+edge[i].cost;
pre[v]=i;
if(!vis[v]){
vis[v]=1;
que.push(v);
}
}
}
}
return (pre[t]!=-1);
}
int mcmf(int s,int t,int &cost){
int flow=0;
cost=0;
while(spfa(s,t)){
int mn=INF;
for(int i=pre[t];i!=-1;i=pre[edge[i^1].to]){
if(mn>edge[i].cap-edge[i].flow){
mn=edge[i].cap-edge[i].flow;
}
}
for(int i=pre[t];i!=-1;i=pre[edge[i^1].to]){
edge[i].flow+=mn;
edge[i^1].flow-=mn;
cost+=edge[i].cost*mn;
}
flow+=mn;
}
return flow;
}
struct mv{
int s,t,w,op;
bool operator <(const mv &o)const{
if(s==o.s)return t<o.t;
return s<o.s;
}
};
mv mvs[maxn];
void solve() {
int t;
sd(t);
while(t--){
int n,m,k,w;
sdd(n,m);
sdd(k,w);
for(int i=0;i<m;i++){
sd(mvs[i].s);
sd(mvs[i].t);
sd(mvs[i].w);
sd(mvs[i].op);
}
init(m*2+3);//原点、次原点、汇点、每个视频拆成两个点,一共2m+3个点
addedge(0,1,k,0);//限制人数
int cur=2;
int ed=2*m+2;
for(int i=0;i<m;i++){
addedge(cur,cur+1,1,-mvs[i].w);//每部电影拆点,连一条流量为1(只能一个人看),费用为-w(最小费用变最大费用),的边
addedge(1,cur,1,0);
addedge(cur+1,ed,1,0);
cur+=2;
}
for(int i=0;i<m;i++){
for(int j=0;j<m;j++){//枚举每两部视频
if(mvs[j].s>=mvs[i].t){//是否重合
int debuf=0;
if(mvs[j].op==mvs[i].op){//如果类型相同,有debuf
debuf=w;
}
addedge(2*(i+1)+1,2*(j+1),1,debuf);//两个视频之间连边
}
}
}
int ans;
mcmf(0,ed,ans);
printf("%d\n",-ans);
}
return ;
}
int main() {
#ifdef LOCAL
freopen("in.txt","r",stdin);
// freopen("out.txt","w",stdout);
#else
// freopen("","r",stdin);
// freopen("","w",stdout);
#endif
solve();
return 0;
}