保研機試験データ構造問題まとめ

辞書の木
1、難しい問題を統計する HDU-1251 
考え方:
直接に1株の辞書の木を建てて、木を建てる過程の中で、プレフィックスの出現の回数を統計します.検索時に文字列の最後の文字の対応する位置のsum値を直接確認することができます.
コード:

#include
#include
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define dep(i,a,b) for(int i=a;i>=b;i--)
#define ll long long
#include
#include
#include
#include
#include
#include 
#include
using namespace std;
const int maxn=1000000+10;
#define mod 1000000007
#define INF 0x3f3f3f3f
int dx[]= {-1,1,0,0};
int dy[]= {0,0,-1,1};
int n,k;
char ch[20];
int trie[maxn][30];
int sum[maxn];
int cnt;
void inserts()
{
    int s=0;
    for(int i=0;ch[i];i++)
    {
        int v=ch[i]-'a';
        if(trie[s][v]==0)
        {
            trie[s][v]=cnt++;
        }
        s=trie[s][v];
        sum[s]++;
    }
}
int finds()
{
    int s=0;
    for(int i=0;ch[i];i++)
    {
        int v=ch[i]-'a';
        if(trie[s][v]==0)
        {
            return 0;
        }
        s=trie[s][v];
    }
    return sum[s];
}
int main()
{
    cnt=1;
    while(gets(ch)&&ch[0]!=NULL)
    {
        inserts();
    }
    while(gets(ch))
    {
        int ans=finds();
        printf("%d
",ans);
    }
    return 0;
}

二尺の取り方
1、三数の和
https://leetcode-cn.com/problems/3sum/
2、四数の和
https://leetcode-cn.com/problems/4sum/
三、プレフィックスとサフィックス