bzoj 1045 [HAOI2008] 糖果传递——设变量推式子

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=1045

费用流TLE。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
#define ll long long
using namespace std;
const int N=1e6+;
const ll INF=1e15;
int n,a[N],hd[N],xnt=,pre[N],dis[N];
ll ans,base;
bool vis[N];
struct Ed{
    int nxt,to,w;
    ll cap;
    Ed(int n=,int t=,ll c=,int w=):nxt(n),to(t),cap(c),w(w) {}
}ed[N<<];
int rdn()
{
    int ret=;char ch=getchar();
    while(ch>''||ch<'')ch=getchar();
    while(ch>=''&&ch<='')(ret*=)+=ch-'',ch=getchar();
    return ret;
}
void add(int x,int y,ll cap)
{
    int k=;if(!x||y==n+)k=;
    ed[++xnt]=Ed(hd[x],y,cap,k);hd[x]=xnt;
    ed[++xnt]=Ed(hd[y],x,,-k);hd[y]=xnt;
}
queue<int> q;
bool spfa()
{
    q.push();vis[]=;
    memset(dis,0x3f,sizeof dis);dis[]=;
    while(q.size())
    {
        int k=q.front();q.pop();vis[k]=;
        for(int i=hd[k],v;i;i=ed[i].nxt)
            if(ed[i].cap&&dis[v=ed[i].to]>dis[k]+ed[i].w)
            {
                dis[v]=dis[k]+ed[i].w;
                pre[v]=i;
                if(!vis[v])vis[v]=,q.push(v);
            }
    }
    return dis[n+]<0x3f3f3f3f;
}
void ek()
{
    ll flow=INF;
    for(int i=n+;i;i=ed[pre[i]^].to)
        flow=min(flow,ed[pre[i]].cap);
    for(int i=n+;i;i=ed[pre[i]^].to)
        ed[pre[i]].cap-=flow,ed[pre[i]^].cap+=flow;
    ans+=flow*dis[n+];
}
int main()
{
    n=rdn();
    for(int i=;i<=n;i++)a[i]=rdn(),base+=a[i];
    base/=n;
    for(int i=;i<=n;i++)
    {
        if(a[i]>base)add(,i,a[i]-base);
        if(a[i]<base)add(i,n+,base-a[i]);
        add(i,i-?i-:n,INF);add(i,i+<=n?i+:,INF);
    }
    while(spfa())ek();
    printf("%lld\n",ans);
    return ;
}

TJ：http://hzwer.com/2656.html

为了计算出每个人传递出去的糖果数量，需要先把它们设出来。

不妨指定一个方向。

然后尽量把每个变量用常量表示出来。

希望自己以后也能想出来。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
const int N=1e6+;
int n;
ll a[N],base,c[N],ans;
int main()
{
    scanf("%d",&n);
    for(int i=;i<=n;i++)
        scanf("%lld",&a[i]),base+=a[i];
    base/=n;
    for(int i=;i<n;i++)c[i]=c[i-]+a[i]-base;
    sort(c+,c+n);
    ll x=c[(n+)>>];
    for(int i=;i<n;i++)ans+=abs(x-c[i]);
    printf("%lld\n",ans);
    return ;
}