poj 1201 Intervals

题目描述

有一个序列，输入n及n个整数组合 ai，bi，ci，[ai，bi，ci]表示在该序列中处于[ai，bi]这个区间的整数至少有ci个。如果存在这样的序列，请求出满足题目要求的最短的序列长度是多少。如果不存在则输出 -1。

Simple input

5
3 7 3
8 10 3
6 8 1
1 3 1
10 11 1

Simple output

6

数据范围

1 <= n <= 50000
0<= ai <= bi <= 50000
1 <= ci <= bi - ai+1

解题思路

差分约束系统例题。

可以将问题转化为1到max（b[i])个物品中，每个选或不选，在a[i]到b[i]中，至少选了c[i]个。那么，如果用一个数组T维护前缀和，那么有
t[b[i]]-t[a[i]-1]>=c[i]
=>t[b[i]]-c[i]>=t[a[i]-1]
于是由b[i]往a[i]-1连一条边权为-c[i]的边。
当然，由于每个只能选或不选，所以有
0<=t[i]-t[i-1]<=1
=>t[i]+0>=t[i-1] 由i往i-1连一条边权为0的边
=>t[i-1]+1>=t[i] 由i-1往i连一条边权为1的边

因为单源最短路跑完后满足dis[i]+w[i,j]>=dis[j](w[i,j]为边权，dis就相当于上面的T数组）
所以有n出发做一遍spfa就可以了。

PS：1.好像不可以从1出发，这样连上的负边权没有作用了，会输出max(b[i])-1.
2.最后输出t[0],因为若a[i]=1那么会连一条边到0，这样会有错误。
3.t[0]为负值，输出时取相反数。

Source Code

#include<iostream>#include<algorithm>#include<cstdio>#include<cstring>#include<queue>using namespace std;int a[50010],b[50010],c[50010],t[50010];int s[150010][4],next[150010],last[50010];int n,m;void init(){    scanf("%d",&n);    int i,l=0;    memset(last,0,sizeof(last));    for (i=1;i<=n;i++)    {        scanf("%d%d%d",&a[i],&b[i],&c[i]);        if (b[i]>m) m=b[i];        l++;        s[l][1]=b[i];s[l][2]=a[i]-1;s[l][3]=-c[i];        next[l]=last[s[l][1]];last[s[l][1]]=l;    }    for (i=1;i<=m;i++)    {        l++;        s[l][1]=i;s[l][2]=i-1;s[l][3]=0;        next[l]=last[s[l][1]];last[s[l][1]]=l;        l++;        s[l][1]=i-1;s[l][2]=i;s[l][3]=1;        next[l]=last[s[l][1]];last[s[l][1]]=l;      }    /*for (i=1;i<=l;i++)      printf("%d %d %d %d %d\n",i,s[i][1],s[i][2],s[i][3],next[i]);    for (i=0;i<=m;i++)      printf("%d \n",last[i]);*/    for (i=0;i<=m;i++)      t[i]=500000;}queue<int> Q;int p[50010];void spfa(){    int i,j,k;    memset(p,0,sizeof(p));    Q.push(m);    t[m]=0;    p[m]=1;    while (!Q.empty())    {        int now=Q.front();        i=last[now];        Q.pop();        while (i!=0)        {            k=s[i][1];j=s[i][2];            if (t[k]+s[i][3]<t[j])            {                /*printf("%d %d %d %d\n",i,k,j,p[j]);*/                t[j]=t[k]+s[i][3];                if (p[j]!=1){Q.push(j);p[j]=1;}            }            i=next[i];          }        p[now]=0;    }       }int main(){    init();    spfa();    printf("%d",-t[0]);    return 0;}