美团笔试题

三、回转寿司

题目描述：
小美请小团吃回转寿司。转盘上有N盘寿司围成一圈，第1盘与第2盘相邻，第2盘与第3盘相邻，…，第N-1盘与第N盘相邻，第N盘与第1盘相邻。小团认为第i盘寿司的美味值为A[i]（可能是负值，如果小团讨厌这盘寿司）。现在，小团要在转盘上选出连续的若干盘寿司，使得这些寿司的美味值之和最大（允许不选任何寿司，此时美味值总和为0）。



输入描述
第一行输入一个整数T（1<=T<=10），表示数据组数。

每组数据占两行，第一行输入一个整数N（1<=N<=10^5）；

第二行输入N个由空格隔开的整数，表示A[1]到A[N]（-10^4<=A[i]<=10^4）。

输出描述
每组数据输出占一行，输出一个整数，表示连续若干盘寿司的美味值之和的最大值。


样例输入
1
4
3 -2 4 -1
样例输出
6

题解：前缀+后缀最大值 或者 前缀最大值

#include <bits/stdc++.h>
#define pf printf
#define sf scanf
using namespace std;

const static int maxn=1e5+5;
const static int inf=1e9+5;
int a[maxn],pre[maxn],pre_min[maxn],saf[maxn],saf_max[maxn];

int main()
{
    int T;sf("%d",&T);
    while(T--)
    {
        int n;sf("%d",&n);
        for(int i=1;i<=n;i++) sf("%d",a+i);
        int ma=-inf,mi=inf,ans=0;
        pre[0]=0;
        for(int i=1;i<=n;i++)
        {
            pre[i]=pre[i-1]+a[i];
            mi=min(mi,pre[i]);
            pre_min[i]=mi;
        }
        saf[n+1]=0;
        for(int i=n;i>=1;i--)
        {
            saf[i]=saf[i+1]+a[i];
            ma=max(ma,saf[i]);
            saf_max[i]=ma;
        }

        for(int i=1;i<=n;i++)
            ans=max(ans,max(pre[i]-pre_min[i],pre[i]+saf_max[i+1]));
        pf("%d\n",ans);
    }
    return 0;
}

四、神秘的苹果树

题目描述：
小团找到一颗有n个节点的苹果树，以1号节点为根，且每个节点都有一个苹果，苹果都有一个颜色，但是这棵树被施加了咒术，这使得小团只能从某一个节点的子树中选取某一种颜色的拿。小团想要拿到数量最多的那种颜色的所有苹果，请帮帮她。每次她会指定一个节点t，如果小团只能从节点t的子树中选取某一种颜色的苹果，选取什么颜色能拿到最多的苹果？如果有多种颜色都可以拿同样多的苹果，输出颜色编号最小的那个对应的编号。

 

节点x的子树定义为所有将x当作祖先的节点，x也视为x的子树的一部分。



输入描述
第一行一个正整数 n 表示这颗树上节点的个数。

接下来n-1行，每行两个正整数x­­i , yi,表示树上第i条边连接的两个节点。

接下来一行n个正整数c­i，分别表示从1~n号节点上的苹果的颜色。

接下来一行一个正整数q,表示接下来有q次独立的询问。

接下来q行，每行一个正整数t表示询问：如果小团只能从节点t的子树中选取某一种颜色的苹果，选取什么颜色能拿到最多的苹果？如果有多种颜色都可以拿同样多的苹果，输出颜色编号最小的那个对应的编号。

n ≤ 5000,  1≤xi,yi,t≤n,  ci≤1000000000,  q≤10000

输出描述
输出q行，每行一个整数，表示答案。


样例输入
7
1 2
1 3
2 4
2 5
3 6
3 7
1 1 2 1 2 2 3
7
1
2
3
4
5
6
7

样例输出
1
1
2
1
2
2
3

#include <bits/stdc++.h>
#define pf printf
#define sf scanf
using namespace std;

const static int maxn=5000+5;
const static int inf=1e9+5;

vector<int>tree[maxn];
int ans[maxn][maxn];
int col[maxn],tt[maxn],d[maxn];
bool vis[maxn];
int num[maxn];
int n,ct;

void dfs(int x,int fa)
{
    ans[x][col[x]]+=1;
    for(int i=0;i<tree[x].size();i++)
    {
        int u=tree[x][i];
        if(u==fa) continue;
        dfs(u,x);
        for(int j=1;j<=ct;j++)
        {
            int t=num[j];
            ans[x][t]+=ans[u][t];
        }
    }
}

int main()
{
    sf("%d",&n);
    for(int i=1;i<=n-1;i++)
    {
        int u,v;sf("%d%d",&u,&v);
        tree[u].push_back(v);
        tree[v].push_back(u);
    }

    for(int i=1;i<=n;i++) sf("%d",d+i),tt[i]=d[i];

    sort(tt+1,tt+n+1);

    ct=0;
    for(int i=1;i<=n;i++)
    {
        col[i]=lower_bound(tt+1,tt+n+1,d[i])-tt;
        if(vis[col[i]]) continue;
        vis[col[i]]=1;
        num[++ct]=col[i];
    }

    dfs(1,1);

    int q;sf("%d",&q);
    while(q--)
    {
        int tmp;sf("%d",&tmp);
        int ma=-inf,p=0;
        for(int i=1;i<=ct;i++)
        {
            int t=num[i];
            if(ma<ans[tmp][t]) ma=ans[tmp][t],p=i;
        }

        for(int i=1;i<=n;i++)
        {
            if(col[i]==num[p])
            {
                pf("%d\n",d[i]);
                break;
            }
        }
    }
    return 0;
}