cogs luogu 1804. [NOIP2014]联合权值 WD

★★   输入文件：linkb.in   输出文件：linkb.out   简单对比
时间限制：1 s   内存限制：128 MB

【题目描述】

爆搜60（cogs），然而在luogu  all  TLE：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cmath>
#include<cstring>
#define ll long long

using namespace std;
const int N=2000100;
const int mod=10007;

int head[N];
int now=1;
ll Answer;
ll AnsMax=-1;
ll va[N];

struct node{
    int u,v,nxt;
}E[N];

inline int read()
{
    int x=0,f=1;
    char c=getchar();
    while(c<'0'||c>'9') {if(c=='-')f=-1; c=getchar();}
    while(c>='0'&&c<='9')x=x*10+c-'0',c=getchar();
    return x*f;
}

inline void add(int u,int v)
{
    E[now].u=u;
    E[now].v=v;
    E[now].nxt=head[u];
    head[u]=now;
    now++;
}

inline void bfs(int x)
{
    for(int i=head[x];i!=-1;i=E[i].nxt)
    {
        int v=E[i].v;
        for(int j=head[v];j!=-1;j=E[j].nxt)
        {
            if(E[j].v==x)continue ;
            AnsMax=max(AnsMax,va[x]*va[E[j].v]);
            Answer+=(va[x]%mod*va[E[j].v]%mod)%mod;
        }
    }
}

int main()
{
    
    freopen("linkb.in","r",stdin);
    freopen("linkb.out","w",stdout);
    int n=read();
    
    for(int i=1;i<=n;i++)
        head[i]=-1;
        
    for(int i=1;i<=n-1;i++)
    {
        int u=read();
        int v=read();
        add(u,v);
        add(v,u);
    }
    
    for(int i=1;i<=n;i++)
        va[i]=read();
    
    for(int i=1;i<=n;i++)
        bfs(i);
        
    printf("%lld %lld",AnsMax,Answer%mod);
    return 0;
}

AC 牛B 代码：

看不懂，甚至语法：

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#include<iomanip>
#include<cstdlib>
#include<vector>
#include<queue>

using namespace std;
const int SIZEN=200010,MOD=10007;

int N;
vector<int> c[SIZEN];
vector<int> son[SIZEN];
int W[SIZEN];
int fa[SIZEN]= {0};
vector<int> Blis;

class PAIR {
    public:
        int mx;
        int s;
        void print(void) {
            cout<<"("<<mx<<" "<<s<<")";
        }
        void clear(void) {
            mx=s=0;
        }
        PAIR(void) {
            mx=s=0;
        }
};

PAIR mkpr(int x,int y) {
    PAIR a;
    a.mx=x,a.s=y;
    return a;
}
PAIR operator + (PAIR a,PAIR b) {
    PAIR c;
    c.mx=max(a.mx,b.mx);
    c.s=(a.s+b.s)%MOD;
    return c;
}

PAIR P[SIZEN];
PAIR sp[SIZEN];
void update_1(int x) { //子之子
    for(int i=0; i<son[x].size(); i++) {
        int u=son[x][i];
        P[x]=P[x]+sp[u];
        sp[x]=sp[x]+mkpr(W[u],W[u]);
    }
}
PAIR pre[SIZEN],suf[SIZEN];
void update_2(int x) { //x兄弟之间
    if(son[x].size()<=1) return;
    int u=son[x].front();
    pre[0]=mkpr(W[u],W[u]);
    for(int i=1; i<son[x].size(); i++) {
        u=son[x][i];
        pre[i]=pre[i-1]+mkpr(W[u],W[u]);
        P[u]=P[u]+pre[i-1];
    }
    u=son[x].back();
    suf[son[x].size()-1]=mkpr(W[u],W[u]);
    for(int i=son[x].size()-2; i>=0; i--) {
        u=son[x][i];
        suf[i]=suf[i+1]+mkpr(W[u],W[u]);
        P[u]=P[u]+suf[i+1];
    }
}
queue<int> Q;
bool vis[SIZEN]= {0};
int f[SIZEN];
void BFS(int S) {
    memset(fa,0,sizeof(fa));
    memset(vis,0,sizeof(vis));
    while(!Q.empty()) Q.pop();
    for(int i=1; i<=N; i++) son[i].clear();
    Blis.clear();
    vis[S]=true;
    Q.push(S);
    f[S]=0;
    while(!Q.empty()) {
        int x=Q.front();
        Q.pop();
        Blis.push_back(x);
        for(int i=0; i<c[x].size(); i++) {
            int u=c[x][i];
            if(u==fa[x]) continue;
            fa[u]=x;
            son[x].push_back(u);
            f[u]=f[x]+1;
            Q.push(u);
        }
    }
}
void work_big(void) {
    BFS(1);
    for(int i=1; i<=N; i++) { //父之父
        if(fa[fa[i]]!=0) {
            int u=fa[fa[i]];
            P[i]=P[i]+mkpr(W[u],W[u]);
        }
    }
    for(int i=Blis.size()-1; i>=0; i--) update_1(Blis[i]); //子之子
    for(int i=0; i<Blis.size(); i++) update_2(Blis[i]); //兄弟
    int mx=0,sum=0;
    for(int i=1; i<=N; i++) {
        mx=max(mx,W[i]*P[i].mx);
        sum+=(W[i]*P[i].s)%MOD;
        sum%=MOD;
    }
    printf("%d %d
",mx,sum);
}
void work_small(void) { //其实这个傻叉了
    int mxw=0,wsum=0;
    for(int i=1; i<=N; i++) {
        BFS(i);
        for(int j=1; j<=N; j++) {
            if(f[j]==2) {
                int now=W[i]*W[j];
                mxw=max(mxw,now);
                wsum+=now;
                wsum%=MOD;
            }
        }
    }
    printf("%d %d
",mxw,wsum);
}
void read(void) {
    scanf("%d",&N);
    int a,b;
    for(int i=1; i<N; i++) {
        scanf("%d%d",&a,&b);
        c[a].push_back(b);
        c[b].push_back(a);
    }
    for(int i=1; i<=N; i++) scanf("%d",&W[i]);
}
int main() {
    freopen("linkb.in","r",stdin);
    freopen("linkb.out","w",stdout);
    read();
    work_big();
    return 0;
}

 高：

#include<cstdio>
#include<iostream>

using namespace std;

const int N=2e5+5,mo=10007;

struct cs{
int to,nxt;
}a[N*2];

int head[N],ll,v[N];
int n,ans,x,y,maxans;

void init(int x,int y)
{
    a[++ll].to=y;
    a[ll].nxt=head[x];
    head[x]=ll;
}

void work(int x)
{
    int sum=0,ma=0,m=0;
    for(int k=head[x];k;k=a[k].nxt)
    {
        if(v[a[k].to]>ma){m=ma;ma=v[a[k].to];}else
        if(v[a[k].to]>m)m=v[a[k].to];
        ans=(ans+sum*v[a[k].to])%mo;
        sum=(sum+v[a[k].to])%mo;
    }
    maxans=max(maxans,ma*m);
}
int main()
{
    freopen("linkb.in","r",stdin);
    freopen("linkb.out","w",stdout); 
    scanf("%d",&n);
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&x,&y);
        init(x,y); init(y,x);
    }
    
    for(int i=1;i<=n;i++)scanf("%d",&v[i]);
    for(int i=1;i<=n;i++)work(i);
    printf("%d %d",maxans,(ans*2)%mo);
    return 0;
}