题目描述
A 国有 n 座城市,编号从 1到 n,城市之间有 m 条双向道路。每一条道路对车辆都有重量限制,简称限重。
现在有 q 辆货车在运输货物, 司机们想知道每辆车在不超过车辆限重的情况下,最多能运多重的货物。
.
输入格式
第一行有两个用一个空格隔开的整数 n,m,表示 A 国有 n 座城市和 m 条道路。
接下来 m行每行三个整数 x,y,z,每两个整数之间用一个空格隔开,表示从 x 号城市到 y号城市有一条限重为 z的道路。
注意: x≠y, 两座城市之间可能有多条道路 。
接下来一行有一个整数 q,表示有 q 辆货车需要运货。
接下来 q 行,每行两个整数 x,y,之间用一个空格隔开,表示一辆货车需要从 x 城市运输货物到 y 城市,保证 x≠y
.
输出格式
共有 q行,每行一个整数,表示对于每一辆货车,它的最大载重是多少。
如果货车不能到达目的地,输出 −1。
本题:
kruscal最大生成树
树链剖分(剖森林!)
并查集(我又写错了啊啊啊啊啊)
别的一堆细节细节细节细节细节
代码:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<queue>
#include<cstring>
using namespace std;
const int N = 10005;
int n,m,en,ee,q,fa[N],val[N],head[N];
int father[N],son[N],size[N],depth[N],top[N],seg[N],rev[N];
bool vis[N];
struct Bian{
int from,to,w;
bool operator <(const Bian &x)const{
return w<x.w ;
}
}bian[50005];
struct Edge{
int next,to,w;
}edge[50005];
priority_queue<Bian> pq;
struct Tree{
int minn;
}tree[N*4];
int cnt;
int read(){
int x;
char ch,last=' ';
while((ch=getchar())<'0'||ch>'9') last=ch;
for(x=0;isdigit(ch);ch=getchar())
x = x*10+ch-'0';
if(last=='-') x=-x;
return x;
}
void build(int x,int y,int z){
bian[++ee].from = x;
bian[ee].to = y;
bian[ee].w = z;
}
void add(int x,int y,int l){
edge[++en].to = y ;
edge[en].next = head[x];
head[x] = en;
edge[en].w = l ;
}
int find(int u){
if(fa[u]!=u) return fa[u] = find(fa[u]);
return u;
}
void kruscal(){
while(!pq.empty()){
// cout<<"?"<<endl;
Bian tmp=pq.top();
pq.pop();
int k1=find(tmp.from);
int k2=find(tmp.to);
if(k1!=k2) {
// ++cnt;
// cout<<tmp.from<<" "<<tmp.to <<endl;
fa[k1] = k2;
// for(int i=1;i<=n;i++)
// cout<<fa[i]<<" ";
// cout<<endl;
add(tmp.from ,tmp.to ,tmp.w );
add(tmp.to ,tmp.from ,tmp.w );
}
}
//cout<<"cnt="<<cnt<<endl;
}
void dfs1(int u,int f){
// cout<<u<< ' '<<endl;
size[u] = 1;
depth[u] = depth[f]+1;
father[u] = f;
for(int p=head[u];p;p=edge[p].next){
int v=edge[p].to ;
if(v==f) continue;
vis[v] = true;
dfs1(v,u);
val[v] = edge[p].w ;
size[u]+=size[v];
if(size[son[u]]<size[v]) son[u] = v;
}
}
void dfs2(int u){
if(son[u]){
vis[son[u]] = true;
top[son[u]] = top[u];
seg[son[u]] = ++seg[0];
// cout<<son[u]<<"seg:"<<seg[0]<<endl;
rev[seg[0]] = son[u];
dfs2(son[u]);
}
for(int p=head[u];p;p=edge[p].next){
int v=edge[p].to ;
if(top[v]) continue;
vis[v] = true;
top[v] = v;
seg[v] = ++seg[0];
rev[seg[0]] = v;
dfs2(v);
}
}
void create(int rt,int l,int r){
if(l==r) {
tree[rt].minn = val[rev[l]];//Rev!!!
return;
}
int mid = (l+r)>>1;
create(rt<<1,l,mid);
create(rt<<1|1,mid+1,r);
tree[rt].minn = min(tree[rt<<1].minn,tree[rt<<1|1].minn);
}
int chaxun(int rt,int l,int r,int x,int y){
if(x<=l&&r<=y) return tree[rt].minn;
if(r<x||l>y) return 100005;
int mid = (l+r)>>1;
return min(chaxun(rt<<1,l,mid,x,y),chaxun(rt<<1|1,mid+1,r,x,y));
}
//
//void tiaoshi(int rt,int l,int r){
// if(l==r) {
// //cout<<l<<" "<<tree[rt].minn<<endl;
// return ;
// }
// int mid = (l+r)>>1;
// tiaoshi(rt<<1,l,mid);
// tiaoshi(rt<<1|1,mid+1,r);
//}
int query(int u,int v){
int tmp=100005;
while(top[u]!=top[v]){
if(depth[top[u]]<depth[top[v]]) swap(u,v);
// cout<<"check "<<top[u]<<" "<<u<<" here"<<endl;
tmp = min(tmp,chaxun(1,1,n,seg[top[u]],seg[u]));
u=father[top[u]];
}
if(depth[u]>depth[v]) swap(u,v);
// cout<<"check "<<rev[seg[u]+1]<<" "<<v<<" here"<<endl;
tmp = min(tmp,chaxun(1,1,n,seg[u]+1,seg[v]));
return tmp;
}
int main(){
n=read();m=read();
for(int i=1;i<=n;i++) fa[i] = i;
for(int i=1,x,y,z;i<=m;i++){
x=read();y=read();z=read();
if(x==y) continue;
build(x,y,z);
pq.push(bian[ee]);
}
kruscal();
for(int i=1;i<=n;i++)
if(!vis[i]){
dfs1(i,0);
val[i] = 100005;
vis[i] = true;
}
memset(vis,0,sizeof(vis));
seg[0] = 0;
for(int i=1;i<=n;i++)
if(!vis[i]){
seg[i]=++seg[0];
rev[seg[0]]=top[i]=i;
dfs2(i);
}
// for(int i=1;i<=n;i++)
// cout<<i<<" "<<father[i]<<" "<<son[i]<<" "<<depth[i]<<" "<<size[i]<<" "<<top[i]<<" "<<seg[i]<<endl;
// cout<<endl;
create(1,1,n);
// for(int i=1;i<=n;i++)
// cout<<val[i]<<" ";
// tiaoshi(1,1,n);
// cout<<endl;
q=read();
for(int i=1,x,y;i<=q;i++){
x=read();y=read();
if(find(fa[x])!=find(fa[y])) printf("-1\n");
else printf("%d\n",query(x,y));
}
return 0;
}
从昨天晚上改到今天下午 从MLE+WA到MLE+RE+WA到95到今天下午改成剖森林最后AC!
遇到的问题:
- 并查集没学到位 k1 k2要写好(小心写错)
//relative
#include<iostream>
using namespace std;
int father[20005];
int find(int x){
if (father[x]!=x) return father[x]=find(father[x]);
else return x;
}
int main(){
int n,m,q;
cin>>n>>m;
for (int i=1;i<=n;i++) father[i] = i;
for (int i=0,x,y,k1,k2;i<m;i++){
cin>>x>>y;
k1 = find(x);
k2 = find(y);
if (k1!=k2) father[k2] = k1;
}
cin>>q;
for (int i=0,x,y;i<q;i++){
cin>>x>>y;
if (find(x)==find(y)) cout<<"Yes\n";
else cout<<"No\n";
}
return 0;
}
- 树剖create总是在tree[rt] = w[?] 的问号处写错!!应该是rev[L]
- kruscal之前不用建双向边,kruscal选出来的边再双向
- 要剖森林啊啊啊啊啊啊啊啊啊啊啊啊
用Bian这个名字容易遭到别人的吐槽
2019.11.13. — 2019.11.14.