Week13 作业（树形dp，单调队列优化dp）

思路：最基础的树形dp，dp[i][0|1]代表不摘/摘i号果子所得到的最大快乐值。
状态转移方程：

dp[x][0]+=max(dp[G[x][i]][0],dp[G[x][i]][1]);
dp[x][1]+=dp[G[x][i]][0];

是不是一看就明白呀，如果汹汹没看清楚，就看完整代码：

#include<iostream>
#include<cstring>
#include<vector>
#include<algorithm>
using namespace std;
vector<int> G[6005];
int w[6005],dp[6005][2];//dp[i][0|1],摘不摘以i为节点的子树 
bool vis[6005];
bool son[6005];
int n,a,b;
void dfs(int x){
 if(vis[x]) return;
 dp[x][0]=0;
 dp[x][1]=w[x];
 for(int i=0;i<G[x].size();i++){
  dfs(G[x][i]);
  dp[x][0]+=max(dp[G[x][i]][0],dp[G[x][i]][1]);
  dp[x][1]+=dp[G[x][i]][0];
 }
 vis[x]=1;
} 
int main(){
 while(cin>>n){
  if(n==0) break;//given unclear condition, this shall be the correct solution 
  memset(w,0,sizeof(w));
  memset(dp,0,sizeof(dp));
  memset(vis,0,sizeof(vis));
  memset(son,0,sizeof(son));
  for(int i=1;i<=n;i++){
   cin>>w[i];
  }
  for(int i=1;i<n;i++){
   cin>>a>>b;
   G[b].push_back(a);
   son[a]=1;  
  }
  int f=1;
  for(int i=1;i<=n;i++){
   if(son[i]==0){//just one tree
    f=i;
    break;
   }
  }
  dfs(f);
  cout<<max(dp[f][1],dp[f][0])<<endl;
 }
 return 0;
}

E

思路:
单调队列优化dp。
题目要求环形最大连续和，我们将数列的前n-1个数接到后面，替代环。
求连续和的方法很简单，我们用前缀和就行，以sum[i]-sum[j]表示区间[j,i]的连续和。
至于求最大连续和，我这有一个绝妙的做法，而且这个页面也写得下，那就是利用单调队列来优化查找最大连续和的过程。这里使用单调递增队列。
我们需要用到双向队列deque，元素从队尾进，队尾元素大于队首元素。
队尾存当前遍历到的元素下标，如果当前遍历到的元素小于队尾元素对应值，就取出队尾直到当前元素大于队尾元素对应值为止。
队列首存前缀和最小的元素的下标，如果队首的下标相距当前遍历到的元素下标已经超过k，就取出队首。
这个过程用代码来阐述会比较直观：

for(int i=1;i<=2*n;i++){
   while(!q.empty()&&sum[q.back()]>sum[i-1])
    q.pop_back()；
   q.push_back(i-1);
   if(i-q.front()>k) 
    q.pop_front();
   if(sum[i]-sum[q.front()]>ans){
    left=q.front()+1;//guarantee the least left and right
    right=i;
    if(right>n)
     right-=n;
    ans=sum[i]-sum[q.front()];
   }
  }

完整代码：

#include<cstdio>
#include<cmath>
#include<queue>
#include<vector>
#include<cstring>
using namespace std;
int main(){
 int T,n,k,a[100010],sum[200010];
 deque<int> q;
 scanf("%d",&T);
 while(T--){
  scanf("%d%d",&n,&k);
  q.clear();
  memset(a,0,sizeof(a));
  memset(sum,0,sizeof(sum));
  for(int i=1;i<=n;i++){
   scanf("%d",&a[i]);
   sum[i]=sum[i-1]+a[i];
  }
  for(int i=n+1;i<=2*n;i++){
   sum[i]+=sum[i-1]+a[i-n];
  }
  int ans=-1e9;
  int left=0,right=0;
  for(int i=1;i<=2*n;i++){
   while(!q.empty()&&sum[q.back()]>sum[i-1])
    q.pop_back();
   q.push_back(i-1);
   if(i-q.front()>k) 
    q.pop_front();
   if(sum[i]-sum[q.front()]>ans){
    left=q.front()+1;//guarantee the least left and right
    right=i;
    if(right>n)
     right-=n;
    ans=sum[i]-sum[q.front()];
   }
  }
  printf("%d %d %d\n",ans,left,right);
 // cout<<ans<<" "<<left<<" "<<right<<endl;
 }
 return 0;
}