P2754 [CTSC1999]家园 / 星际转移问题

P2754 [CTSC1999]家园 / 星际转移问题

关键

一个是需要建立分层图，但是分层图如何建立？
使用一个while循环，对每次的人数跑一次就可以了
挺奇妙的

代码

#include <bits/stdc++.h>
using namespace std;
const int N=1005,M=1e6+5;
const int inf=1e9;
 
int h[N],ne[M],e[M],w[M],tot=1;
void add(int from,int to,int wi) {
    e[++tot]=to;  w[tot]=wi;  ne[tot]=h[from];  h[from]=tot;
    e[++tot]=from;w[tot]=0;  ne[tot]=h[to];    h[to]=tot;
}
 
int S=N-2,T=N-1;
int cur[N],dep[N];
bool bfs() {
    memcpy(cur,h,sizeof(h));
    memset(dep,0,sizeof(dep));
    queue<int>q;
    q.push(S);
    dep[S]=1;
    while(!q.empty()) {
        int now=q.front();
        q.pop();
        for(int i=h[now];i;i=ne[i]) {
            int to=e[i];
            if(dep[to]==0&&w[i]>0)
                dep[to]=dep[now]+1,q.push(to);
        }
    }
    return dep[T];
}
 
int dfs(int now,int sum) {
    if(now==T)return sum;
    int ans=0;
    for(int i=cur[now];i&&sum;i=ne[i]) {
        cur[now]=i;
        int to=e[i];
        if(dep[to]==dep[now]+1&&w[i]>0) {
            int k=dfs(to,min(sum,w[i]));
            if(k==0)dep[to]=0;
            w[i]-=k;
            w[i^1]+=k;
            sum-=k;
            ans+=k;
        }
    }
    return ans;
}
 
int dinic() {
    int ans=0;
    while(bfs())ans+=dfs(S,inf);
    return ans;
}

int fa[N];
int find(int x) {
    return x==fa[x]?fa[x]:find(fa[x]);
}
int n,m,k;
int a[N],b[N];
int c[50][50];
int id(int x,int y) {
    return (x)*(n+2)+y;
}

int main() {
    ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
    for(int i=0;i<N;i++)fa[i]=i;//初始化
    cin>>n>>m>>k;
    for(int i=1;i<=m;i++) {
        cin>>a[i]>>b[i];
        int pre=0;
        for(int j=0;j<b[i];j++) {
            cin>>c[i][j];
            if(c[i][j]==0)c[i][j]=n+1;
            if(c[i][j]==-1)c[i][j]=n+2;
            if(pre)fa[find(c[i][j])]=find(pre);//进行合并
            pre=c[i][j];
        }
    }
    //对天数进行暴力转移
    //起点是n+1,终点是n+2
    if(find(n+1)!=find(n+2))return cout<<0,0;
    int ans=1,sum=0;
    add(S,id(0,n+1),inf);
    add(id(0,n+2),T,inf);
    while(1) {
        for(int i=1;i<=n+2;i++)add(id(ans-1,i),id(ans,i),inf);
        for(int i=1;i<=m;i++) {
            int  x=c[i][(ans-1)%b[i]],y=c[i][ans%b[i]];
            add(id(ans-1,x),id(ans,y),a[i]);//上一天向下一天进行连边
        }
        add(S,id(ans,n+1),inf);
        add(id(ans,n+2),T,inf);
        sum+=dinic();
        if(sum>=k)break;
        ans++;
    }
    cout<<ans<<endl;
    return 0;
}