BZOJ-3171 && 2661 循环格&&连连看 最小费用最大流

我的程序实现出了一点问题，大约是常用的算法上出现了一些偏差QAQ，不过建图都能一眼建出来...自己的代码MCMF刚写，太丑就不放了..

2661: [BeiJing wc2012]连连看
Time Limit: 10 Sec Memory Limit: 128 MB
Submit: 944 Solved: 375
[Submit][Status][Discuss]

Description
凡是考智商的题里面总会有这么一种消除游戏。不过现在面对的这关连连看可不是QQ游戏里那种考眼力的游戏。我们的规则是，给出一个闭区间[a,b]中的全部整数，如果其中某两个数x,y（设x>y）的平方差x2-y2是一个完全平方数z2，并且y与z互质，那么就可以将x和y连起来并且将它们一起消除，同时得到x+y点分数。那么过关的要求就是，消除的数对尽可能多的前提下，得到足够的分数。快动手动笔算一算吧。

Input
只有一行，两个整数，分别表示a,b。

Output
两个数，可以消去的对数，及在此基础上能得到的最大分数。

Sample Input
1 15

Sample Output
2 34

HINT
对于30%的数据，1<=a,b<=100
对于100%的数据，1<=a,b<=1000

Source

建图：
拆点，枚举数对i，j，如果满足以上的限制，就向连，容量为1，费用为i+j；
再由源向各入点连容量为1，费用为0
由各出点向汇连容量为1，费用为0
再枚举所有到汇点的点/2即为合并对数，费用为最大得分

code by hzwer：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#define ll long long
#define inf 0x7fffffff
#define T 2001
using namespace std;
inline ll read()
{
    ll x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int a,b,cnt=1;
int ans1,ans2;
int head[2005],q[2005],dis[2005],from[2005];
bool inq[2005];
struct data{int to,from,next,v,c;}e[2000005];
void ins(int u,int v,int w,int c)
{
    e[++cnt].to=v;e[cnt].from=u;
    e[cnt].next=head[u];head[u]=cnt;
    e[cnt].v=w;e[cnt].c=c;
}
void insert(int u,int v,int w,int c)
{ins(u,v,w,c);ins(v,u,0,-c);}
inline int gcd(int a,int b){return b==0?a:gcd(b,a%b);}
bool spfa()
{
    for(int i=0;i<=T;i++)dis[i]=-inf;
    int t=0,w=1;
    dis[0]=0;q[0]=0;inq[0]=1;
    while(t!=w)
    {
        int now=q[t];t++;if(t==2001)t=0;
        for(int i=head[now];i;i=e[i].next)
            if(e[i].v&&e[i].c+dis[now]>dis[e[i].to])
            {
                dis[e[i].to]=e[i].c+dis[now];
                from[e[i].to]=i;
                if(!inq[e[i].to])
                {
                    inq[e[i].to]=1;
                    q[w++]=e[i].to;
                    if(w==2001)w=0;
                }
            }
        inq[now]=0;
    }
    if(dis[T]==-inf)return 0;
    return 1;
}
void dfs()
{
    int x=inf;
    for(int i=from[T];i;i=from[e[i].from])
        x=min(e[i].v,x);
    for(int i=from[T];i;i=from[e[i].from])
        {ans2+=x*e[i].c;e[i].v-=x;e[i^1].v+=x;}
}
bool jud(int x,int y)
{
    if(x<y)swap(x,y);
    int t=int(sqrt(x*x-y*y));
    return (gcd(y,t)==1&&x*x-y*y==t*t);
}
int main()
{
    a=read();b=read();
    for(int i=a;i<=b;i++)
        for(int j=a;j<=b;j++)
            if(jud(i,j)&&i!=j)insert(i,j+1000,1,i+j);
    for(int i=a;i<=b;i++)
        insert(0,i,1,0),insert(i+1000,T,1,0);
    while(spfa())dfs();
    for(int i=2;i<=cnt;i+=2)
        if(e[i].to==T&&!e[i].v)ans1++;
    printf("%d %d",ans1/2,ans2/2);
    return 0;
}

3171: [Tjoi2013]循环格
Time Limit: 1 Sec Memory Limit: 128 MB
Submit: 742 Solved: 464
[Submit][Status][Discuss]

Description
一个循环格就是一个矩阵，其中所有元素为箭头，指向相邻四个格子。每个元素有一个坐标（行，列），其中左上角元素坐标为（0,0）。给定一个起始位置（r，c）
，你可以沿着箭头防线在格子间行走。即如果（r,c）是一个左箭头，那么走到（r，c-1）;如果是右箭头那么走到（r，c+1）；如果是上箭头那么走到（r-1，c）；如果是下箭头那么走到（r+1，c）；每一行和每一列都是循环的，即如果走出边界，你会出现在另一侧。
一个完美的循环格是这样定义的：对于任意一个起始位置，你都可以i沿着箭头最终回到起始位置。如果一个循环格不满足完美，你可以随意修改任意一个元素的箭头直到完美。给定一个循环格，你需要计算最少需要修改多少个元素使其完美。

Input
第一行两个整数R，C。表示行和列，接下来R行，每行C个字符LRUD，表示左右上下。

Output
一个整数，表示最少需要修改多少个元素使得给定的循环格完美

Sample Input
3 4
RRRD
URLL
LRRR

Sample Output
2

HINT
1<=R,L<=15

Source

建模：
同样拆点
由源到各入点，连容量为1，费用为0；
由各出点向汇点，连容量为1，费用为0；
各点向周围4个点练容量为1，费用0 or 1；（初始可到为1，不可为0）
最小费用即为答案

code by hzwer：

#include<iostream>
#include<cstdio>
#include<cstring>
#define inf 0x7fffffff
#define T 600
using namespace std;
int n,m,cnt=1,ans;
int mp[16][16],mark[16][16],xx[4]={0,0,-1,1},yy[4]={-1,1,0,0};
int from[601],head[601],q[601],dis[601];
bool inq[601];
struct data{int to,from,next,v,c;}e[1000001];
void ins(int u,int v,int w,int c)
{
     e[++cnt].to=v;e[cnt].from=u;
     e[cnt].next=head[u];head[u]=cnt;
     e[cnt].v=w;e[cnt].c=c;
}
void insert(int u,int v,int w,int c)
{ins(u,v,w,c);ins(v,u,0,-c);}
void build()
{
     for(int i=1;i<=n;i++)
        for(int j=1;j<=m;j++)
            for(int k=0;k<4;k++)
            {
                int nowx=i+xx[k],nowy=j+yy[k];
                if(nowx>n)nowx=1;if(nowx<1)nowx=n;
                if(nowy>m)nowy=1;if(nowy<1)nowy=m;
                if(k==mp[i][j])
                   insert(mark[i][j],mark[nowx][nowy]+300,1,0);
                else insert(mark[i][j],mark[nowx][nowy]+300,1,1);
                }
     for(int i=1;i<=n*m;i++)
     {insert(0,i,1,0);insert(i+300,T,1,0);}
     }
bool spfa()
{
     for(int i=0;i<=T;i++)dis[i]=inf;
    int t=0,w=1,i,now;
    dis[0]=0;inq[0]=1;q[0]=0;
    while(t!=w)
    {
        now=q[t];t++;if(t==601)t=0;
        for(i=head[now];i;i=e[i].next)
        {
                if(e[i].v&&e[i].c+dis[now]<dis[e[i].to])
                {
                    dis[e[i].to]=e[i].c+dis[now];
                    from[e[i].to]=i;
                    if(!inq[e[i].to])
                    {
                        inq[e[i].to]=1;
                        q[w++]=e[i].to;
                        if(w==601)w=0; 
                                     }
                }
        }
        inq[now]=0;
    }
    if(dis[T]==inf)return 0;return 1;
}
void mcf()
{
    int x=inf;
    for(int i=from[T];i;i=from[e[i].from])
    {
        x=min(e[i].v,x);
    }
    for(int i=from[T];i;i=from[e[i].from])
    {
        e[i].v-=x;e[i^1].v+=x;
        ans+=x*e[i].c;
    }
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)
       for(int j=1;j<=m;j++)
          mark[i][j]=(i-1)*m+j;
    for(int i=1;i<=n;i++)
    {
       char ch[21];scanf("%s",ch);
       for(int j=1;j<=m;j++)
       {
          if(ch[j-1]=='L')mp[i][j]=0;
          if(ch[j-1]=='R')mp[i][j]=1;
          if(ch[j-1]=='U')mp[i][j]=2;
          if(ch[j-1]=='D')mp[i][j]=3;
          }
       }
    build();
    while(spfa())mcf();
    printf("%d",ans);
    return 0;
 }