差分约束 - Partial Teacher - CodeForces 67A

差分约束 - Partial Teacher - CodeForces 67A

题意：

$老师分糖果，有n名同学，$

$给定一个长度为n-1的字符串，{由}^{\prime }{L}^{\prime }{,}^{\prime }{R}^{\prime }{,}^{\prime }{=}^{\prime }组成，$

$若第i位字符{为}^{\prime }{L}^{\prime }，则表示第i位同学的成绩高于第i+1位同学的成绩。$

$若第i位字符{为}^{\prime }{R}^{\prime }，则表示第i+1位同学的成绩高于第i位同学的成绩。$

$若第i位字符{为}^{\prime }{=}^{\prime }，则表示第i位同学的成绩与第i+1位同学的成绩相等。$

$成绩高的同学要求糖果数量多于相邻的成绩较低的同学。$

$每位同学至少分得一个糖果，计算老师至少需要准备多少糖果。$

输入：

$首行包括一个正整数n，表示学生的数量，$

$接着输入一个长度为n-1的字符串，表示相邻两个同学之间的关系。$

输出：

$一个正整数，表示老师至少需要准备的糖果的数量。$

Examples
Input

5
LRLR

Output

2 1 2 1 2

Input

5
=RRR

Output

1 1 2 3 4

数据范围：

$2\le n\le 1000$

---

分析：

$裸的差分约束题，记第i个同学得到的糖果数量为C_i，$

$要求最小值，跑最长路，将不等式转化为A\ge B+k的形式。$

$①、L：C_i\ge C_{i+1}+1，建边：i+1->i，边权为1$

$②、R：C_{i+1}\ge C_i+1，建边：i->i+1，边权为0$

$③、=：C_i=C_{i+1}即C_i\ge C_{i+1}且C_{i+1}\ge C_i，建边:i-i+1，无向边，边权为0。$

$④、C_i\ge 1$

$由于图不一定连通，因此我们需要遍历所有未被计算过的点。$

$建立虚拟源点0似乎会TLE，要避免重复更新同一个点，$

$由于需要保证条件④，我们每次从未被计算过的点开始跑一遍单源最短路，源点的dis初始化为1即可。$

$因为所有边的权值都是大于等于0的，就能够保证其他点的dis大于等于1.$

注意：

$跑最长路，距离数组应当初始化一个较小的值。$

代码：

#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>

using namespace std;

const int N=1010, M=4*N, inf=0x3f3f3f3f;

int n;
int e[M],ne[M],w[M],h[N],idx;
int dis[N];
bool st[N];
char str[N];

void add(int a,int b, int c)
{
    e[idx]=b,w[idx]=c,ne[idx]=h[a],h[a]=idx++;
}

void spfa(int S)
{
    queue<int> Q;
    Q.push(S);
    dis[S]=1;
    st[S]=true;

    while(Q.size())
    {
        int u=Q.front();
        Q.pop();
        st[u]=false;
        for(int i=h[u];~i;i=ne[i])
        {
            int j=e[i];
            if(dis[j]<dis[u]+w[i])
            {
                dis[j]=dis[u]+w[i];
                if(!st[j])
                {
                    st[j]=true;
                    Q.push(j);
                }
            }
        }
    }
}

int main()
{
    scanf("%d",&n);
    scanf("%s",str+1);

    memset(h,-1,sizeof h);
    for(int i=1;i<n;i++)
        if(str[i]=='L') add(i+1,i,1);  //i>=(i+1)+1
        else if(str[i]=='R') add(i,i+1,1);
        else add(i,i+1,0), add(i+1,i,0);

    memset(dis,-1,sizeof dis);
    for(int i=1;i<=n;i++)
        if(dis[i]==-1)
            spfa(i);
    
    for(int i=1;i<=n;i++) printf("%d ",dis[i]);
    
    return 0;
}