2024牛客寒假算法基础集训营4补题（G题,H题,F题）

G题：数三角形（easy）

题意：

求满足条件的等腰三角形个数

思路

从每个三角形的顶点开始往下搜索，用前缀和判断能不能连成一条线，统计个数即可

代码

#include <bits/stdc++.h>
using namespace std;
#define ll long long
const ll N=2e5+5;
const ll mod=1e9+7;
#define inf 0x7f7f7f7f7f7f
char s[550][550];
ll n,m;
ll a[550][550];
ll b[550][550];//前缀和
int main()
{
	ios::sync_with_stdio(false);    //快读 
	cin.tie(0);
	cout.tie(0);
    ll ans=0;
    cin>>n>>m;
    for(ll i=0;i<n;i++)
    {
        for(ll j=0;j<m;j++)
        {
            cin>>s[i][j];
            if(s[i][j]=='*') a[i][j]=1;
        }
    }
    
    for(ll i=0;i<n;i++)
    {
        for(ll j=0;j<m;j++)
        {
            if(j==0) b[i][j]=a[i][j];
            else b[i][j]=b[i][j-1]+a[i][j];
        }
    }

    for(ll i=0;i<n;i++)
    {
        for(ll j=0;j<m;j++)
        {
            if(s[i][j]=='*')
            {
                ll k=i+1,l=j-1,r=j+1;
                while(k<n&&k>=0&&l>=0&&l<m&&r>=0&&r<m)//出界退出循环
                {
                    if(s[k][l]!='*'||s[k][r]!='*') break;//三角形腰不满足
                    if(b[k][r]-b[k][l]==r-l) ans++;//如果底边满足ans++
                    k++,l--,r++;//下一层
                }
            }
        }
    }
    cout<<ans;
    
    
	
	return 0;
}

H题：数三角形（hard)

题意：

通easy版本，但是n，m范围变成（1，3000）

思路

一个等腰三角形可以由一个顶点和一个底边中点确定，考虑枚举每一个顶点，求的在这个顶点的下方的能作为底边且有贡献的有多少行（顶点和底边中点同一列并且 所以我们采取一列一列枚举，并且从下往上的方法，从下往上是因为我们先要将底边的贡献放进树状数组里，同时往上的过程中还要不断删除底边的贡献）。当然，在进行这些操作之前，我们先要预处理好四样东西：                           

 sl[N][N];//以(i,j)为顶点，左腰最长长度
 sr[N][N];//以(i,j)为顶点，右腰最长长度
 ls[N][N];//在第i行上，(i,j)左侧包括本身有几个连续点
 rs[N][N];//在第i行上，(i,j)右侧包括本身有几个连续点 

 同时我们要不断修改项，也要不断求前n项和，所以选取树状数组最优（两种操作都是logN)。

 我们还要用一个开一个vector<vector<ll>>(n+1) 用来存放那些需要被删除元素（对一个作为底边   的点，它的最高顶点位置，注意不要越界），v[i][j]即遍历完第i行后，第j行将不能再作为底边提供   贡献（第i行作为顶点已经是第j行作为底边的最高位置了）。

代码

#include <bits/stdc++.h>
using namespace std;
#define ll int//这里不能用long long了，会超内存
const ll N=3e3+10;
char a[N][N];//存放矩阵
ll n,m;
ll sl[N][N];//以(i,j)为顶点，左腰最长长度
ll sr[N][N];//以(i,j)为顶点，右腰最长长度
ll ls[N][N];//在第i行上，(i,j)左侧包括本身有几个连续点
ll rs[N][N];//在第i行上，(i,j)右侧包括本身有几个连续点
vector<long long>tr;//树状数组

//取二进制最后一位1
ll lowbit(ll x){return x & -x;}


//修改第x项元素加d
void add(ll x,ll d){
    for(ll i=x;i<=n;i+=lowbit(i)) tr[i]+=d;
}

//询问前N项和
ll ask(ll x){
    ll res=0;
    for(ll i=x;i;i-=lowbit(i)) res+=tr[i];
    return res;
}

int main()
{
    cin>>n>>m;
    
    for(ll i=1;i<=n;i++)
    {
        for(ll j=1;j<=m;j++)
        {
            cin>>a[i][j];
            //预处理(i,j)左侧包括本身有几个连续点
            if(a[i][j]=='*') ls[i][j]=ls[i][j-1]+1;
            else ls[i][j]=0;
        }
        for(ll j=m;j>=1;j--)
        {
            //预处理(i,j)右侧包括本身有几个连续点
            if(a[i][j]=='*') rs[i][j]=rs[i][j+1]+1;
            else rs[i][j]=0;
        }
    }
    
    for(ll i=n;i>=1;i--)
    {
        for(ll j=1;j<=m;j++)
        {
            //预处理以(i,j)为顶点，左腰最长长度和以(i,j)为顶点，右腰最长长度
            if(a[i][j]=='*')
            {
                sl[i][j]=sl[i+1][j-1]+1;
                sr[i][j]=sr[i+1][j+1]+1;
            }
            else
            {
                sl[i][j]=0;
                sr[i][j]=0;
            }
        }
    }

    long long ans=0;

    for(ll j=1;j<=m;j++)
    {
        tr=vector<long long>(n+1);
        //每一列开一个新的树状数组,便于修改元素以及得到前N项和（都是logN的级别)

        vector<vector<ll>>v(n+1);
        //用来存放需要删除的点(i,j)
        
        for(ll i=n;i>=1;i--)
        {
            //没有贡献的点直接删
            if(a[i][j]=='.')
            {
                for(auto x:v[i]) add(x,-1);
                continue;
            }
            //统计(i,j)做顶点
            ll l=i;//第i行
            ll r=i+min(sl[i][j],sr[i][j])-1;//两腰最大能到第r行
            ans+=ask(r)-ask(l);//在第l+1行和第r行之间能够提供的贡献之和

            //if(ask(r)-ask(l)!=0) cout<<i<<' '<<j<<' '<<l<<' '<<r<<' '<<ask(r)<<' '<<ask(l)<<endl;
            
            add(i,1);//第i行贡献加1
            
            ll lim=i-min(ls[i][j],rs[i][j])+1;
            //以(i,j)为底边中点，所对应的顶点位置(i,lim)
            //是这个点能提供贡献的最上方的一行，即第lim行
            lim=max(0,lim);//防止越界
            v[lim].push_back(i);
            //在运行到第lim行时
            //第i行这个元素不再作为底边为第lim行的顶点提供构成等腰三角形的贡献
            //当遍历完第lim行后将这些点，即v[lim]这个数组里的元素贡献删除
            
            //if(lim==i) cout<<i<<' '<<j<<endl;
            //cout<<i<<' '<<j<<' ';
            //cout<<lim<<' '<<endl;
            
            //删掉第i行不再提供底边贡献的元素所对应行
            for(auto x:v[i]) add(x,-1);
           
//             for(auto x:v[i]) cout<<i<<"-"<<x<<"-";
//             cout<<endl;
        }
    }
    
    cout<<ans<<endl;
    return 0;
}

F题：来点每日一题

题意:

按题给要求选数，求最大得分

思路

考察了区间DP。解法:先进行区间DP,然后合并区间即可
注意:有乘法，可能会负负得正，所以要同时维护最大和最小值

代码

#include <bits/stdc++.h>
using namespace std;
#define ll long long
const ll N=1010;
const ll mod=1e9+7;
const ll inf=1e18+5;
#define inf 0x7f7f7f7f7f7f
//思路：先进行区间DP，然后合并区间即可
//注意:有乘法，可能会负负得正，所以要同时维护最大和最小值
ll mi[N][N][7];//区间i到j选k个数的min
ll ma[N][N][7];//区间i到j选k个数的max

ll dp[N];//保存合并后区间后的最大值

ll a[N];

int main()
{
	ios::sync_with_stdio(false);    //快读 
	cin.tie(0);
	cout.tie(0);
	ll n;
    cin>>n;
    for(ll i=1;i<=n;i++) cin>>a[i];
    for(ll i=0;i<=n;i++)
    {
        for(ll j=0;j<=n;j++)
        {
            for(ll k=1;k<=6;k++)
            {
                mi[i][j][k]=inf;
                ma[i][j][k]=-inf;
            }
        }
    }
    
    
    //区间DP
    for(ll i=1;i<=n;i++)
    {
        for(ll j=i;j<=n;j++)
        {
            //维护区间最大值
            ma[i][j][1]=max(a[j],ma[i][j-1][1]);
            if(j-i+1>=2) ma[i][j][2]=max(ma[i][j-1][2],ma[i][j-1][1]-a[j]); 
            if(j-i+1>=3) ma[i][j][3]=max({ma[i][j-1][3],ma[i][j-1][2]*a[j],mi[i][j-1][2]*a[j]});
            if(j-i+1>=4) ma[i][j][4]=max(ma[i][j-1][4],ma[i][j-1][3]-a[j]);
            if(j-i+1>=5) ma[i][j][5]=max({ma[i][j-1][5],ma[i][j-1][4]*a[j],mi[i][j-1][4]*a[j]});
            if(j-i+1>=6) ma[i][j][6]=max(ma[i][j-1][6],ma[i][j-1][5]-a[j]);
            
            //维护区间最小值
            mi[i][j][1]=min(a[j],mi[i][j-1][1]);
            if(j-i+1>=2) mi[i][j][2]=min(mi[i][j-1][2],mi[i][j-1][1]-a[j]); 
            if(j-i+1>=3) mi[i][j][3]=min({mi[i][j-1][3],mi[i][j-1][2]*a[j],ma[i][j-1][2]*a[j]});
            if(j-i+1>=4) mi[i][j][4]=min(mi[i][j-1][4],mi[i][j-1][3]-a[j]);
            if(j-i+1>=5) mi[i][j][5]=min({mi[i][j-1][5],mi[i][j-1][4]*a[j],ma[i][j-1][4]*a[j]});
            if(j-i+1>=6) mi[i][j][6]=min(mi[i][j-1][6],mi[i][j-1][5]-a[j]);
        }
    }
    

    
    //合并区间,i表示当前区间的结尾，j表示上一区间结尾
    for(ll i=1;i<=n;i++)
    {
        for(ll j=0;j<=n;j++)
        {
            if(i-j+1>=6) dp[i]=max({dp[i],dp[j]+ma[j+1][i][6]}); 
            else dp[i]=max(dp[i],dp[j]);
        }
    }
    
	cout<<dp[n];
    
	return 0;
}