1. 程式人生 > >【BZOJ2330】糖果

【BZOJ2330】糖果

題目連結:https://www.lydsy.com/JudgeOnline/problem.php?id=2330


 

差分約束系統的模板題吧,一看題目各種不等式。。。

因為是要求最小值,所以我們選擇跑最長路好了。根據題目描述建邊,這個應該不難,x=1就是建從a到b和從b到a的邊權為0的邊,x=2就是建從a到b邊權為1的邊,x=3是建從b到a邊權為0的邊,x=4是建從b到a邊權為1的邊,x=5是建從a到b邊權為0的邊。

注意,當x=2或4時,肯定是無解的,據說,有些SPFA會被卡,反正我的不會。。。

然後我們需要加一個超級源,但注意,如果從1到n逐個與0連邊,對於某個測試點就會超時,因為人家是一條超長的鏈,所以應該倒過來,從n到1連邊,當然,如果鄰接連結串列不是從頭部插入請忽略。

還有答案會爆int。

 1 #include <cstdio>
 2 #include <cstring>
 3 #include <queue>
 4 
 5 using namespace std;
 6 
 7 inline int get_num() {
 8     int num = 0;
 9     char c = getchar();
10     while (c < '0' || c > '9') c = getchar();
11     while (c >= '0' && c <= '
9') 12 num = num * 10 + c - '0', c =getchar(); 13 return num; 14 } 15 16 const int maxn = 1e5 + 5, maxk = 1e5 + 5; 17 18 int head[maxn], eid; 19 20 struct Edge { 21 int v, w, next; 22 } edge[2 * maxk + maxn]; 23 24 inline void insert(int u, int v, int w) { 25 edge[++eid].v = v;
26 edge[eid].w = w; 27 edge[eid].next = head[u]; 28 head[u] = eid; 29 } 30 31 int n, k, inq[maxn], cnt[maxn]; 32 long long dist[maxn]; 33 34 queue<int> q; 35 36 inline int spfa() { 37 inq[0] = 1; 38 q.push(0); 39 while (!q.empty()) { 40 int u = q.front(); 41 q.pop(); 42 inq[u] = 0; 43 if (++cnt[u] >= n) return 0; 44 for (int p = head[u]; p; p = edge[p].next) { 45 int v = edge[p].v, w = edge[p].w; 46 if (dist[v] < dist[u] + w) { 47 dist[v] = dist[u] + w; 48 if (!inq[v]) {q.push(v); inq[v] = 1;} 49 } 50 } 51 } 52 return 1; 53 } 54 55 int main() { 56 n = get_num(), k = get_num(); 57 for (int i = 1; i <= k; ++i) { 58 int x = get_num(), a = get_num(), b = get_num(); 59 if (x == 1) {insert(a, b, 0); insert(b, a, 0);} 60 else if (x == 2) { 61 if (a == b) {printf("-1"); return 0;} 62 insert(a, b, 1); 63 } 64 else if (x == 3) insert(b, a, 0); 65 else if (x == 4) { 66 if (a == b) {printf("-1"); return 0;} 67 insert(b, a, 1); 68 } 69 else insert(a, b, 0); 70 } 71 for (int i = n; i >= 1; --i) insert(0, i, 1); 72 if (!spfa()) printf("-1"); 73 else { 74 long long ans = 0; 75 for (int i = 1; i <= n; ++i) ans += dist[i]; 76 printf("%lld", ans); 77 } 78 return 0; 79 }
AC程式碼