京东 算法校招笔试满分代码

第一题 01字符串分割：

#include <bits/stdc++.h>

using namespace std;
map<pair<int, int>, int>mp;
// 1 0
// 0 1
const int maxn=5e5+5;
char a[maxn];
int ans[maxn];
int main()
{
    pair<int ,int>p;
    int zero=0, one=0;
    int n;
    cin>>n;
    scanf("\n%s", a);
    for(int i=0; i<n; i++){
        if(a[i]=='0')zero++;
        else one++;
        if(zero==0){
            p.first=0, p.second=1;
            mp[p]=mp[p]+1;
        }
        else if(one==0){
            p.first=1, p.second=0;
            mp[p]=mp[p]+1;
        }
        else{
            int gcd=__gcd(zero, one);
            p.first=zero/gcd;
            p.second=one/gcd;
            mp[p]=mp[p]+1;
        }
        ans[i]=mp[p];
    }
    for(int i=0; i<n; i++)printf(i==n-1?"%d\n":"%d ", ans[i]);
    return 0;
}

第二题 分竹子，a,b,c三种长度竹子：

#include <bits/stdc++.h>

using namespace std;
int dp[10005];
int main()
{
    int n, a, b, c;
    cin>>n>>a>>b>>c;
    for(int i=0; i<=n; i++){
        if(i==0 || dp[i]){
        dp[i+a]=max(dp[i+a], dp[i]+1);
        dp[i+b]=max(dp[i+b], dp[i]+1);
        dp[i+c]=max(dp[i+c], dp[i]+1);
        }
    }
    printf("%d\n", max(dp[n], 1));
    return 0;
}