poj 1201 差分约束
大致题意:给出n个闭区间[ai,bi],每个区间对应一个ci,表示集合Z在区间[ai,bi]内ci个相同元素,问集合Z至少有几个元素。
思路:将区间[ai,bi]的每个端点看做一个顶点,以dist[i+1]表示源点x到i之间至少有与集合Z相同元素的个数。如果将至少含有的相同元素的个数看做边的权值则一个区间[ai,bi]的两个端点的dist[ai]和dist[bi+1]分别表示源点x到ai-1和bi的距离。那么区间[ai,bi+1]又可以看做什么呢?可以看做ai到bi的一条边。依据题意 x、ai、bi构成了一个边权三角形dist[bi+1]-dist[ai]>=ai->bi,即dist[bi+1]-dist[ai]>=ci;所以变形一下就得:dist[bi+1]>=dist[ai]+ci;注意:这个结果只是题目的要求,所以我们if(dist[bi+1]<dist[ai]+ci) dist[bi+1]=dist[ai]+ci;这也决定了求的是最长路径(当然有的人求的是最短路也行)。
如果只有题目所给的约束条件的话,是远远不够的。所以还需要挖掘隐藏的条件:0<=dist[i+1]-dist[i]<=1。有了这个条件我们便可以建立数据结构了,然后我们以出现的最小元素min1为源点spfa,dist[max1]就是解了。
#include<iostream>
#include<cstring>
#include<queue>
using namespace std;
#define MAX_INT 123456789
struct node
{
int v;
int value;
int next;
};
int head[51000],dist[51000],visit[51000],N;
node edge[151000];
queue <int> Q;
int add(int s,int t,int w)
{
node e={t,w,0};
edge[N]=e;
edge[N].next=head[s];
head[s]=N++;
return 0;
}
int spfa(int min1,int max1)
{
int i,j,e,v;
for(i=min1;i<=max1;i++)
dist[i]=-MAX_INT;
Q.push(min1); visit[min1]=1;
dist[min1]=0;
while(!Q.empty())
{
e=Q.front(),Q.pop();
visit[e]=0;
for(j=head[e];j;j=edge[j].next)
if(dist[v=edge[j].v]<dist[e]+edge[j].value)
{
dist[v]=dist[e]+edge[j].value;
if(!visit[v])
{
visit[v]=1;
Q.push(v);
}
}
}
return dist[max1];
}
int main()
{
int i,n,s,t,w,min1,max1;
while(cin>>n)
{
min1=MAX_INT; max1=-MAX_INT;
memset(head,0,sizeof(head));
for(N=1,i=0;i<n;i++)
{
cin>>s>>t>>w;
add(s,t+1,w);
if(min1>s) min1=s;
if(max1<t+1) max1=t+1;
}
for(i=min1; i<max1;i++)
{
add(i,i+1,0);
add(i+1,i,-1);
}
cout<<spfa(min1,max1)<<endl;
}
return 0;
}
浙公网安备 33010602011771号