[Luogu] P3701 「伪模板」主席树

题目背景

byx和手气君都非常都非常喜欢种树。有一天，他们得到了两颗奇怪的树种，于是各自取了一颗回家种树，并约定几年后比一比谁种出来的树更加牛x。

题目描述

很快，这棵树就开花结果了。byx和手气君惊讶的发现，这是一棵主席树，树上长满了主席和主席的朋友们。这棵树上一共有五种人，主席(J)，记者(HK)，高人(W)，女王(E)和膜法师(YYY)。他们发现，他们的主席树上的人数相同，都为N。

研究发现，这五种人的输赢如上图所示（一样的人不能PK），箭头指向输的人。至于为什么，留给同学们自己思考。

比赛如期进行。

byx和手气君要进行M场比赛，每一场比赛他们会选出树上的两个人来比较看谁更牛x。

第i个人寿命为Lifei秒，每次比完赛他们就会-1s。当他们生命为0s时他们就不能再比赛了。

同时，当J的寿命为0时，同一棵树上的YYY可以为他+1s。每个YYY只能给.每个J续一次。

那么问题来了

现在给定N,M（1≤N≤100,1≤M≤1000），A和B每一个人所属种类(J,HK,W,YYY或E)以及每一个人的生命，生命不超过50.请你算算A最多能够赢得多少场比赛呢。

数据保证每一场一定都有人用。两个人之间只能比一场。

题目解析

其实没有那么难吧

最大流

S向byx的人连边，边权是寿命。手气君的人向T连边，边权是寿命。

如果byx的人向ta能战胜的手气君的人连边，边权是1。

然后DINIC

不过貌似什么地方写挂了，只有90。完了再填坑吧

Code

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;

const int MAXN =  + ;
const int MAXM =  + ;
const int S = ;
const int T = ;
const int INF = 0x3f3f3f3f; 

struct Tree {
    string name;
    int life;
} a[MAXN],b[MAXN];
struct Edge {
    int nxt;
    int to,w;
} l[MAXN*MAXN*];

int n,m;
int maxflow;
int num[];
int head[MAXN*],cnt=;
int deep[MAXN*],cur[MAXN*];

inline void add(int x,int y,int z) {
    cnt++;
    l[cnt].nxt = head[x];
    l[cnt].to = y;
    l[cnt].w = z;
    head[x] = cnt;
    return;
}

inline void build() {
    for(int i = ;i <= n;i++) {
        add(S,i,a[i].life);add(i,S,);
        add(i+n,T,b[i].life);add(T,i+n,);
    }
    for(int i = ;i <= n;i++) {
        for(int j = ;j <= n;j++) {
            if(a[i].name == "W" && (b[j].name == "YYY" || b[j].name == "E")) add(i,j+n,),add(j+n,i,);
            else if(a[i].name == "HK" && (b[j].name == "W" || b[j].name == "E")) add(i,j+n,),add(j+n,i,);
            else if(a[i].name == "E" && (b[j].name == "J" || b[j].name == "YYY")) add(i,j+n,),add(j+n,i,);
            else if(a[i].name == "YYY" && (b[j].name == "J" || b[j].name == "HK")) add(i,j+n,),add(j+n,i,);
            else if(a[i].name == "J" && (b[j].name == "W" || b[j].name == "HK")) add(i,j+n,),add(j+n,i,);
        }
    }
    return;
}
queue<int> Q;
inline bool bfs(){
    memset(deep,,sizeof(deep));
    while(Q.size()) Q.pop();
    deep[S]=;Q.push(S);
    while(Q.size()){
        int x=Q.front();Q.pop();
        for(register int i=head[x];i;i=l[i].nxt){
            int u=l[i].to;
            if(!deep[u] && l[i].w){
                deep[u]=deep[x]+;
                if(u==T) return true;
                Q.push(u);
            }
        }
    }
    return false;
}

int dfs(int x,int flow){
    if(x==T) return flow;
    int res=flow,k;
    for(register int i=head[x];i && res;i=l[i].nxt){
        int u=l[i].to;
        if(l[i].w && deep[u]==deep[x]+){
            k=dfs(u,min(l[i].w,res));
            if(!k) deep[u]=;
            l[i].w -= k;
            res-=k;
        }
    }
    return flow-res;
}

inline void Dinic(int s,int t) {
    while(bfs()) maxflow += dfs(s,INF);
    return;
}

int main() {
    scanf("%d%d",&n,&m);
    for(int i = ;i <= n;i++) {
        cin>>a[i].name;
        if(a[i].name == "YYY") num[]++;
    }
    for(int i = ;i <= n;i++) {
        cin>>b[i].name;
        if(b[i].name == "YYY") num[]++;
    }
    for(int i = ;i <= n;i++) {
        scanf("%d",&a[i].life);
        if(a[i].name == "J") a[i].life += num[];
    }
    for(int i = ;i <= n;i++) {
        scanf("%d",&b[i].life);
        if(b[i].name == "J") b[i].life += num[];
    }
    build();
    Dinic(S,T);
    int debug = min(m,maxflow);
    if(debug == ) debug += ;//QAQ
    printf("%d",debug);
    return ;
}