链接:http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemId=1770
题意是陆逊能根据动静看到刘备某几个连续大营里至少有多少士兵,并且给出每个营的容纳极限,问刘备阵营至少有多少兵。
是一道标准的差分约束的题。建立一个虚拟兵营零,容纳是零,作用是把所有的大营人数连接起来。判断能否建立连通图只需要先在输入时判断是否有某段最少士兵数超过该段兵营容纳极限。数组建立aa[i]-aa[0]表示前i号营至少有多少士兵,aa[tot]-aa[0]即为刘备至少有多少士兵。用SPFA建立单源最短路,建立从i-1->i的权为容纳量的路来标记各营至少有多少士兵。
代码:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<map>
#include<queue>
#include<stack>
#include<vector>
#include<ctype.h>
#include<algorithm>
#include<string>
#define PI acos(-1.0)
#define maxn 1005
#define maxm 10005
#define INF 1<<25
#define MAX 0x7fffffff
typedef long long ll;
using namespace std;
struct Edge
{
int v,w;
int next;
} edge [maxm];
int top=0,point[maxn],vv[maxn],dis[maxn],jud,t,ans,tot,tt,aa[maxn];
int init()
{
top=0,jud=0,t=0,ans=0;
memset(point,-1,sizeof(point));
memset(vv,0,sizeof(vv));
memset(dis,0,sizeof(dis));
memset(edge,0,sizeof(edge));
}
int add_edge(int u,int v,int w)
{
edge[top].v=v;
edge[top].w=w;
edge[top].next=point[u];
point[u]=top++;
}
int main()
{
while(scanf("%d%d",&tot,&tt)!=EOF)
{
init();
for(int i=1; i<=tot; i++)
{
scanf("%d",&aa[i]);
t+=aa[i];
aa[i]=t;
}
for(int i=0; i<tt; i++)
{
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
if(c>aa[b]-aa[a-1])
jud=1;
add_edge(b,a-1,-c);
}
for(int i=1; i<=tot; i++)
{
add_edge(i,i-1,0);
add_edge(i-1,i,aa[i]-aa[i-1]);
}
if(!jud)
{
queue <int> qq;
qq.push(tot);
vv[tot]=1;
dis[tot]=0;
while(!qq.empty())
{
int T=qq.front();
qq.pop();
vv[T]=0;
for(int i=point[T]; i!=-1; i=edge[i].next)
{
if(dis[T]+edge[i].w<dis[edge[i].v])
{
dis[edge[i].v]=dis[T]+edge[i].w;
if(vv[edge[i].v]!=1)
{
vv[edge[i].v]=1;
qq.push(edge[i].v);
}
}
}
}
}
if(jud)
printf("Bad Estimations\n");
else
{
printf("%d\n",dis[tot]-dis[0]);
}
}
}