C/C++,图算法——求强联通的Tarjan算法之源程序

1 文本格式

#include <bits/stdc++.h>

using namespace std;

const int maxn = 1e4 + 5;

const int maxk = 5005;

int n, k;

int id[maxn][5];

char s[maxn][5][5], ans[maxk];

bool vis[maxn];

struct Edge {

int v, nxt;

} e[maxn * 100];

int head[maxn], tot = 1;

void addedge(int u, int v) {

e[tot].v = v;

e[tot].nxt = head[u];

head[u] = tot++;

}

int dfn[maxn], low[maxn], color[maxn], stk[maxn], ins[maxn], top, dfs_clock, c;

void tarjan(int x) { // tarjan算法求强联通

stk[++top] = x;

ins[x] = 1;

dfn[x] = low[x] = ++dfs_clock;

for (int i = head[x]; i; i = e[i].nxt) {

int v = e[i].v;

if (!dfn[v]) {

tarjan(v);

low[x] = min(low[x], low[v]);

} else if (ins[v])

low[x] = min(low[x], dfn[v]);

}

if (dfn[x] == low[x]) {

c++;

do {

color[stk[top]] = c;

ins[stk[top]] = 0;

} while (stk[top--] != x);

}

}

int main() {

scanf("%d %d", &k, &n);

for (int i = 1; i <= n; i++) {

for (int j = 1; j <= 3; j++) scanf("%d%s", &id[i][j], s[i][j]);

for (int j = 1; j <= 3; j++) {

for (int k = 1; k <= 3; k++) {

if (j == k) continue;

int u = 2 * id[i][j] - (s[i][j][0] == 'B');

int v = 2 * id[i][k] - (s[i][k][0] == 'R');

addedge(u, v);

}

}

}

for (int i = 1; i <= 2 * k; i++)

if (!dfn[i]) tarjan(i);

for (int i = 1; i <= 2 * k; i += 2)

if (color[i] == color[i + 1]) {

puts("-1");

return 0;

}

for (int i = 1; i <= 2 * k; i += 2) {

int f1 = color[i], f2 = color[i + 1];

if (vis[f1]) {

ans[(i + 1) >> 1] = 'R';

continue;

}

if (vis[f2]) {

ans[(i + 1) >> 1] = 'B';

continue;

}

if (f1 < f2) {

vis[f1] = 1;

ans[(i + 1) >> 1] = 'R';

} else {

vis[f2] = 1;

ans[(i + 1) >> 1] = 'B';

}

}

ans[k + 1] = 0;

printf("%s\n", ans + 1);

return 0;

}

2 代码格式

cpp 复制代码
#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e4 + 5;
const int maxk = 5005;

int n, k;
int id[maxn][5];
char s[maxn][5][5], ans[maxk];
bool vis[maxn];

struct Edge {
  int v, nxt;
} e[maxn * 100];

int head[maxn], tot = 1;

void addedge(int u, int v) {
  e[tot].v = v;
  e[tot].nxt = head[u];
  head[u] = tot++;
}

int dfn[maxn], low[maxn], color[maxn], stk[maxn], ins[maxn], top, dfs_clock, c;

void tarjan(int x) {  // tarjan算法求强联通
  stk[++top] = x;
  ins[x] = 1;
  dfn[x] = low[x] = ++dfs_clock;
  for (int i = head[x]; i; i = e[i].nxt) {
    int v = e[i].v;
    if (!dfn[v]) {
      tarjan(v);
      low[x] = min(low[x], low[v]);
    } else if (ins[v])
      low[x] = min(low[x], dfn[v]);
  }
  if (dfn[x] == low[x]) {
    c++;
    do {
      color[stk[top]] = c;
      ins[stk[top]] = 0;
    } while (stk[top--] != x);
  }
}

int main() {
  scanf("%d %d", &k, &n);
  for (int i = 1; i <= n; i++) {
    for (int j = 1; j <= 3; j++) scanf("%d%s", &id[i][j], s[i][j]);

    for (int j = 1; j <= 3; j++) {
      for (int k = 1; k <= 3; k++) {
        if (j == k) continue;
        int u = 2 * id[i][j] - (s[i][j][0] == 'B');
        int v = 2 * id[i][k] - (s[i][k][0] == 'R');
        addedge(u, v);
      }
    }
  }

  for (int i = 1; i <= 2 * k; i++)
    if (!dfn[i]) tarjan(i);

  for (int i = 1; i <= 2 * k; i += 2)
    if (color[i] == color[i + 1]) {
      puts("-1");
      return 0;
    }

  for (int i = 1; i <= 2 * k; i += 2) {
    int f1 = color[i], f2 = color[i + 1];
    if (vis[f1]) {
      ans[(i + 1) >> 1] = 'R';
      continue;
    }
    if (vis[f2]) {
      ans[(i + 1) >> 1] = 'B';
      continue;
    }
    if (f1 < f2) {
      vis[f1] = 1;
      ans[(i + 1) >> 1] = 'R';
    } else {
      vis[f2] = 1;
      ans[(i + 1) >> 1] = 'B';
    }
  }
  ans[k + 1] = 0;
  printf("%s\n", ans + 1);
  return 0;
}
相关推荐
海洲探索-Hydrovo1 小时前
TTP Aether X 天通透传模块丨国产自主可控大数据双向通讯定位模组
网络·人工智能·科技·算法·信息与通信
2401_841495644 小时前
【计算机视觉】基于复杂环境下的车牌识别
人工智能·python·算法·计算机视觉·去噪·车牌识别·字符识别
Jonkin-Ma4 小时前
每日算法(1)之单链表
算法
奔跑吧邓邓子5 小时前
【C语言实战(8)】C语言循环结构(do-while):解锁编程新境界
c语言·实战·do-while
晚风残5 小时前
【C++ Primer】第六章:函数
开发语言·c++·算法·c++ primer
小莞尔5 小时前
【51单片机】【protues仿真】基于51单片机温度测量系统
c语言·单片机·嵌入式硬件·物联网·51单片机
杨云强5 小时前
离散积分,相同表达式数组和公式
算法
地平线开发者5 小时前
征程 6 | BPU trace 简介与实操
算法·自动驾驶
满天星83035775 小时前
【C++】AVL树的模拟实现
开发语言·c++·算法·stl
晓风凌殇5 小时前
单片机按键检测与长短按识别实现
c语言·单片机