A. Yet Another Tetris Problem(思维)
B. Yet Another Palindrome Problem(思维)
C. Frog Jumps(思维)
D . Pair of Topics (思维+二分)
E . Sleeping Schedule (O(n^2) dp)
F . Maximum White Subtree (换根dp)
A. Yet Another Tetris Problem(思维)
题意描述: 类似俄罗斯方块(每列加偶数个,每次消去满层的),问是否能把所有方块消去;
题解: 每组数奇偶性质都相同时输出YES
#include<bits/stdc++.h>
using namespace std;
int main()
{
int T;
cin>>T;
while(T--){
int n; cin>>n;
int odd=0,even=0;
for(int i=0;i<n;i++) {
int t; cin>>t;
if(t&1) odd++;
else even++;
}
if(odd==n||even==n) cout<<"YES\n";
else cout<<"NO\n";
}
return 0;
}
B. Yet Another Palindrome Problem(思维)
题意描述: 在子序列中找长度 >= 3的回文串;
题解: 只要存在两个数相同并且这两个数中间还有不少于一个数就YES;
那么就只要两层for循环就好了;
#include<bits/stdc++.h>
using namespace std;
int main()
{
int T,a[5050];
cin>>T;
while(T--){
int n;
cin>>n;
for(int i=0;i<n;i++) cin>>a[i] ;
int ans=0;
for(int i=0;i<n;i++){
int j=i,k=n-1,num=0;
while(j<=k){
if(a[j]==a[k]) k--,j++,num+=2;
else k--;
}
ans = max(ans,num);
}
if(ans>=3) cout<<"YES\n";
else cout<<"NO\n";
}
return 0;
}
C. Frog Jumps(思维)
题意描述: 每个格子只能按字母‘L’或‘R’ 往左右两边跳 d 个格子,求从0到n+1的最小d;
题解: 实际只考虑R的位置就好了,找到相邻R的最大值就是最小d,起点和终点都处理成R就可以了;
#include<bits/stdc++.h>
using namespace std;
int main()
{
int T;
cin>>T;
while(T--){
string s ;
cin>>s;
s = "R"+s;
s.push_back('R');
int n = s.size();
vector<int> ans;
for(int i=0;i<n;i++){
if(s[i]=='R') ans.push_back(i);
}
int res=1;
for(int i=1;i<ans.size();i++){
res = max(res,ans[i]-ans[i-1]);
}
cout<<res<<endl;
}
return 0;
}
D . Pair of Topics (思维+二分)
题意描述: 寻找满足上式的a,b的对数。
题解: 令 x = a - b; 那么就是寻找 xi+xj > 0,这就化简成了,找 x 数组中两两之和大于0的对数;
排序后二分查找一下就好了,;
#include<bits/stdc++.h>
#define LL long long
using namespace std;
vector<int> a;
int n,b;
int main()
{
scanf("%d",&n);
for(int i=0;i<n;i++) scanf("%d",&b),a.push_back(b);
for(int i=0;i<n;i++) scanf("%d",&b), a[i] -= b;
sort(a.begin(),a.end());
LL ans=0;
for(int i=0,j=1;j<n;i++,j++)
ans += a.end() - upper_bound(a.begin()+j,a.end(),0-a[i]) ;
cout<<ans<<endl;
return 0;
}
E . Sleeping Schedule (O(n^2) dp)
题意描述:
七个数 ai 相加,每次可以选择加ai 或 ai-1,每次相加后求余24,如果结果在[21,23] 内 就是好睡眠,求最大好睡眠数;
题解: 很明显的dp,因为每个步骤对应只有两种情况,也可以用记忆递归写。
1、DP: dp[i][j] 代表 在第i天时 前 i 天减去了j个1 所获得的好睡眠数量;
#include<bits/stdc++.h>
using namespace std;
int n,h,l,r,sum=0;
int a[2010];
int dp[2010][2010];
int main()
{
cin>>n>>h>>l>>r;
for(int i=1;i<=n;i++) cin>>a[i];
for(int i=1;i<=n;i++)
{
sum += a[i];
for(int j=0;j<=i;j++)
{
int t = (sum-j)%h ;
if(j==0) dp[i][j] = dp[i-1][j] + (t>=l&&t<=r) ; //都不减一,直接赋值
else dp[i][j] = max(dp[i-1][j],dp[i-1][j-1]) + (t>=l&&t<=r) ; //否则就计算 前i天分别减j得到的值,然后取最大;
}
}
cout<<*max_element(dp[n],dp[n]+n+1)<<endl; //查询下标 [0] ~ [n]
}
2、记忆递归(dp的转换):
#include<bits/stdc++.h>
using namespace std;
int n,h,l,r,a[2020];
int dp[2020][2020];
int DFS(int i,int sum) // 第 i 天 ,在sum 点醒来;
{
if(i>=n) return 0 ; //超过n天为零
if(dp[i][sum]!=-1) return dp[i][sum]; //i,sum出现过就不用再递归;
int x = (a[i]+sum) % h;
int c = DFS(i+1,x) + (x>=l&&x<=r); // 每次不-1;
int y = (a[i]-1+sum) % h;
int d = DFS(i+1,y) + (y>=l&&y<=r); // 每次 -1;
return dp[i][sum] = max (c , d); //两种选择取最大值;
}
int main()
{
cin>>n>>h>>l>>r;
for(int i=0;i<n;i++) cin>>a[i];
memset(dp,-1,sizeof(dp));
cout<<DFS(0,0)<<endl; //下标 , 累和
return 0;
}
F . Maximum White Subtree (换根dp)
题意描述: 给一颗树(无根),每个节点非黑即白,求以每个节点为根节点的子树(这里的子树相对于其他节点作为根来说)中num白 - num黑的最大值。
题解: 貌似和DP没太大关系,因为没用严谨的状态转移方程,其核心思想是换根。 给一个联通无向图,寻找以每个点出发的联通子图里num白 - num黑的最大值,那么这里我们采用换根dp来做比较简单,详情看注释解释换根过程。
#include<bits/stdc++.h>
using namespace std;
const int MAX = 2e5+7;
int col[MAX];
vector<int> edge[MAX];
int dp[MAX];
int ans[MAX];
void Root(int x,int f)
{
dp[x] = (col[x]==1 ? 1:-1); //初始化节点值
for(auto v :edge[x]){ // v是x的 孩子
if(v==f) continue;
Root(v,x);
if(dp[v] > 0) dp[x] += dp[v]; //自叶子节点从下往上依次更新每个节点的最优值
}
}
void CRoot(int x,int f)
{
for(auto v : edge[x]){ //v是x的孩子
if(v==f) continue;
int xx = dp[x], vv = dp[v]; //与孩子换根
if(dp[v] > 0) dp[x] -= dp[v]; //若此孩子对父节点有贡献,则父节点减去它的贡献,因为该反过来贡献了
if(dp[x] > 0) dp[v] += dp[x]; //换根,就是改成父节点对儿子做贡献;
ans[v] = dp[v]; //存以v为根的最大值;
CRoot(v,x); //相当于前序遍历换根;
dp[v] = vv, dp[x] = xx ; //到叶子节点后,依次换回跟,直到访问过所有节点
}
}
int main()
{
int n;
cin>>n;
for(int i=1;i<=n;i++) scanf("%d",&col[i]);
for(int i=1;i<n;i++) {
int x,y;scanf("%d %d",&x,&y);
edge[x].push_back(y); //建图
edge[y].push_back(x);
}
Root(1,0); //先去找以1为根的答案;
ans[1] = dp[1]; //答案存进ans
CRoot(1,0); //利用之前保存的dp[]值,进行换根操作;
for(int i=1;i<=n;i++)
if(i!=n) printf("%d ",ans[i]);
else printf("%d\n",ans[i]);
return 0;
}