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;
}
相关推荐
写个博客6 分钟前
暑假算法日记第一天
算法
绿皮的猪猪侠8 分钟前
算法笔记上机训练实战指南刷题
笔记·算法·pta·上机·浙大
hie988941 小时前
MATLAB锂离子电池伪二维(P2D)模型实现
人工智能·算法·matlab
杰克尼1 小时前
BM5 合并k个已排序的链表
数据结构·算法·链表
.30-06Springfield2 小时前
决策树(Decision tree)算法详解(ID3、C4.5、CART)
人工智能·python·算法·决策树·机器学习
我不是哆啦A梦2 小时前
破解风电运维“百模大战”困局,机械版ChatGPT诞生?
运维·人工智能·python·算法·chatgpt
xiaolang_8616_wjl2 小时前
c++文字游戏_闯关打怪
开发语言·数据结构·c++·算法·c++20
small_wh1te_coder2 小时前
硬件嵌入式学习路线大总结(一):C语言与linux。内功心法——从入门到精通,彻底打通你的任督二脉!
linux·c语言·汇编·嵌入式硬件·算法·c
挺菜的2 小时前
【算法刷题记录(简单题)002】字符串字符匹配(java代码实现)
java·开发语言·算法
凌肖战6 小时前
力扣网编程55题:跳跃游戏之逆向思维
算法·leetcode