银河「建图 + Tarjan缩点 + 拓扑排序 + 最长路」
题目描述:
思路1「差分约束」:
#include <bits/stdc++.h>
using namespace std;
#define endl '\n'
#define inf 0x3f3f3f3f
#define mod7 1000000007
#define mod9 998244353
#define m_p(a,b) make_pair(a, b)
#define mem(a,b) memset((a),(b),sizeof(a))
#define io ios::sync_with_stdio(false); cin.tie(0); cout.tie(0)
#define debug(a) cout << "Debuging...|" << #a << ": " << a << "\n";
typedef long long ll;
typedef pair <int,int> pii;
#define MAX 300000 + 50
int n, m, k;
int tot;
int head[MAX];
struct ran{
int to, nex, val;
}tr[MAX];
inline void add(int u, int v, int c){
tr[++tot].to = v;
tr[tot].val = c;
tr[tot].nex = head[u];
head[u] = tot;
}
int dis[MAX];
bool vis[MAX];
int num[MAX];
void SPFA(){
deque<int>q;
for(int i = 1; i <= n; ++i){
q.push_back(i);dis[i] = 1;vis[i] = 1;
}
while (!q.empty()) {
int u = q.front();q.pop_front();vis[u] = 0;
for(int i = head[u]; i; i = tr[i].nex){
int v = tr[i].to;
if(dis[v] < dis[u] + tr[i].val){
dis[v] = dis[u] + tr[i].val;
num[v] = num[u] + 1;
if(num[v] >= n){
cout << -1 << endl;
return;
}
if(!vis[v]){
if(dis[v] > dis[q.front()])q.push_front(v);
else q.push_back(v);
vis[v] = 1;
}
}
}
}
ll ans = 0;
for(int i = 1; i <= n; ++i){
ans += dis[i];
}
cout << ans << endl;
}
void work(){
cin >> n >> m;
for(int i = 1, op, a, b; i <= m; ++i){
cin >> op >> a >> b;
if(op == 1){
add(a, b, 0);
add(b, a, 0);
}
else if(op == 2)add(a, b, 1);
else if(op == 3)add(b, a, 0);
else if(op == 4)add(b, a, 1);
else add(a, b, 0);
}
SPFA();
}
int main(){
io;
work();
return 0;
}
思路2「Tarjan缩点 + 拓扑排序 + 最长路」
#include <bits/stdc++.h>
using namespace std;
#define endl '\n'
#define inf 0x3f3f3f3f
#define mod7 1000000007
#define mod9 998244353
#define m_p(a,b) make_pair(a, b)
#define mem(a,b) memset((a),(b),sizeof(a))
#define io ios::sync_with_stdio(false); cin.tie(0); cout.tie(0)
#define debug(a) cout << "Debuging...|" << #a << ": " << a << "\n";
typedef long long ll;
typedef pair <int,int> pii;
#define MAX 300000 + 50
int n, m, k;
int tot;
int head[MAX];
struct ran{
int to, nex, val;
}tr[MAX];
inline void add(int u, int v, int c){
tr[++tot].to = v;
tr[tot].val = c;
tr[tot].nex = head[u];
head[u] = tot;
}
int tmd;
int dfn[MAX], low[MAX];
bool vis[MAX];
int cnt;
int col[MAX];
int num[MAX];
int du[MAX];
stack<int>st;
void tarjan(int u){
st.push(u);vis[u] = 1;
dfn[u] = low[u] = ++tmd;
for(int i = head[u]; i; i = tr[i].nex){
int v = tr[i].to;
if(!dfn[v]){
tarjan(v);
low[u] = min(low[u], low[v]);
}
else if(vis[v])low[u] = min(low[u], dfn[v]);
}
if(low[u] == dfn[u]){
int now;
++cnt;
do {
now = st.top();
st.pop();vis[now] = 0;
col[now] = cnt;
++num[cnt];
} while (now != u);
}
}
vector<pii>G[MAX];
int dis[MAX];
void topu(){
queue<int>q;
for(int i = 1; i <= cnt; ++i){
if(!du[i]){
q.push(i);
dis[i] = 1;
}
}
ll ans = 0;
while (!q.empty()) {
int u = q.front();q.pop();
ans += dis[u] * num[u];
for(auto [v, d] : G[u]){
--du[v];
if(!du[v])q.push(v);
if(dis[v] < dis[u] + d){
dis[v] = dis[u] + d;
}
}
}
cout << ans << endl;
}
void work(){
cin >> n >> m;
for(int i = 1, op, a, b; i <= m; ++i){
cin >> op >> a >> b;
if(op == 1){
add(a, b, 0);
add(b, a, 0);
}
else if(op == 2)add(a, b, 1);
else if(op == 3)add(b, a, 0);
else if(op == 4)add(b, a, 1);
else add(a, b, 0);
}
for(int i = 1; i <= n; ++i){
if(!dfn[i])tarjan(i);
}
for(int u = 1; u <= n; ++u){
for(int i = head[u]; i; i = tr[i].nex){
int v = tr[i].to;
if(col[u] == col[v]){
if(tr[i].val != 0){
cout << -1 << endl;
return;
}
}
else {
G[col[u]].push_back(m_p(col[v], tr[i].val));
++du[col[v]];
}
}
}
topu();
}
int main(){
io;
work();
return 0;
}