codevs 1033 蚯蚓的游戏问题 费用流+拆点

传送门：codevs1033

思路：一开始没审清楚题，智障的写成了最大流，结果发现是要求点权和。。

既然是关于点权的问题，并且还要限制每个点的流量，那么必定要拆点将点权转化为边权，然后按照题目给定的走法连边就好了。

代码：

#include<bits/stdc++.h>
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
const int MAXN=5050;
const int MAXM=100010;
int head[MAXN],pre[MAXN],dis[MAXN],book[MAXN];
int cnt=0,N;//N为点数
struct node{
	int to,next,cap,flow,cost;
}edge[MAXM];
void init()
{
	cnt=0;
	memset(head,-1,sizeof(head));
}
void addedge(int u,int v,int cap,int cost)
{
	edge[cnt].to=v;edge[cnt].cap=cap;edge[cnt].flow=0;
	edge[cnt].cost=cost;edge[cnt].next=head[u];head[u]=cnt++;
	edge[cnt].to=u;edge[cnt].cap=0;edge[cnt].flow=0;
	edge[cnt].cost=-cost;edge[cnt].next=head[v];head[v]=cnt++;
}
bool spfa(int s,int t)
{
	queue<int>q;
	while(!q.empty())q.pop();
	for(int i=0;i<=N;i++)
	{
		dis[i]=inf;
		pre[i]=-1;
		book[i]=0;
	}
	dis[s]=0;book[s]=1;q.push(s);
	while(!q.empty())
	{
		int u=q.front();q.pop();book[u]=0;
		for(int i=head[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(!book[v])
				{
					book[v]=1;
					q.push(v);
				}
			}
		 }
	}
	return pre[t]!=-1;
}
int min_cost_max_flow(int s,int t,int &cost)
{
	int flow=0;
	cost=0;
	while(spfa(s,t))
	{
		int temp=inf;
		for(int i=pre[t];i!=-1;i=pre[edge[i^1].to])
		temp=min(temp,edge[i].cap-edge[i].flow);
		for(int i=pre[t];i!=-1;i=pre[edge[i^1].to])
		{
			edge[i].flow+=temp;
			edge[i^1].flow-=temp;
			cost+=edge[i].cost*temp;
		}
		flow+=temp;
	}
	return flow;
}
int w[66][66];
void debug()
{
    for(int i=0;i<cnt;i+=2)
    {
        printf("%d %d %d %d\n",edge[i^1].to,edge[i].to,edge[i].flow,edge[i].cost);
    }
}
int main()
{
    int n,m,k,dd=2500;
    init();
    int source=0,sink;
    cin>>n>>m>>k;
    for(int i=0;i<n;i++)
    {
        for(int j=0;j<m+i;j++)
            cin>>w[i][j];
    }
    int t=1;
    for(int i=0;i<n-1;i++)
    {
        for(int j=0;j<m+i;j++)
        {
            if(!i)
            addedge(source,t+j,inf,0);
            addedge(t+j,t+j+dd,1,-w[i][j]);
            addedge(t+j+dd,t+m+i+j,inf,0);
            addedge(t+j+dd,t+m+i+j+1,inf,0);
        }
        t+=m+i;
    }
    sink=t+m+n-1;
    for(int i=0;i<m+n-1;i++)
    {
		addedge(t+i,t+i+dd,1,-w[n-1][i]);
		addedge(t+i+dd,sink,1,0);
    }
    addedge(sink+1,source,k,0);
    N=sink+dd+1;
    int cost;
    min_cost_max_flow(sink+1,sink,cost);
    //debug();
    cout<<-cost;
}