NOIP 2014 提高组 复赛 第一天 第二题 联合权值 link AC代码(最大值 次大值 乘法原理 加法结合律)+30分代码(Floyd)+70分代码(树的遍历+回溯算法)
总目录详见:NOIP 提高组 复赛 试题 目录 信奥 历年
在线测评地址:https://www.luogu.com.cn/problem/P1351
1.AC代码(最大值 次大值 乘法原理 加法结合律)
思路部分摘自:https://www.cnblogs.com/zbtrs/p/5750387.html
分析:注意到n个点n-1条边,说明了这是一棵树,每两个点之间只有一条路径,因为距离为2,这一条路径必定经过一个点,而这两个点在这一个点的左右,那么思路来了,我们可以枚举中间的一个点,然后枚举它所连到的另外两个点,去最大值和第二大值.可以用邻接表来实现.那么怎么求和最好呢?假设一个点与a,b,c相连.那么和就是a * b + a * c + b * a + b * c + c * a + c * b = a * (b + c) + b * (a + c) + c * (a + b),可以注意到这是一个对称式,a,b,c是不能化简得,那么后一项发现是总的和减去前一项,那么算法就出来了,设o = a + b + c,那么sum = a * (o - a) + b * (o - b) + c * (o - c),不过注意到和需要取余,以前我比较恶心取余的题,但是现在知道了一个技巧,凡是结果可能大于mod的数就取余,此题还有一个坑点就是权值可能就是负数,那么就要加上mod再取余.
AC代码(最大值 次大值 乘法原理 加法结合律) 代码如下:
#include <cstdio>
#include <algorithm>
#define mod 10007
#define maxn 200010
using namespace std;
int n,head[maxn],tot,w[maxn],ans1,ans2;
struct node{
int to,next;
}e[maxn<<1];
void add_edge(int u,int v){
tot++,e[tot].to=v,e[tot].next=head[u],head[u]=tot;
}
void solve(){
int b,v,u,sum,mx1,mx2;//mx1最大值,mx2次大值
for(u=1;u<=n;u++){
mx2=mx1=sum=0;
for(b=head[u];b;b=e[b].next){//寻找以u为中点,相邻的两点中的最大值,次大值
v=e[b].to;
sum=(sum+w[v])%mod;//与u相邻的所有元素权值之和
if(w[v]>mx1)mx2=mx1,mx1=w[v];
else{//w[v]<=mx1
if(w[v]>mx2)mx2=w[v];
}
}
ans1=max(ans1,mx1*mx2);
for(b=head[u];b;b=e[b].next){
v=e[b].to;
ans2=(ans2+w[v]*(sum-w[v]))%mod;
}
}
ans2=(ans2%mod+mod)%mod;//ans2可能是负数,需进行处理
printf("%d %d\n",ans1,ans2);
}
int main(){
int i,u,v;
scanf("%d",&n);
for(i=1;i<n;i++){
scanf("%d%d",&u,&v);
add_edge(u,v),add_edge(v,u);//无向图
}
for(i=1;i<=n;i++)scanf("%d",&w[i]);
solve();
return 0;
}
2.30分代码(Floyd)
30分代码(Floyd) 如下:
#include <cstdio>
#include <algorithm>
#define mod 10007
#define maxn 110
using namespace std;
int n,w[maxn],map[maxn][maxn],sum,mx;
int main(){
int u,v,i,j,k;
scanf("%d",&n);
for(i=1;i<=n;i++)
for(j=1;j<=n;j++)
if(i==j)map[i][j]=0;
else map[i][j]=maxn;
for(i=1;i<n;i++){
scanf("%d%d",&u,&v);
map[u][v]=map[v][u]=1;
}
for(i=1;i<=n;i++)scanf("%d",&w[i]);
for(k=1;k<=n;k++)
for(i=1;i<=n;i++)
for(j=1;j<=n;j++)
if(map[i][j]>map[i][k]+map[k][j])
map[i][j]=map[i][k]+map[k][j];
for(i=1;i<=n;i++)
for(j=1;j<=n;j++)
if(map[i][j]==2)sum=(sum+w[i]*w[j])%mod,mx=max(mx,w[i]*w[j]);
printf("%d %d\n",mx,sum);
return 0;
}
3.70分代码(树的遍历)
70分代码(树的遍历) 如下:
#include <cstdio>
#include <algorithm>
#define maxn 200010
#define mod 10007
using namespace std;
struct node{
int to,next,w;
}e[maxn<<1];
int n,head[maxn],tot,cnt,sum,mx,w[maxn];
void add_edge(int u,int v){
tot++,e[tot].to=v,e[tot].next=head[u],head[u]=tot;
}
void dfs(int x,int u,int fa){//x标记距离为2的线段的起始点
int b,v;
if(cnt==2){
sum=(sum+w[x]*w[u])%mod;
mx=max(mx,w[x]*w[u]);
return;
}
for(b=head[u];b;b=e[b].next){
v=e[b].to;
if(v==fa)continue;
cnt++;//回溯算法
dfs(x,v,u);
cnt--;//回溯算法
}
}
int main(){
int u,v,i;
scanf("%d",&n);
for(i=1;i<n;i++){
scanf("%d%d",&u,&v);
add_edge(u,v),add_edge(v,u);
}
for(i=1;i<=n;i++)scanf("%d",&w[i]);
for(i=1;i<=n;i++){
cnt=0;
dfs(i,i,0);
}
printf("%d %d\n",mx,sum);
return 0;
}