sgu 213 分类： sgu 2015-06...

最新推荐文章于 2019-01-08 08:27:59 发布

aa288288

最新推荐文章于 2019-01-08 08:27:59 发布

阅读量115

点赞数

原文链接：http://www.cnblogs.com/dashgua/p/4722998.html

版权

SPFA最短路，dist(s→t)即为最少颜色数，
答案可以这样构造：对于一条边(u,v)，染色为min(max(dist(u),dist(v)),dist(t))

为什么可以这么构造？

我们可以发现，所有边都为1的情况下, 对于一条边(u,v), dist(v)<=dist(u)+1

按 dist 的取值，可以把原图划分成一个层次图，
这样就可以保证起点到终点最少跨过了dist(t)层，

那么只要把连接两个不在同一层的结点的边染色，
这样就可以保证起点到终点最少经过了dist(t)种不同颜色。

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

#include<ctime>
#include<cmath>

const int maxn = 405, INF = 0x3f3f3f3f;
#define fr first
#define to second
#define mp(x,y) std::make_pair(x,y)
#define pb(x) push_back(x)

int n, m, s, t;
bool w[maxn][maxn];
int dist[maxn];
bool hash[maxn];
std::queue<int> line;
std::pair<int,int> edge[maxn*maxn];
std::vector<int> color[maxn*maxn];

int SPFA()
{
    memset(dist,INF,sizeof(dist));

    dist[s] = 0, line.push(s), hash[s] = true;

    while(true)
    {
        if(line.empty()) break;

        int u = line.front(); 
        hash[u] = false, line.pop();

        for(int i = 1; i <= n; i++)
            if(w[u][i] && dist[u]+1 < dist[i])
            {
                dist[i] = dist[u]+1;
                if(!hash[i]) 
                    line.push(i), hash[i] = true;
            }   
    }

    return dist[t];
}

void prtcolor()
{
    for(int i = 1; i <= m; i++)
    {
        int opt = std::max(dist[edge[i].fr], dist[edge[i].to]);
        color[std::min(opt, dist[t])].pb(i);
    }

    for(int i = 1; i <= dist[t]; i++)
    {
        std::cout << color[i].size();
        for(int j = 0; j < color[i].size(); j++)
            std::cout << ' ' << color[i][j];
        std::cout << std::endl; 
    }
}

int main()
{
#ifndef ONLINE_JUDGE
    freopen("sgu213.in","r",stdin);
    freopen("sgu213.out","w",stdout);
#endif

    std::cin >> n >> m >> s >> t;

    for(int i = 1, u, v; i <= m; i++)
    {
        std::cin >> u >> v, edge[i] = mp(u,v);
        w[u][v] = w[v][u] = true;
    }

    std::cout << SPFA() << std::endl, prtcolor();

#ifndef ONLINE_JUDGE
    fclose(stdin);
    fclose(stdout);
#endif
    return 0;       
}