洛谷 P1631 序列合并

题目描述

有两个长度都是N的序列A和B，在A和B中各取一个数相加可以得到N^2个和，求这N^2个和中最小的N个。

输入输出格式

输入格式：

第一行一个正整数N；
第二行N个整数Ai,满足Ai<=Ai+1且Ai<=10^9;
第三行N个整数Bi, 满足Bi<=Bi+1且Bi<=10^9.

输出格式：

输出仅一行，包含N个整数，从小到大输出这N个最小的和，相邻数字之间用空格隔开。

输入输出样例

输入样例#1：

3
2 6 6
1 4 8

输出样例#1：

3 6 7

【数据规模】

对于50%的数据中，满足1<=N<=1000；
对于100%的数据中，满足1<=N<=100000。

分析

其实算法很简单，将A与B序列从大到小排列，再将A序列第一项与B序列所有项加上入队得到初始队列，
之后再比较A中第2项与B中所有项的和，如果有一个和比当前Sum的最大值（即Sum.top()）还要大，就跳出，否则就弹出Sum.top()，
入队。

代码

#include <bits/stdc++.h>

#define N 105

int ans=0;
int n;
int len[N],use[N];
std::string s[N];

bool find(int l,int x,int y)
{
    for (int i = l; i <= len[x] - 1; i++)
    {
        int j = i - l;
        if (s[x][i] != s[y][j])
            return false;
    }
    return true;
}

int getSame(int x,int y)
{
    for (int i = len[x] - 1; i >= 0; i--)
    {
        int same = len[x] - i;
        if(find(i,x,y))
            return same;
    }
    return 0;
}

void dfs(int l,int dep)
{
    if (l > ans)
        ans=l;
    use[dep] ++;
    for (int i = 1; i <= n; i++)
    {
        int same = getSame(dep,i);
        if (same > 0 && use[i] < 2)
            if (same  != len[i])
                dfs(l + len[i] - same,i);
    }
    use[dep] --;
}

int main()
{
    scanf("%d",&n);
    for (int i = 1; i <= n; i++)
        std::cin>>s[i],len[i] = s[i].size();

    char start;
    std::cin>>start;
    for (int i = 1; i <= n; i++)
        if (s[i][0] == start)
            dfs(len[i],i);

    printf("%d\n",ans);
}