POJ 3169 差分约束系统

来源:互联网 发布:access数据库2007 编辑:程序博客网 时间:2024/05/22 03:14
大致题意:

    有n头牛,按照序号1……n排队,多头牛可以站在同一个位置(暂且这么认为),也就是任意两头牛之间的距离都大于等于0。先给出ml组约束关系(u,v,w)代表第u牛和第v牛之间的距离要小于等于w。再给出md组关系(u,v,w),代表第u牛和第v牛之间的距离要大于等于w。求这n头牛排成的队伍能否符合以上的约束,不能的话(也就是出现负环),输出“-1”,如果距离是inf,输出“-2”。否则输出这个最短距离。

大致思路:

    转化为差分约束模型。dis[i]表示第i头牛所在的位置。如果spfa判断出负环的话就输出-1,如果dis[n]==inf则输出-2。否则输出dis[n]。

#include<cstdio>#include<queue>#include<cstring>using namespace std;const int inf=1<<30,maxv=1005,maxe=20005;struct edge{int u,v,k,next;void init(int tu,int tv,int tk,int tnext){u=tu, v=tv, k=tk, next=tnext;}}e[maxe];int link[maxv],cnt[maxv],d[maxv];int ne;bool vis[maxv];void addedge(int u,int v,int k){e[ne].init(u,v,k,link[u]);link[u]=ne++;}int spfa(int s,int t){int i,n=t-s;for(i=s;i<=t;i++){d[i]=inf;}queue<int> q;memset(vis,0,sizeof(vis));memset(cnt,0,sizeof(cnt));q.push(s);vis[s]=true;d[s]=0;cnt[s]++;while(!q.empty()){int p,v,u=q.front();q.pop();vis[u]=false;for(p=link[u];p!=-1;p=e[p].next){v=e[p].v;if(d[v]>d[u]+e[p].k){d[v]=d[u]+e[p].k;cnt[v]++;if(cnt[v]>n){return -1;}if(!vis[v]){vis[v]=true;q.push(v);}}}}if(d[t]==inf){return -2;}return d[t];}int main(){int n,ml,md;while(~scanf("%d%d%d",&n,&ml,&md)){ne=0;memset(link,-1,sizeof(link));int u,v,k;while(ml--){scanf("%d%d%d",&u,&v,&k);addedge(u,v,k);}while(md--){scanf("%d%d%d",&u,&v,&k);addedge(v,u,-k);}printf("%d\n",spfa(1,n));}return 0;}




原创粉丝点击