程序设计思维week13 选做题

TT的苹果树（没有上司的舞会）

在大家的三连助攻下，TT 一举获得了超级多的猫咪，因此决定开一间猫咖，将快乐与大家一同分享。并且在开业的那一天，为了纪念这个日子，TT 在猫咖门口种了一棵苹果树。
一年后，苹果熟了，到了该摘苹果的日子了。
已知树上共有 N 个节点，每个节点对应一个快乐值为 w[i] 的苹果，为了可持续发展，TT 要求摘了某个苹果后，不能摘它父节点处的苹果。
TT 想要令快乐值总和尽可能地大，你们能帮帮他吗？
Input
结点按 1～N 编号。
第一行为 N (1 ≤ N ≤ 6000) ，代表结点个数。
接下来 N 行分别代表每个结点上苹果的快乐值 w[i]（-128 ≤ w[i] ≤ 127)。
接下来 N-1 行，每行两个数 L K，代表 K 是 L 的一个父节点。
输入有多组，以 0 0 结束。
Output
每组数据输出一个整数，代表所选苹果快乐值总和的最大值。
输入样例
7
1
1
1
1
1
1
1
1 3
7 4
2 3
4 5
6 4
3 5
0 0
输出样例
5
解题思路
没有上司的舞会是树形dp经典例题。树形dp就是在树上跑的dp，经常会结合dfs等方法。这道题首先通过入度确定根节点，设计状态为f[i][2],f[i][]表示以I为根节点的子树能达到的最大快乐值，第二维0代表不取这个节点，1就代表取这个节点，初始化所有的f[i][1]都为w[i],要求的答案为max(f[root][1],f[root][0])

#include<iostream>
#include <cstdio>
#include <algorithm>
#include <string>
#include <vector>
#include <cstring>
#include <queue>
#include <cmath>
using namespace std;
#define ll long long
const int maxn = 6005;
int n,m;
int w[maxn],du[maxn];
vector<int>son[maxn];//存每个结点的儿子
ll f[maxn][2];
void dfs(int x)
{
    f[x][1] = w[x];
    for(int i=0;i<son[x].size();i++)
    {
        dfs(son[x][i]);
        f[x][0] += max(f[son[x][i]][0],f[son[x][i]][1]);//不取这个点，增加的最大值为是否取儿子节点的最大值
        f[x][1] += f[son[x][i]][0];//取这个点，儿子就不能取了
    }
}
int main()
{
    while(cin>>n)
    {
        if(n==0)break;
        for(int i=1;i<=n;i++)
        {
            scanf("%d",w+i);
            son[i].clear();
        }
        memset(du,0, sizeof(du));
        memset(f,0, sizeof(f));
        for(int i=1;i<n;i++)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            son[y].push_back(x);
            du[x]++;
        }
        int i=1;
        for(i=1;i<=n;i++)if(du[i]==0)break;
        dfs(i);
        /*for(int j=1;j<=n;j++)
        {
            cout<<j<<"   ";
            for(int k=0;k<son[j].size();k++)
            {
                cout<<son[j][k]<<" ";
            }
            cout<<"\n";
        }*/
        cout<<max(f[i][0],f[i][1])<<"\n";
    }
    return 0;
}

TT的神秘任务3

TT 猫咖的生意越来越红火，人越来越多，也越来越拥挤。
为了解决这个问题，TT 决定扩大营业规模，但猫从哪里来呢？
TT 第一时间想到了神秘人，想要再次通过完成任务的方式获得猫咪。
而这一次，神秘人决定加大难度。
给定一个环，A[1], A[2], A[3], … , A[n]，其中 A[1] 的左边是 A[n]。要求从环上找出一段长度不超过 K 的连续序列，使其和最大。
这一次，TT 陷入了沉思，他需要你们的帮助。
Input
第一行一个整数 T，表示数据组数，不超过 100。
每组数据第一行给定两个整数 N K。（1 ≤ N ≤ 100000, 1 ≤ K ≤ N）
接下来一行，给出 N 个整数。（-1000 ≤ A[i] ≤ 1000）。
Output
对于每一组数据，输出满足条件的最大连续和以及起始位置和终止位置。
如果有多个结果，输出起始位置最小的，如果还是有多组结果，输出长度最短的。
Sample Input
4
6 3
6 -1 2 -6 5 -5
6 4
6 -1 2 -6 5 -5
6 3
-1 2 -6 5 -5 6
6 6
-1 -1 -1 -1 -1 -1
Sample Output
7 1 3
7 1 3
7 6 2
-1 1 1
一道单调队列应用题，同时也考察了破环成链、前缀和等实用技巧。
破环成链一般都是开两倍数组，把1~n的数据再复制到n+1到2n中
这个题目序列最长为k，也就是破出来的最大链为n+k
最大连续序列和一定会用到前缀和来解决，单调队列尾指针控制队列中的sum始终是递增的，头指针则控制选择的区间长度不超过k

#include<iostream>
#include <cstdio>
#include <algorithm>
#include <string>
#include <vector>
#include <cstring>
#include <queue>
#include <cmath>
using namespace std;
#define ll long long
const int maxn = 2000005;
ll sum[maxn];
int n,k;
int de_queue[maxn];
ll ans,ansl,ansr;
int main()
{
    int T;
    cin>>T;
    while(T--)
    {
        ans = 0;
        ansl = 0;
        ansr = 0;
        sum[0] = 0;
        cin>>n>>k;
        for(int i=1;i<=n;i++)
        {
            scanf("%lld",sum+i);
            sum[i] += sum[i-1];
        }
        for(int i=n+1;i<=n+k;i++)
        {
            sum[i] = sum[n] + sum[i-n];
        }
        int h=0,t=0;
        ans = -2147483647;
        for(int i=1;i<=n+k-1;i++)
        {
            while(h<t && sum[i-1] < sum[de_queue[t-1]])
            {
                t--;
            }
            de_queue[t] = i-1;
            t++;
            while(h<t && i-de_queue[h]>k)
            {
                h++;
            }
            if(sum[i] - sum[de_queue[h]] > ans)
            {
                ans = sum[i] - sum[de_queue[h]];
                ansl = de_queue[h]+1;
                ansr = i;
                if(ansr>n)ansr%=n;
            }
        }
        cout<<ans<<" "<<ansl<<" "<<ansr<<"\n";
    }
    return 0;
}